Golang WebSocket连接:为何一个标签页刷新后其他标签页无法收发信息?

Golang WebSocket连接:为何一个标签页刷新后其他标签页无法收发信息?

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 对象,解决了标签页刷新导致其他标签页连接中断的问题。 这确保了每个客户端连接拥有独立的资源,互不干扰。

© 版权声明
THE END
喜欢就支持一下吧
点赞13 分享