go中websocket多客户端管理通过Client和ClientManager结构实现:Client含连接和发送通道,ClientManager用map、注册/注销/广播通道及读写协程统一管理连接生命周期与消息分发。

在Go语言中实现WebSocket多客户端管理,核心是利用Goroutine和channel构建一个高效、安全的并发模型。通过一个中心化的管理器来注册、注销连接,并进行消息广播,可以轻松应对聊天室、实时通知等场景。
设计客户端与管理器结构
每个客户端连接需要被封装,以便统一处理读写操作。同时,一个全局的管理器负责维护所有活跃连接。
– Client结构体:包含*websocket.Conn连接实例和一个带缓冲的Send chan []byte,用于向该客户端发送消息。
– ClientManager结构体:这是核心,包含以下字段:
• clients map[*Client]bool:用读写锁保护的集合,存储所有在线客户端。
• register/unregister chan *Client:通道,用于安全地添加或移除客户端,避免直接操作map导致的并发问题。
• broadcast chan []byte:接收要广播给所有客户端的消息。
实现连接的生命周期管理
管理器的Run方法是一个永不退出的for-select循环,它监听三个事件通道。
– 注册 (register):当有新连接时,将Client实例加入clients映射,并启动其读写协程(readPump和writePump)。
– 注销 (unregister):当连接断开时,从clients映射中删除该Client,并关闭其Send通道,释放资源。
– 广播 (broadcast):当收到一条消息时,遍历所有clients,尝试将消息发送到每个Client的Send通道。如果发送失败(通常是因为通道已满或已关闭),则触发unregister流程清理该连接。
处理http升级与读写分离
ServeHTTP函数负责将普通的HTTP请求升级为WebSocket连接。
立即学习“go语言免费学习笔记(深入)”;
– 升级连接:使用websocket.Upgrader.CheckOrigin设置为true(生产环境需严格校验来源)来接受请求,调用Upgrade方法获得*websocket.Conn。
– 读写分离:为每个新连接创建一个Client实例后,立即启动两个独立的Goroutine:
• readPump:在一个for循环中调用conn.ReadMessage(),持续监听客户端发来的消息。收到的消息可以直接转发给manager.broadcast进行全局广播,或根据业务逻辑进行路由。
• writePump:在一个for循环中监听client.Send这个channel。一旦有数据写入此channel,就立刻调用conn.WriteMessage()推送给客户端。当连接关闭时,该循环会因channel被关闭而退出。
基本上就这些。