Go语言 WebSocket 多连接频繁断开:如何避免连接丢失?
go语言 websocket 在一个端口上建立多个连接导致频繁断开
您提供的 go 语言代码中,websocket 发送消息的函数如下:
// 通过 websocket 发送字符串 func (d *device) websocketsendstring(str string) bool { if !d.online { return false } log.println(d.ws.request().remoteaddr) if err := websocket.message.send(d.ws, str); err != nil { log.println("设备", d.ws.request().remoteaddr, "[",d.ws.request().form.get("id"),"] websocket 发送消息失败:%s", err) _ = d.ws.close() return false } return true }
此函数中存在一个潜在问题:您在发送消息之前没有检查 websocket 连接状态。当多个设备尝试同时连接到同一个端口时,可能会出现连接丢失的情况。
解决方法是在发送消息之前添加一个检查,以确保 websocket 仍处于活动状态。以下是修改后的代码:
// 通过 WebSocket 发送字符串 func (d *Device) WebsocketSendString(str string) bool { if !d.online || d.ws.WriteMessage == nil { return false } log.Println(d.ws.Request().RemoteAddr) if err := websocket.Message.Send(d.ws, str); err != nil { log.Println("设备", d.ws.Request().RemoteAddr, "[",d.ws.Request().Form.Get("id"),"] websocket 发送消息失败:%s", err) _ = d.ws.Close() return false } return true }