如何在Golang中处理WebSocket连接_Golang gorilla WebSocket包实践

2026-01-20

如何在Golang中处理WebSocket连接_Golang gorilla WebSocket包实践

gorilla/websocket.Upgrader.Upgrade() 返回 http.ErrUpgradeRequired 是因客户端未发送正确的升级请求头，必须包含 Upgrade: websocket 和 Connection: Upgrade。

如何在golang中处理websocket连接_golang gorilla websocket包实践

为什么 `gorilla/websocket` 的 `Upgrader.Upgrade()` 会返回 `http.ErrUpgradeRequired`

这个错误不是连接失败，而是你没发对 HTTP 请求头。WebSocket 握手本质是 HTTP 协议升级，客户端必须带 Upgrade: websocket 和 Connection: Upgrade，否则 Upgrade() 直接拒绝并返回该错误。

检查前端是否用 new WebSocket("ws://...") 而非普通 fetch 或 XMLHttpRequest

若用 curl 测试，需手动加头：

curl -i -N /
  -H "Connection: Upgrade" /
  -H "Upgrade: websocket" /
  -H "Sec-WebSocket-Version: 13" /
  -H "Sec-WebSocket-Key: $(openssl rand -base64 16)" /
  http://localhost:8080/ws

Nginx 反向代理时，必须显式透传升级头：

location /ws {
    proxy_pass http://backend;
    proxy_http_version 1.1;
    proxy_set_header Upgrade $http_upgrade;
    proxy_set_header Connection "upgrade";
}

如何安全地并发读写 `*websocket.Conn`

gorilla/websocket 的连接对象**不支持并发读或并发写**，但允许一个 goroutine 读、另一个写——这是设计前提。直接多个 goroutine 调用 conn.WriteMessage() 会 panic：「concurrent write to websocket connection」。

写操作必须串行：用 conn.SetWriteDeadline() 防僵死，再通过 channel 把消息统一交给一个「writer goroutine」处理
读操作建议单 goroutine + conn.SetReadDeadline()，收到消息后投递到业务逻辑 channel（如 chan Message）
不要在 ReadMessage() 循环里直接调用耗时函数（比如 DB 查询），容易阻塞整个连接读取

`conn.ReadMessage()` 返回 `websocket.CloseMessage` 时该怎么做

这不是错误，是对方主动关闭连接的信号。此时你不该再调用 WriteMessage()，否则会报 use of closed network connection。

Frase

Frase是一款出色的长篇 AI 写作工具，快速创建seo优化的内容。

下载

检测到 messageType == websocket.CloseMessage 后，应立即跳出读循环，并调用 conn.Close()
如果需要响应关闭帧，先调用 conn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))，再关连接
注意：conn.Close() 内部会发一次 CloseMessage，所以通常只需自己发一次响应帧，避免重复关闭

如何给每个 WebSocket 连接绑定用户身份并管理生命周期

别把用户 ID 存在全局 map 里裸用，容易竞态和泄漏。推荐用 sync.Map + 显式清理：

立即学习“go语言免费学习笔记（深入）”；

连接建立后，从 JWT token 或 query 参数解析 user_id，存入 conn 的自定义字段（例如用 context.WithValue() 包裹 http.HandlerFunc）
用 sync.Map 存 map[string]*websocket.Conn，key 是 user_id；但注意：连接断开时必须显式 Delete()，不能只靠 GC

在 defer conn.Close() 前加清理逻辑：

defer func() {
    if userID != "" {
        clients.Delete(userID)
    }
}()

心跳超时（如 30 秒无 PING）也得触发清理，否则僵尸连接堆积

实际跑起来最常卡住的地方，是忘记设读写 deadline，或者在 handler 里漏掉 defer conn.Close()。这两个点一旦出问题，连接不会立刻断，但内存和 goroutine 会缓慢上涨。

https://www.php.cn/faq/2005764.html