答案是利用goroutine和channel实现非阻塞通信,通过map和互斥锁管理连接,使用广播channel转发消息,读写分离确保并发安全,配合defer及时释放资源,避免内存泄漏。

用Go做并发WebSocket消息处理,核心是利用goroutine和channel实现非阻塞通信。关键不在技术选型,而在于如何组织连接、广播和错误处理的逻辑。
连接管理与客户端注册
每个WebSocket连接上来,都要作为一个独立的客户端实例纳入统一管理。通常用一个全局map存储活跃连接,并配合互斥锁防止并发写冲突。
- 定义clients map[*websocket.Conn]bool来跟踪在线用户
- 使用sync.Mutex保护map的读写操作,避免多个goroutine同时修改引发panic
- 在握手阶段通过upgrader.CheckOrigin允许跨域请求,方便前端调试
消息广播机制设计
服务端收到任意客户端消息后,需要推送给所有在线用户。这个过程不能阻塞主接收循环,所以引入一个广播channel作为中转。
- 声明broadcast chan string作为消息中继通道
- 启动单独goroutine监听该channel,一旦有消息就遍历clients发送
- 发送时加锁,确保同一时间只有一个goroutine在写连接
- 遇到写失败(比如客户端已断开),清理对应连接并关闭资源
并发安全与异常处理
大量客户端同时收发消息时,容易出现资源竞争。除了锁之外,每个连接应有自己的读写分离goroutine。
立即学习“go语言免费学习笔记(深入)”;
- 每个连接开启两个协程:一个负责ReadJSON,一个负责WriteJSON
- 读取到消息后发送到broadcast channel,由广播协程统一处理
- 任何一端出错(如网络中断),立即释放连接、从clients移除并关闭相关channel
- defer语句中执行清理动作,保证异常情况下也能回收资源
基本上就这些,不复杂但容易忽略细节。比如忘记加锁、没处理channel阻塞、或未及时清理失效连接,都会导致内存泄漏或程序崩溃。










