当前位置：首页 > 文章列表 > Golang > Go教程 > Golang实现WebSocket多客户端实时推送教程

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

2026-02-01 19:58:36 0浏览收藏

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

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

如何使用Golang实现WebSocket实时通信_多客户端数据推送

用 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)，防止僵尸连接堆积