golang 读取了大量的 websockets

Question

在过去的几周里，我一直潜伏在 Stack Overflow 上寻找与阅读大量 websocket 相关的信息。基本上，我有许多主机都通过 websocket 发出消息，我需要聚合它们。到目前为止，我已经使用 Golang 完成了单个 websocket 连接。我也使用 Python 完成了我正在寻找的东西，但我真的很想在 Go 中做到这一点！我使用了 gorilla 的 websocket 示例以及其他一些示例，并且可以在 Go 中成功读取套接字。然而，websocket 服务器似乎并不完全符合典型的开发实践，如在 JS 中使用 .forEach 或 .Each 等方法；导致握手失败。原版    package main    import (            "fmt"            "golang.org/x/net/websocket"            "log"    )    var url = "ws://10.0.1.19:5000/data/websocket"    func main() {            ws, err := websocket.Dial(url, "", origin)            if err != nil {                    log.Fatal(err)            }            var msg = make([]byte, 512)            _, err = ws.Read(msg)            if err != nil {                    log.Fatal(err)            }            fmt.Printf("Receive: %s\n", msg)    }我实际上不需要向套接字发送任何数据，我只需要连接并继续读取它，然后我会将这些数据聚合到单个流中以执行以后的操作。

Answer 1

启动一个 goroutine 来读取每个连接。将收到的消息发送到频道。从该通道接收以从所有连接获取消息。

// Create channel to receive messages from all connections

messages := make(chan []byte)

// Run a goroutine for each URL that you want to dial.

for _, u := range urls {

    go func(u string) {

        // Dial with Gorilla package. The x/net/websocket package has issues.

        c, _, err := websocket.DefaultDialer.Dial(u, http.Header{"Origin":{origin}})

        if err != nil {

            log.Fatal("dial:", err)

        }

        // Clean up on exit from this goroutine

        defer c.Close()

        // Loop reading messages. Send each message to the channel.

        for {

            _, m, err := c.ReadMessage()

            if err != nil {

                log.Fatal("read:", err)

                return

            }

            messages <- m

        }

    }(u)

}

// Print all messages received from the goroutines.

for m := range messages {

    fmt.Printf("%s\n", m)

}