Golang 实现 WebSocket 实时通信需用 gorilla/websocket 库,核心是通过 ClientManager 结构体(含 clients map、broadcast/register/unregister channel 及 sync.RWMutex)统一管理连接并安全广播;HTTP 升级须用 websocket.Upgrader 并校验 origin。

用 Golang 实现 WebSocket 实时通信并支持多客户端数据推送,核心是使用 gorilla/websocket 库——它稳定、轻量、文档清晰,是 Go 生态中最主流的 WebSocket 方案。关键不在“连上”,而在“管住连接”和“安全广播”。
初始化 WebSocket 连接管理器
不能让每个连接裸奔,需统一维护在线客户端。推荐用带互斥锁的 map 存储 *websocket.Conn,避免并发写 panic:
- 定义结构体:type ClientManager struct { clients map[*websocket.Conn]bool; broadcast chan []byte; register chan *websocket.Conn; unregister chan *websocket.Conn; mu sync.RWMutex }
- 启动 goroutine 监听 register/unregister/broadcast 三个 channel,实现连接注册、下线清理、消息分发三件事
- register 和 unregister 操作前加
mu.Lock(),读 clients 时用mu.RLock()
处理 HTTP 升级与连接握手
WebSocket 是基于 HTTP 的升级协议,需明确返回 101 状态码。别手写 header,用 gorilla 提供的 Upgrader:
- 声明全局 var upgrader = websocket.Upgrader{CheckOrigin: func(r *http.Request) bool { return true }}(生产环境务必校验 origin)
- 在 handler 中调用 conn, err := upgrader.Upgrade(w, r, nil),成功即获得 *websocket.Conn
- 立即调用 clientManager.register 将其纳入管理器
实现多客户端广播推送
广播不是遍历 send,而是通过 channel 解耦:业务逻辑只往 broadcast channel 发字节流,由 manager 统一投递给所有活跃连接:
立即学习“go语言免费学习笔记(深入)”;
- manager 的主循环中,收到 broadcast 消息后,对每个已注册的 conn 调用 conn.WriteMessage(websocket.TextMessage, msg)
- 每次 WriteMessage 前检查 conn 是否 closed(可用 conn.Close() == nil 粗略判断,更准的是监听 conn.ReadMessage() 返回的 error)
- 写失败时执行 clientManager.unregister 并 close(conn),防止僵尸连接堆积
主动向指定客户端推送(可选增强)
若需点对点或按标签推送(如“推给用户ID=123的所有设备”),可在 Client 结构中增加字段:
- 扩展 clientManager.clients 为 map[string][]*websocket.Conn,key 为用户 ID 或 group ID
- 注册时解析 token 或 URL query(如
/ws?uid=123),将 conn 加入对应 slice - 推送时查 map 获取目标 slice,逐个 write;注意同样要跳过已断开的连接
不复杂但容易忽略:心跳保活(服务端定期 ping)、错误日志记录、连接数限制、消息大小限制(设置 conn.SetReadLimit(512 * 1024))、TLS 部署(用 https + wss)。跑通后,你拥有的是一个可横向扩展的实时通道底座。










