Go并发WebSocket客户端需用goroutine隔离读写、channel统一收发、sync.RWMutex安全维护连接池,封装ConnWrapper结构体管理状态,并通过context控制生命周期与优雅关闭。

在 Go 中实现并发 WebSocket 客户端并管理多个连接,核心在于:用 goroutine 隔离每个连接的读写逻辑,用 channel 统一收发消息,用 map + sync.RWMutex 安全维护连接状态,再配合 context 控制生命周期。
每个连接独立运行在 goroutine 中
不要让多个连接共用一个 goroutine。为每个 WebSocket 连接启动专属 goroutine,分别处理读、写和心跳。例如:
- 启动一个 goroutine 调用
conn.ReadMessage()持续读取,将收到的消息发到该连接对应的接收 channel(如recvCh chan Message) - 另启一个 goroutine 监听该连接的发送 channel(如
sendCh chan Message),取出消息调用conn.WriteMessage() - 再启一个 goroutine 定期发 ping(用
conn.SetPingHandler和conn.WriteControl配合time.Ticker)
用结构体封装连接状态,避免裸指针操作
定义一个 WSClient 或 ConnWrapper 结构体,内嵌 *websocket.Conn,并带上 ID、状态、读写 channel、关闭信号等字段:
type ConnWrapper struct {
ID string
Conn *websocket.Conn
SendCh chan Message
RecvCh chan Message
CloseCh chan struct{}
closed uint32 // 用 atomic 判断是否已关闭
}
这样便于统一管理、调试和扩展(比如加重连计数、最后活跃时间等)。
立即学习“go语言免费学习笔记(深入)”;
全局连接池用 sync.Map 或带锁 map 管理
若需按 ID 查找或广播,用 sync.Map 存储活跃连接(key=ID, value=*ConnWrapper)。注意:sync.Map 适合读多写少;若频繁增删,可用 map[string]*ConnWrapper 配 sync.RWMutex:
- 添加连接时写锁:
mu.Lock(); clients[id] = wrapper; mu.Unlock() - 遍历广播时读锁:
mu.RLock(); for _, c := range clients { ... }; mu.RUnlock() - 关闭连接后务必从 map 中删除,避免内存泄漏和误发
优雅关闭与错误恢复不可少
每个连接 goroutine 都要监听 CloseCh 或 context.Done(),收到信号后主动调用 conn.Close() 并关闭本地 channel。读/写循环中遇到网络错误(如 websocket.IsUnexpectedCloseError、io.EOF)应退出 goroutine,并触发清理逻辑(如从连接池移除、尝试重连)。重连建议加退避策略(如 1s → 2s → 4s),避免雪崩。










