本文介绍如何在服务端确保每个用户（以 session id 为标识）仅存在一个活跃的 websocket 连接，新连接建立时自动关闭旧连接，并提供基于 go + gorilla websocket 的线程安全、低开销实现。
本文介绍如何在服务端确保每个用户（以 session id 为标识）仅存在一个活跃的 websocket 连接，新连接建立时自动关闭旧连接，并提供基于 go + gorilla websocket 的线程安全、低开销实现。
在构建实时 Web 应用（如聊天系统、协作编辑或状态同步服务）时，常需限制单个用户只能拥有一条活跃的 WebSocket 连接——例如用户在新标签页打开页面时，应自动踢掉前一个连接，避免重复接收消息、资源浪费或状态冲突。虽然看似简单，但其实现需兼顾正确性（避免竞态）、性能（不成为瓶颈）和健壮性（连接异常断开时能及时清理）。
核心思路是：以用户会话标识（如 sessionID）为键，在服务端维护一个全局映射表，记录当前活跃连接。每当新连接握手完成，即用该 sessionID 尝试注册；若已存在旧连接，则主动调用 .Close() 终止它。注意：WebSocket 的 Close() 方法是线程安全的，可并发调用，且会立即中断读写循环，无需等待超时。
以下为生产就绪的 Go 实现（依赖 Gorilla WebSocket）：
import (
"sync"
"github.com/gorilla/websocket"
)
var (
mu sync.Mutex
conns = make(map[string]*websocket.Conn)
)
// addConn 注册新连接，同时关闭同 session 的旧连接
func addConn(sessionID string, conn *websocket.Conn) {
mu.Lock()
prev := conns[sessionID]
conns[sessionID] = conn
mu.Unlock()
if prev != nil {
// 主动关闭旧连接（非阻塞，安全）
_ = prev.Close() // 忽略错误：连接可能已断开，不影响逻辑
}
}
// removeConn 在连接关闭前安全地从映射中移除自身
func removeConn(sessionID string, conn *websocket.Conn) {
mu.Lock()
// 仅当当前连接仍为 map 中所存值时才删除，防止误删新连接
if conns[sessionID] == conn {
delete(conns, sessionID)
}
mu.Unlock()
}✅ 关键设计说明：
- 轻量级锁保护：sync.Mutex 开销极小，远低于 WebSocket 握手、TLS 加解密、网络 I/O 等操作，无需过度担忧性能；
- 防竞态清理：removeConn 中的指针比对（conns[sessionID] == conn）避免了“旧连接刚被关闭、新连接尚未注册完成”时的误删；
- 无内存泄漏风险：每次连接生命周期内仅 addConn 和 removeConn 各触发一次，map 大小与在线用户数严格正相关，可控；
- Session ID 来源建议：应由服务端生成并持久化（如 Cookie + HTTP-only），避免前端伪造；推荐使用 http.Session 或 JWT payload 中的唯一用户标识。
⚠️ 注意事项：
- 若部署多实例（集群），需将 conns 映射替换为分布式缓存（如 Redis + pub/sub 或 RedLock），并监听连接关闭事件做跨节点同步；
- 客户端应实现重连退避与连接状态检测（如心跳响应超时），避免因服务端强制关闭导致“假离线”；
- 生产环境务必添加连接数监控（如 Prometheus 指标 websocket_active_connections{session_id}），便于及时发现异常长连接。
综上，单机场景下，带互斥锁的内存 map 是最简洁、高效且可靠的方案——它不追求零开销，而是将复杂度控制在远低于网络层的层级，让架构保持清晰与可维护。
