答案是使用全局map和channel实现WebSocket广播。通过sync.RWMutex保护clients map存储连接,注册与注销客户端;利用broadcast channel接收消息并由单独goroutine遍历map推送,结合write deadline和异常恢复机制确保稳定性。

用Golang实现WebSocket消息广播,核心是管理所有活跃连接,并在收到消息时推送给每个客户端。关键在于设计一个全局的客户端集合,配合goroutine进行消息分发。
建立连接池管理客户端
服务器需要知道哪些客户端在线,才能向它们广播消息。通常使用一个全局的map来存储所有活动的WebSocket连接。
定义一个并发安全的客户端集合:- 声明一个
clients map[*websocket.Conn]bool,记录当前所有连接 - 使用
sync.RWMutex保护map,避免并发读写冲突 - 在用户连接时(
handleConnection)将conn加入map - 断开时从map中删除,确保连接不泄漏
实现广播逻辑
广播的本质是遍历所有连接,逐一发送相同的消息。可以借助一个全局channel集中处理。
典型实现方式:- 创建一个
broadcast chan []byte作为消息中转站 - 启动一个独立goroutine,循环监听该channel
- 每当有新消息进入channel,就遍历
clientsmap,调用每个conn的WriteMessage方法 - 发送时建议设置write deadline,防止慢连接阻塞整个广播过程
处理异常连接
广播过程中可能遇到已断开或不可用的连接,必须妥善处理,避免程序崩溃。
立即学习“go语言免费学习笔记(深入)”;
- 在
WriteMessage外层加recover(),捕获panic - 如果发送失败(如客户端已关闭),立即关闭该conn,从
clients中移除 - 可结合心跳机制,定期检测连接状态,提前清理无效连接
基本上就这些。一个稳定的消息广播系统,重点在连接管理的健壮性和并发安全性。用好channel和mutex,就能支撑起千人级别的实时推送场景。










