使用Go和Gorilla WebSocket实现广播系统,核心是维护客户端连接集合与消息广播通道;02. 服务端通过upgrade处理WebSocket连接,将新连接加入clients map,并启动handleMessages协程监听broadcast通道;03. 每个连接读取消息后推送到broadcast,由广播协程转发给所有在线client;04. 前端通过WebSocket API连接,发送消息并实时接收显示他人消息,形成简单聊天室。

用Go语言实现WebSocket消息广播,核心是管理客户端连接,并把收到的消息发给所有在线用户。下面是一个简单实用的示例,使用Gorilla WebSocket库,适合做聊天室、通知系统等实时应用。
基本结构设计
服务端需要两个主要部分:一个全局的客户端集合,用来保存所有活跃的连接;一个广播通道,用来中转消息。
- clients:用map存储每个WebSocket连接,方便动态增删
- broadcast:一个channel,接收来自任一客户端的消息,等待分发
后端代码实现
使用gorilla/websocket库处理协议升级和通信。启动一个goroutine专门负责广播,避免阻塞主流程。
package mainimport ( "log" "net/http" "github.com/gorilla/websocket" )
var upgrader = websocket.Upgrader{CheckOrigin: func(r http.Request) bool { return true }} var clients = make(map[websocket.Conn]bool) var broadcast = make(chan string)
func handleConnections(w http.ResponseWriter, r *http.Request) { ws, err := upgrader.Upgrade(w, r, nil) if err != nil { log.Fatal(err) } defer ws.Close() clients[ws] = true
for { var msg string err := ws.ReadJSON(&msg) if err != nil { delete(clients, ws); break } broadcast <- msg }}
立即学习“go语言免费学习笔记(深入)”;
func handleMessages() { for { msg :=
func main() { http.HandleFunc("/ws", handleConnections) go handleMessages() log.Println("服务器启动在 :8080") err := http.ListenAndServe(":8080", nil) if err != nil { log.Fatal("启动失败:", err) } }
前端页面配合
写一个简单的HTML页面,通过浏览器原生WebSocket API连接服务端,输入内容后发送,并能实时看到别人的消息。
基本上就这些,不复杂但容易忽略错误处理和连接清理。










