gpt4 book ai didi

pointers - go 中的引用 - 属性不会改变

转载 作者:数据小太阳 更新时间:2023-10-29 03:05:45 24 4
gpt4 key购买 nike

我很难理解 go 中的引用是如何工作的。我想编写一个非常简单的内存中发布-订阅机制。这是代码:

package sockets

import (
"fmt"

"github.com/gorilla/websocket"
)

type hubSingleton struct {
Clients map[string][]*websocket.Conn
}

var instance *hubSingleton

func Hub() *hubSingleton {

if instance == nil {
fmt.Println("New instance created")
instance = &hubSingleton{}
}
instance.Clients = make(map[string][]*websocket.Conn, 6)
return instance
}

func (hub *hubSingleton) Subscribe(channel string, socket *websocket.Conn) error {
if _, ok := hub.Clients[channel]; !ok {
hub.Clients[channel] = make([]*websocket.Conn, 0)
}
hub.Clients[channel] = append(hub.Clients[channel], socket)

fmt.Println("Subscribe: ", hub.Clients)
return nil
}

func (hub *hubSingleton) Publish(channel string, message interface{}) {
fmt.Println("Publish: ", hub.Clients)

if _, ok := hub.Clients[channel]; !ok {
return
}
for i := 0; i < len(hub.Clients[channel]); i++ {
conn := hub.Clients[channel][i]
conn.WriteJSON(Message{status: "ok", content: message})
}
}

问题是似乎每次我调用 Hub().Publish() 或 Hub().Subscribe() 时都会创建一个新的 hubSingleton.Client。好吧,我不确定会发生什么,但这是运行程序的输出:

Publish:  map[]
Subscribe: map[chan:[0xc820170000]]
Publish: map[]
Subscribe: map[chan:[0xc82008ac80]]
Publish: map[]
Publish: map[]
Publish: map[]
Subscribe: map[chan:[0xc820170140]]
Publish: map[]

每次我调用 Hub() 时,添加到 map 的订阅者都会消失。我该怎么做才能在调用之间保留 map 中的更改?

最佳答案

Hub 函数在每次调用时都会创建一个新的客户端映射。将函数更改为:

func Hub() *hubSingleton {
if instance == nil {
fmt.Println("New instance created")
instance = &hubSingleton{}
instance.Clients = make(map[string][]*websocket.Conn, 6)
}
return instance
}

如果对 Hub 的第一次调用来自请求处理程序,则 instance 上存在数据竞争。使用锁来固定比赛:

var (
instance *hubSingleton
mu sync.Mutex
)

func Hub() *hubSingleton {
mu.Lock()
defer mu.Unlock()
if instance == nil {
fmt.Println("New instance created")
instance = &hubSingleton{}
instance.Clients = make(map[string][]*websocket.Conn, 6)
}
return instance
}

一个更简单的方法是在使用前初始化一次实例:

var instance *hubSingleton

func newHub() *hubSingleton {
return &hubSingleton{Clients: make(map[string][]*websocket.Conn, 6)}
}

func main() {
instance = newHub()
...
}

如果处理程序同时调用PublishSubscribe,则Publish 中的Clients 上存在数据竞争和订阅

关于pointers - go 中的引用 - 属性不会改变,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/39640284/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com