Golang实现WebSocket多客户端实时推送教程

哈喽！今天心血来潮给大家带来了《Golang实现WebSocket多客户端实时推送教程》，想必大家应该对Golang都不陌生吧，那么阅读本文就都不会很困难，以下内容主要涉及到，若是你正在学习Golang，千万别错过这篇文章~希望能帮助到你！

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 统一投递给所有活跃连接：

• 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）。跑通后，你拥有的是一个可横向扩展的实时通道底座。

到这里，我们也就讲完了《Golang实现WebSocket多客户端实时推送教程》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！