GolangWebSocket刷新后多标签页收发解析

本文解决Golang WebSocket应用中一个棘手问题：Chrome浏览器多标签页连接同一服务器，刷新其中一个标签页后，其他标签页连接中断，无法收发信息。问题源于服务器端代码错误地使用了全局变量`websocket.Conn`，导致所有连接复用同一个对象。解决方案是移除全局变量，为每个客户端连接创建独立的`websocket.Conn`对象，确保每个连接拥有独立资源，避免互相干扰。文章详细分析了问题原因，并提供了完整的代码示例及修正方案，有效解决了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连接正常，但另一个标签页却无法收发信息，除非也刷新该标签页。

代码片段（服务器端）：

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

今天关于《GolangWebSocket刷新后多标签页收发解析》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！