本文介绍如何在服务端确保每个用户（以 session id 为标识）仅存在一个活跃的 websocket 连接，旧连接会在新连接建立时自动关闭；通过带锁的内存映射表实现，兼顾正确性与性能。
本文介绍如何在服务端确保每个用户（以 session id 为标识）仅存在一个活跃的 websocket 连接，旧连接会在新连接建立时自动关闭；通过带锁的内存映射表实现，兼顾正确性与性能。
在构建实时 Web 应用（如聊天系统、协同编辑或实时通知服务）时，常需限制单个用户只能拥有一条活跃的 WebSocket 连接——例如用户在新标签页打开同一页面时，应主动断开前一页面的连接，避免资源冗余、状态冲突或计费异常。虽然直觉上“全局映射 + 并发控制”可能引发性能担忧，但实际中，该方案的开销远低于 WebSocket 握手、TLS 协商及网络 I/O 等核心操作，是合理且推荐的实践。
以下基于 Go 语言与 Gorilla WebSocket 包给出轻量、线程安全的实现：
import (
"sync"
"github.com/gorilla/websocket"
)
var (
mu sync.Mutex
conns = make(map[string]*websocket.Conn) // key: sessionID
)
// addConn 将新连接注册到 sessionID 下，并安全关闭已存在的旧连接
func addConn(sessionID string, conn *websocket.Conn) {
mu.Lock()
prev := conns[sessionID]
conns[sessionID] = conn
mu.Unlock()
if prev != nil {
// Close() 是并发安全的：它立即终止读写，触发对方 onError/onClose
prev.Close()
}
}
// removeConn 在连接关闭时清理映射（需传入当前连接对象，防止误删）
func removeConn(sessionID string, conn *websocket.Conn) {
mu.Lock()
if conns[sessionID] == conn {
delete(conns, sessionID)
}
mu.Unlock()
}关键设计说明：
- ✅ 会话粒度精准：使用 sessionID（建议从 HTTP Cookie 或 JWT 中提取并校验）作为键，天然支持跨 Tab/Window 的单连接约束；
- ✅ 无竞态清理：removeConn 中比对指针值，避免因连接快速重连导致的误删；
- ✅ 低延迟关闭：prev.Close() 不阻塞主线程，底层会异步发送 close frame 并释放资源；
- ⚠️ 注意扩展性边界：单机方案适用于万级并发连接；若需水平扩展（多实例部署），应改用 Redis 的 SET key value NX EX 原子操作配合 Pub/Sub 实现分布式连接互斥；
- ? 安全提示：务必验证 sessionID 的合法性与签名，防止客户端伪造 ID 绕过限制。
总结而言，内存映射 + 互斥锁并非性能瓶颈，而是清晰、可控、可测试的首选方案。真正需要优化的是连接生命周期管理（如心跳保活、异常断连检测）和会话上下文的一致性维护——这些远比一个 O(1) 的 map 操作复杂得多。
