如何在Golang中实现WebSocket心跳机制_保持连接活跃

Go中WebSocket心跳需服务端启用PingHandler并每25秒发ping、客户端每30秒发JSON心跳包,双方均需超时检测(服务端45秒、客户端60秒)并主动断连,同时注意Nginx超时配置与WriteMessage并发安全。

在 Go 中实现 WebSocket 心跳机制,核心是客户端与服务端协同发送 ping/pong 帧或自定义心跳消息,防止连接因中间代理(如 Nginx、负载均衡器)超时断开。标准 WebSocket 协议本身支持 Ping/Pong 控制帧,但很多 Go 的 WebSocket 库(如 gorilla/websocket)默认不自动响应 pong,需手动处理;同时,应用层心跳(如 JSON 消息)更可控、可监控,推荐两者结合使用。

服务端:启用 Ping 处理 + 定期发送 Ping

使用 gorilla/websocket 时,需显式设置 CheckOrigin、启用 SetPingHandler 并启动协程定期写入 ping:

  • 调用 conn.SetPingHandler 注册 pong 响应逻辑(收到 ping 自动回 pong)
  • 启动一个独立 goroutine,用 time.Ticker 每 25 秒调用 conn.WriteMessage(websocket.PingMessage, nil)
  • 设置读写超时(如 conn.SetReadDeadline),并在读循环中捕获 websocket.CloseMessage 或超时错误及时关闭连接

客户端:响应 Ping + 发送应用心跳

浏览器原生 WebSocket 会自动响应服务端 ping,无需额外处理;但为兼容性及可观测性,建议在 JS 客户端也定时发送 JSON 格式的心跳包(如 {"type":"heartbeat"}):

  • 使用 setInterval 每 30 秒 ws.send(JSON.stringify({type:'heartbeat'}))
  • 监听 message 事件,忽略服务端返回的纯 pong,只处理业务消息和自定义心跳响应
  • 连接异常时(onclose / onerror)触发重连逻辑,避免假死

双向超时控制:防连接滞留

仅发心跳不够,必须配合超时检测才能真正识别“假在线”:

  • 服务端为每个连接维护最后收到消息的时间戳(含 ping/pong 和业务消息),在读循环中更新
  • 另起 goroutine 定期扫描连接列表,若某连接超过 45 秒无任何活动,则主动 conn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseGoingAway, "")) 关闭
  • 客户端同样记录最后收到消息时间,若 60 秒内无响应(包括 pong 和心跳 ACK),视为断连并重连

注意事项与常见坑

实际部署中容易忽略以下细节:

  • Nginx 默认 proxy_read_timeout 是 60 秒,需设为大于服务端心跳间隔(如 75s),并配置 proxy_set_header Connection '' 避免连接被复用干扰
  • gorilla/websocket 的 WriteMessage 不是并发安全的,多个 goroutine 写需加锁或用 conn.WriteJSON + 单写协程通道模型
  • 不要依赖 SetPongHandler 来做业务逻辑——它只用于维持链路,业务心跳应走文本消息通道,便于日志追踪和协议扩展