我最近開始探索 Go 并且非常喜歡它。我在嘗試檢測 websocket 連接超時時遇到了問題。我正在無限期地監聽 websocket 連接,當我在 X 秒內沒有得到響應時,我嘗試重新連接。為此,我不得不修改我的 for 循環以包含一個選擇。然后我創建了一個類型和一個通道來監聽 websocket 響應。然而,這導致我的 websocket 連接出現錯誤,提示無法獲取閱讀器:之前的消息未讀完。我將項目中的代碼替換為獨立的。下面是兩個循環的完整腳本(工作和非工作可用)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 } } }}正在打印日志:{"e":"trade","E":1577140149102,"s":"BTCUSDT","t":220054947,"p":"7304.40000000","q":"0.07153400","b":933798088 "a":933798124,"T":1577140149099,"m":true,"M":true}未能獲得閱讀器:上一條消息未讀完{"e":"trade","E":1577140149107,"s":"BTCUSDT","t":220054948,"p":"7304.95000000","q":"0.28826900","b":933798126 "a":933798125,"T":1577140149104,"m":false,"M":true}未能獲得閱讀器:上一條消息未讀完所以它的工作,但它仍然返回錯誤。閱讀器功能源在這里。https://github.com/nhooyr/websocket/blob/master/conn.go#L390。假設我可以在那里提出一個問題。
1 回答

holdtom
TA貢獻1805條經驗 獲得超10個贊
正如錯誤所暗示的,必須先完整閱讀一條消息,然后才能閱讀下一條消息。使用代碼的第一個版本或更改第二個版本以將消息發送到 []byte 并將該 []byte 發送到通道。
假設您使用的是 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 關注
- 204 瀏覽
添加回答
舉報
0/150
提交
取消