本文详解如何基于 gorilla websocket 构建支持全局广播的服务器，通过连接池（hub）管理客户端，并实现“一发即达全体”的实时通信能力，附完整可运行示例与关键注意事项。
本文详解如何基于 gorilla websocket 构建支持全局广播的服务器，通过连接池（hub）管理客户端，并实现“一发即达全体”的实时通信能力，附完整可运行示例与关键注意事项。
在 Go Websocket 开发中，单点回显（如 conn.WriteMessage() 仅回复发送者）是默认行为；若需实现类似聊天室的“全员通知”——即任一客户端发消息后，所有已连接客户端（包括发送方自身）均实时收到该消息——必须引入中心化连接管理机制：连接池（Connection Pool）+ 广播通道（Broadcast Channel）。
核心思路是：不再让每个 handler 独立处理连接，而是将所有活跃连接注册到一个共享的 hub 结构体中，由 hub 统一接收广播指令并分发至各客户端的发送通道。
以下是一个精简、可直接运行的生产级广播服务示例（基于 gorilla/websocket v1.5+）：
package main
import (
"log"
"net/http"
"sync"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool { return true }, // 生产环境请严格校验 Origin
}
// connection 表示一个 WebSocket 客户端连接
type connection struct {
ws *websocket.Conn
send chan []byte // 缓冲发送通道，解耦读写
h *hub
}
func (c *connection) writer() {
defer c.ws.Close()
for message := range c.send {
if err := c.ws.WriteMessage(websocket.TextMessage, message); err != nil {
break
}
}
}
// hub 管理所有连接和广播逻辑
type hub struct {
connections map[*connection]bool
broadcast chan []byte
register chan *connection
unregister chan *connection
mu sync.RWMutex
}
func newHub() *hub {
return &hub{
connections: make(map[*connection]bool),
broadcast: make(chan []byte, 128),
register: make(chan *connection, 128),
unregister: make(chan *connection, 128),
}
}
func (h *hub) run() {
for {
select {
case c := <-h.register:
h.mu.Lock()
h.connections[c] = true
h.mu.Unlock()
case c := <-h.unregister:
h.mu.Lock()
if _, ok := h.connections[c]; ok {
delete(h.connections, c)
close(c.send)
}
h.mu.Unlock()
case message := <-h.broadcast:
h.mu.RLock()
for c := range h.connections {
select {
case c.send <- message:
default: // 发送失败（如客户端断连、send channel 已满），清理连接
delete(h.connections, c)
close(c.send)
}
}
h.mu.RUnlock()
}
}
}
var h = newHub()
func serveWs(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println("Upgrade error:", err)
return
}
c := &connection{ws: conn, send: make(chan []byte, 256), h: h}
h.register <- c
// 启动写协程（异步推送）
go c.writer()
// 主读循环：接收消息并广播
for {
_, message, err := conn.ReadMessage()
if err != nil {
if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
log.Printf("Read error: %v", err)
}
break
}
log.Printf("Received: %s", message)
h.broadcast <- message // 关键：推入广播通道，由 hub 统一分发
}
// 连接关闭时注销
h.unregister <- c
close(c.send)
}
func main() {
http.HandleFunc("/ws", serveWs)
log.Println("Server started on :8080")
log.Fatal(http.ListenAndServe(":8080", nil))
}✅ 关键设计说明：
- hub 是线程安全的中央调度器，broadcast 通道接收待广播的原始字节流；
- 每个 connection 持有独立 send 通道，writer() 协程负责从该通道取数据并调用 ws.WriteMessage；
- select + default 的写法确保：若某客户端 send 通道已满或阻塞（如网络中断），立即清理该连接，避免内存泄漏；
- 使用 sync.RWMutex 保护连接映射读写，RWMutex 在高并发读（广播）场景下性能优于普通 Mutex。
⚠️ 注意事项：
- 永远不要在 handler 中直接调用 conn.WriteMessage() 广播 —— 这会导致竞态与连接泄漏；
- send 通道需设合理缓冲（如 256），过小易触发 default 分支误删连接，过大则增加内存压力；
- 生产环境务必替换 CheckOrigin 实现，防止跨站 WebSocket 劫持；
- 建议为 connection 增加心跳检测（SetPingHandler/SetPongHandler）与超时控制（SetReadDeadline），提升健壮性。
通过此架构，你获得的不再是一个“回声服务器”，而是一个可横向扩展的实时广播中枢——无论是聊天、协同编辑、实时通知还是 IoT 设备状态同步，均可在此基础上快速构建。
