本文详解如何基于 gorilla websocket 构建支持广播功能的服务器，通过连接池（hub）统一管理客户端连接，并在任一客户端发消息时实时推送给所有在线用户，附完整可运行示例与关键注意事项。
本文详解如何基于 gorilla websocket 构建支持广播功能的服务器，通过连接池（hub）统一管理客户端连接，并在任一客户端发消息时实时推送给所有在线用户，附完整可运行示例与关键注意事项。
在 Go 中实现 WebSocket 广播（即“一发全收”），核心在于解耦消息接收与发送逻辑——不能像原始代码中那样仅对单个连接调用 WriteMessage，而需维护一个全局可访问的活跃连接池（connection pool），并借助 goroutine 和 channel 实现异步、线程安全的消息分发。
以下是一个精简但生产可用的广播架构实现，基于 Gorilla WebSocket 官方推荐模式（参考经典 chat 示例）：
✅ 核心组件设计
- hub：中心协调器，持有所有活跃连接的引用（map[*connection]bool）及广播通道 broadcast chan []byte；
- connection：封装每个 WebSocket 连接及其专属发送通道 send chan []byte，避免阻塞读协程；
- 所有连接注册/注销、消息广播均通过 hub 统一调度，确保并发安全。
✅ 完整可运行示例（Go 1.21+）
package main
import (
"log"
"net/http"
"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
}
func newHub() *hub {
return &hub{
broadcast: make(chan []byte, 128),
register: make(chan *connection),
unregister: make(chan *connection),
connections: make(map[*connection]bool),
}
}
func (h *hub) run() {
for {
select {
case c := <-h.register:
h.connections[c] = true
case c := <-h.unregister:
if _, ok := h.connections[c]; ok {
delete(h.connections, c)
close(c.send)
}
case message := <-h.broadcast:
// 向所有已注册连接广播
for c := range h.connections {
select {
case c.send <- message:
default:
// 发送失败（如客户端断开或 send channel 满），清理连接
delete(h.connections, c)
close(c.send)
}
}
}
}
}
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 {
break
}
log.Printf("Received: %s", message)
// 广播给所有客户端（含发送者自身）
h.broadcast <- message
}
// 断开时注销
h.unregister <- c
}
func main() {
http.HandleFunc("/ws", serveWs)
log.Println("Server starting on :8080")
go h.run() // 启动 hub 主循环
log.Fatal(http.ListenAndServe(":8080", nil))
}⚠️ 关键注意事项
- send 通道必须带缓冲（如 make(chan []byte, 256)）：防止 writer() 协程因接收端未就绪而阻塞，进而拖垮整个 hub 的广播性能；
- select + default 是优雅降级的关键：当某客户端网络异常导致 c.send 阻塞时，default 分支立即触发清理，避免“僵尸连接”累积；
- 广播不区分发送者与接收者：如需排除发送者，可在广播前遍历 h.connections 并跳过当前 c；
- 生产部署务必启用 TLS：WebSocket 的 wss:// 协议是标配，http.ListenAndServeTLS 替代 ListenAndServe；
- 连接数监控与限流：高并发场景下建议为 h.connections 加读写锁（sync.RWMutex）或改用 sync.Map，并集成 Prometheus 指标采集。
通过该模式，你将获得一个健壮、可扩展的 WebSocket 广播服务基础——无论是实时聊天、协同编辑还是行情推送，均可在此架构上快速演进。
