我最近開始探索 Go 并且非常喜歡它。我在嘗試檢測 WebSocket 連接超時時遇到了問題。我無限期地監聽 websocket 連接,當我在 X 秒內沒有得到響應時,我嘗試重新連接。為了實現這一點,我必須修改我的 for 循環以包含選擇。然后我創建了一個類型和一個通道來監聽 websocket 響應。然而,這導致我的 websocket 連接出現錯誤,提示 failed to get reader: previous message not read to Complete。我將從項目中獲取的代碼替換為獨立的。下面是包含兩個循環的完整腳本(工作和非工作可用)package mainimport ( "bytes" "context" "fmt" "io" "time" "nhooyr.io/websocket")func main() { ctx := context.Background() c, _, err := websocket.Dial(ctx, "wss://stream.binance.com:9443/ws/btcusdt@trade", nil) if err != nil { fmt.Println(err) return } type wsResponse struct { Msg io.Reader Err error MsgType websocket.MessageType } // THIS LOOP WORKS // for { // _, msg, err := c.Reader(ctx) // buf := new(bytes.Buffer) // buf.ReadFrom(msg) // fmt.Println(buf.String()) // if err != nil { // fmt.Println(err) // return // } // } // The following goroutine and loop produces errors wsChan := make(chan wsResponse) go func() { for { msgType, msg, err := c.Reader(ctx) res := wsResponse{Msg: msg, Err: err, MsgType: msgType} //fmt.Printf("%+v\n", res) wsChan <- res } }() ticker := time.NewTicker(30 * time.Second) for { select { case res := <-wsChan: ticker.Stop() if res.Err != nil { fmt.Println(res.Err) break } buf := new(bytes.Buffer) buf.ReadFrom(res.Msg) s := buf.String() fmt.Println(s) ticker = time.NewTicker(5 * time.Second) case <-ticker.C: fmt.Println("timeout error") break } }}
1 回答

瀟瀟雨雨
TA貢獻1833條經驗 獲得超4個贊
正如該錯誤所暗示的那樣,必須先完全讀取一條消息,然后才能讀取下一條消息。使用代碼的第一個版本或更改第二個版本以將消息吸收為 [] 字節并將該 [] 字節發送到通道。
假設您正在使用 nhooyr.io/websocket 包,第二個版本將如下所示:
for {
// Read returns the entire message as a []byte
msgType, msg, err := c.Read(ctx)
// bytes.NewReader creates an io.Reader on a []byte
res := wsResponse{Msg: bytes.NewReader(msg), Err: err, MsgType: msgType}
wsChan <- res
if res.Err {
// Always exit the loop on error. Otherwise, the goroutine will run forever.
return
}
}
- 1 回答
- 0 關注
- 197 瀏覽
添加回答
舉報
0/150
提交
取消