Golang WebSocket连接问题排查与解决方案
在使用Golang的gorilla/websocket库构建WebSocket应用时，可能会遇到一些挑战。本文将分析一个常见问题：使用Chrome浏览器打开多个标签页连接同一个WebSocket地址（例如localhost或IP地址），只有一个标签页能正常工作，其他标签页刷新后才能恢复连接。
问题根源在于WebSocket连接的管理方式。一些开发者错误地使用全局变量来存储websocket.Conn对象：
var ws *websocket.Conn
每次有新的连接时，都会更新ws的值，覆盖之前的连接。这意味着服务器端只能维持一个活跃连接。当多个标签页连接时，后连接的标签页会断开之前的连接。
立即学习“go语言免费学习笔记（深入）”；
解决方案：为每个连接创建独立的websocket.Conn对象
为了解决这个问题，必须为每个WebSocket连接创建独立的websocket.Conn对象，避免使用全局变量。正确的做法是在处理函数内部，针对每个连接创建新的websocket.Conn对象，并绑定到该连接。 一个参考自GitHub聊天案例的改进代码如下：
func Ws(c *gin.Context) {
ws, err := upgrader.Upgrade(c.Writer, c.Request, nil)
if err != nil {
return
}
defer ws.Close() // 关闭连接非常重要
go func(conn *websocket.Conn) {
// 处理每个连接的读写操作
for {
mt, message, err := conn.ReadMessage()
if err != nil {
log.Println("read:", err) // 添加错误日志
break
}
log.Printf("Received: %s", message) // 添加日志方便调试
err = conn.WriteMessage(mt, message)
if err != nil {
log.Println("write:", err) // 添加错误日志
break
}
}
}(ws)
}
通过goroutine为每个连接创建一个独立的处理函数，并使用函数内部的conn变量，避免了全局变量带来的冲突，确保每个WebSocket连接独立运行，互不干扰。 这才是正确处理多个并发WebSocket连接的最佳实践。 此外，添加了错误日志和接收消息日志，方便调试和排错。
通过以上改进，可以有效解决多个浏览器标签页连接同一个WebSocket地址时，只有一个标签页能正常工作的问题。 记住，妥善管理连接，避免使用全局变量是关键。
