Question

我是Golang的新手，我正在尝试创建一个WebSocket服务器，它将向连接的客户端广播消息。此处的消息将从服务器端生成（通过创建默认客户端）。

这是我的client.go

c, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
if err != nil {
    log.Fatal("dial:", err)
}
defer c.Close()
done := make(chan struct{})
new_chan := make(chan string)
//defer new_chan.Stop()
go func() {
    for {
        new_chan <- "my message"
    }
}()

hub := newHub()
go hub.run()
client := &Client{hub: hub, ws: c, send: make(chan []byte, 256)}

for {
    select {
    case <-done:
        return
    case t := <-new_chan:
        err := c.WriteMessage(websocket.TextMessage, []byte(t))
        if err != nil {
            log.Println("write:", err)
            return
        }
        log.Printf(t)
        client.hub.broadcast <- bytes.TrimSpace(bytes.Replace([]byte(t), newline, space, -1))
    }
}

此函数将生成消息并尝试向其他客户端广播。

server.go会将客户端添加到集线器

func echo(w http.ResponseWriter, r *http.Request) {
c, err := upgrader.Upgrade(w, r, nil)
if err != nil {
    log.Print("upgrade:", err)
    return
}
hub := newHub()
go hub.run()

client := &Client{hub: hub, ws: c, send: make(chan []byte, 256)}
client.hub.register <- client

go client.writePump()

writePump（）这里将监听client.send频道和广播消息现在，连接的客户端集线器与服务器上的客户端集线器不同。所以当我尝试发送消息时，我没有收到任何消息。如何使其属于同一个集线器（上下文）？

Answer 1

看起来你是从Gorilla chat example开始的。在该示例中，使用hub.broadcast <- message向hub中*Hub创建main()的所有客户广播消息。

无需创建客户端连接即可发送来自服务器的消息。

这是main（）的修改版本，每秒向所有客户广播“hello”：

func broadcast(hub *Hub) {
    m := []byte("hello")
    for range time.NewTicker(time.Second).C {
        hub.broadcast <- m
    }
}

func main() {
    flag.Parse()
    hub := newHub()
    go hub.run()
    go broadcast(hub)
    http.HandleFunc("/", serveHome)
    http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
        serveWs(hub, w, r)
    })
    err := http.ListenAndServe(*addr, nil)
    if err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}

使用Websockets向客户端广播服务器结束消息

1 个答案: