如何在Golang服务中接入Go-websocket处理长链接消息-Golang-PHP中文网

因为go标准库无原生websocket支持，gorilla/websocket提供完整生命周期管理、安全跨域校验、读写goroutine分离、可靠心跳机制及生产级稳定性保障。

如何在golang服务中接入go-websocket处理长链接消息

为什么用 `gorilla/websocket` 而不是标准库

Go 标准库没有原生 WebSocket 实现，net/http 只能做握手响应，后续帧解析、心跳、关闭流程都得自己写。实际项目里几乎没人手撸——容易漏掉 CloseMessage 处理、分片帧、ping/pong 超时等细节，导致连接假死或内存泄漏。gorilla/websocket 是事实标准，稳定、有完整生命周期控制，且支持 SetReadDeadline 和 SetWriteDeadline 防僵死。

`Upgrader` 配置必须设 `CheckOrigin`

默认情况下 Upgrader 会拒绝所有跨域请求，但线上服务常被前端页面直连，不设校验会导致 403 Forbidden。别直接写 func(r *http.Request) bool { return true }——这是安全漏洞。按需放开：

var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool {
        origin := r.Header.Get("Origin")
        return origin == "https://your-app.com" || origin == "http://localhost:3000"
    },
}

生产环境禁止返回 true，哪怕只跑内网也要显式白名单
如果用 Nginx 做反向代理，确保它透传了 Origin 头（默认不透传）
某些 CDN 或网关会改写 Origin，此时要根据 X-Forwarded-Host 或其他可信头判断

读写并发必须用独立 goroutine + channel 控制

WebSocket 连接是双向流，ReadMessage 和 WriteMessage 都是阻塞操作，混在同一个 goroutine 里会互相卡住。典型错误是：收到消息后立刻 WriteMessage，但对方还没读完上一条，写缓冲区满就 hang 住。

Go语言（Golang）1.26.0

Go语言（Golang）1.26.0版本提供 Go 官方 Windows amd64 MSI 安装包下载入口，版本号 1.26.0，可用于旧项目维护、兼容性测试和指定版本开发环境配置。

下载

每个连接起两个 goroutine：一个专读（for { c.ReadMessage(...) }），把消息发到 chan Message
另一个专写（for msg := range sendChan），调用 c.WriteMessage(...)
用 context.WithTimeout 包裹 WriteMessage，避免单条消息卡死整个写通道
关闭连接前，先 close(sendChan)，再 c.Close()，否则写 goroutine 可能 panic

心跳必须由服务端主动发 `Ping` 并监听 `Pong`

浏览器 WebSocket API 不自动回 Pong，得靠 SetPongHandler 注册回调；而客户端不发 Pong 时，服务端得靠 SetPingHandler + 定时 WriteControl(websocket.PingMessage, ...) 来探测。只做一边等于没做。

c.SetPingHandler(func(appData string) error {
    return c.WriteControl(websocket.PongMessage, []byte{}, time.Now().Add(10*time.Second))
})
c.SetPongHandler(func(string) error { c.PongHandler() })