Go 语言中 WebSocket 多连接频繁断开，如何解决？

在 go 语言中使用 websocket 建立多个连接时可能会出现频繁断开的问题。这可能是由于未能正确处理并发连接导致的。以下是如何解决此问题的建议：

你的代码中，使用 go func() 函数以并发的形式调用发送心跳消息的匿名函数。这意味着该函数会在一个独立 goroutine 中运行，同时 websocket 连接也在另一个 goroutine 中运行。这两个 goroutine 可能以不同的速度执行，导致发送心跳消息不及时，从而导致连接断开。

为了解决这个问题，可以采用以下方法：

BibiGPT-哔哔终结者

B站视频总结器-一键总结 音视频内容

下载

• 使用一个带有锁（sync.mutex 或 sync.rwmutex）的单一 goroutine 来处理所有 websocket 连接。
• 使用通道来在 goroutine 之间进行通信。

这里是一个示例代码，演示了使用通道来处理 websocket 连接：

type Device struct {
    online bool
    ws     *websocket.Conn
    msgCh  chan string
}

func DeviceRequestProc(ws *websocket.Conn) {
    defer ws.Close()
    defer ws.Request().Body.Close()

    _ = ws.Request().ParseForm()
    mac := ws.Request().Form.Get("id")

    log.Println("设备", ws.Request().RemoteAddr, "[", mac, "] websocket 连接建立")

    dev := devInfo.getDeviceByMac(mac)
    if dev == nil {
        _ = websocket.Message.Send(ws, "未找到对应设备配置")
        log.Println("设备", ws.Request().RemoteAddr, "[", mac, "] websocket 连接未找到对应设备")
        return
    }

    dev.online = true
    dev.ws = ws
    dev.msgCh = make(chan string)

    go dev.WSCheck()
    go dev.ProcessWSMsg()

    for {
        select {
        case msg := <-dev.msgCh:
            if err := websocket.Message.Send(ws, msg); err != nil {
                break
            }
        }
    }

    dev.online = false
    dev.ws = nil
    log.Println("设备", ws.Request().RemoteAddr, "[", ws.Request().Form.Get("id"), "] websocket 连接断开")
}

func (d *Device) WSCheck() {
    for d.online {
        d.msgCh <- fmt.Sprintf(`{"eventType":"sundyn_ws_check","time":"%s"}`, time.Now().Format("2006-01-02 15:04:05"))
        time.Sleep(time.Second * 20)
    }
}

func (d *Device) ProcessWSMsg() {
    for {
        if !d.online {
            return
        }

        // 此处阻塞，等待有数据可读
        var reply string
        if err := websocket.Message.Receive(d.ws, &reply); err != nil {
            break
        }

        d.ProcessWSMsg(reply)
    }
}

通过使用通道来通信，两个 goroutine 可以有序地交换消息，从而确保发送心跳消息及时且可靠。