golang websocket连接问题:解决标签页刷新导致连接中断
本文分析并解决一个使用Golang Gorilla/WebSocket库开发的WebSocket应用中出现的棘手问题:chrome浏览器打开多个标签页连接同一服务器,刷新其中一个标签页后,其他标签页的WebSocket连接中断,无法收发信息。
问题描述:
在Go 1.16和Gorilla/WebSocket 1.4.2版本,windows 10环境下,使用Chrome浏览器打开两个标签页访问同一个localhost地址(一个使用localhost,另一个使用IP地址),通过IP地址连接WebSocket服务器。刷新其中一个标签页后,该标签页的WebSocket连接正常,但另一个标签页却无法收发信息,除非也刷新该标签页。
立即学习“go语言免费学习笔记(深入)”;
代码片段(服务器端):
服务器端代码使用了Gorilla/WebSocket库和gin框架。关键代码片段如下,展示了错误的全局变量用法:
package ws import ( "encoding/json" "fmt" "net/http" "time" "github.com/gin-gonic/gin" "github.com/gorilla/websocket" ) var upgrader = websocket.Upgrader{} // 全局变量,错误之处 func ws(c *gin.Context) { ws, err := upgrader.Upgrade(c.Writer, c.Request, nil) // 错误:使用全局变量ws if err != nil { return } defer ws.Close() // 添加连接关闭时的资源释放 for { mt, message, err := ws.ReadMessage() if err != nil { break } err = ws.WriteMessage(mt, message) if err != nil { break } } } // ... (路由配置省略) ...
问题原因及解决方案:
问题根源在于代码使用了全局变量 ws (websocket.Conn)。这导致所有连接都复用同一个WebSocket连接对象,一个连接的操作会影响其他连接。
解决方案: 移除全局 ws 变量,在 ws 函数中,为每个客户端连接创建独立的 websocket.Conn 对象。 在 upgrader.Upgrade 后,直接使用返回的 websocket.Conn 对象进行读写操作,并在连接关闭时释放资源(使用 defer ws.Close())。 正确的代码如下:
package ws import ( // ... (导入包省略) ... ) var upgrader = websocket.Upgrader{} func ws(c *gin.Context) { ws, err := upgrader.Upgrade(c.Writer, c.Request, nil) if err != nil { return } defer ws.Close() for { mt, message, err := ws.ReadMessage() if err != nil { break } err = ws.WriteMessage(mt, message) if err != nil { break } } } // ... (路由配置省略) ...
通过移除全局变量并为每个连接创建独立的 websocket.Conn 对象,解决了标签页刷新导致其他标签页连接中断的问题。 这确保了每个客户端连接拥有独立的资源,互不干扰。