在過去的幾周里，我一直潛伏在 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) }我實際上不需要向套接字發送任何數據，我只需要連接并繼續讀取它，然后我會將這些數據聚合到單個流中以執行以后的操作。
1 回答
交互式愛情
TA貢獻1712條經驗 獲得超3個贊
啟動一個 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)
}
- 1 回答
- 0 關注
- 215 瀏覽
添加回答
舉報
0/150
提交
取消