使用map和互斥锁管理客户端连接,通过Client结构体封装Conn和Send通道,确保并发安全;2. 每个连接启动readPump和writePump协程处理读写,避免阻塞;3. 广播时遍历客户端map,利用select+default防止阻塞,结合Gorilla库适用于聊天室等场景。

在Go语言中实现WebSocket多客户端管理,关键在于维护一个全局的客户端连接池,并通过读写协程处理消息的收发。下面介绍一种常见且高效的实现方式。
1. 使用map和互斥锁管理客户端连接
为了安全地在多个goroutine之间操作客户端集合(如广播消息时),需要使用sync.Mutex保护共享资源。
通常将每个WebSocket连接封装为一个Client结构体,包含连接实例、发送消息通道等字段:
- Conn:*websocket.Conn,实际的WebSocket连接
- Send:chan []byte,用于向该客户端发送数据的通道
用一个map[*Client]bool或map[string]*Client存储所有活跃连接,配合Mutex进行增删操作。
立即学习“go语言免费学习笔记(深入)”;
2. 启动读写协程处理通信
每当有新客户端连接,创建Client实例并注册到全局管理器中。然后启动两个协程:
- writePump:从Send通道读取数据,写入WebSocket连接
- readPump:从WebSocket连接读取消息,转发给业务逻辑或广播
这样可以避免读写阻塞,提升并发性能。当连接关闭时,记得从管理器中注销该Client并关闭其Send通道。
3. 实现广播机制
广播消息是多客户端管理的核心功能。可通过遍历客户端map,将消息写入每个Client的Send通道实现:
func (h *Hub) Broadcast(message []byte) {
for client := range h.Clients {
select {
case client.Send <- message:
default:
close(client.Send)
delete(h.Clients, client)
}
}
}使用select + default防止因某个客户端卡住导致广播阻塞。
基本上就这些。结合Gorilla WebSocket库,这种模式稳定可靠,适用于聊天室、实时通知等场景。不复杂但容易忽略细节,比如异常断开的清理和并发安全。










