使用Golang开发WebSocket聊天室需先建立持久连接，通过net/http和gorilla/websocket库升级HTTP连接；2. 管理客户端状态采用全局clients映射存储活跃连接，结合broadcast通道传递消息；3. 实现广播机制，在handleMessages循环中读取broadcast通道内容并发送给所有客户端。
用Golang开发WebSocket聊天室，核心是建立持久连接、管理客户端状态、实现消息广播。Go语言的高效并发模型特别适合这种实时通信场景。下面一步步说明如何构建一个基础但完整的WebSocket聊天室。
搭建WebSocket服务器
使用标准库net/http结合第三方库gorilla/websocket可以快速实现WebSocket服务。先安装依赖：
go get github.com/gorilla/websocket
创建HTTP处理器，将普通请求升级为WebSocket连接：
立即学习“go语言免费学习笔记（深入）”；
var upgrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool { return true }, // 允许跨域
}
func handleConnections(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Fatal(err)
return
}
defer conn.Close()
// 连接成功后，等待接收消息
for {
var msg string
err := conn.ReadJSON(&msg)
if err != nil {
log.Printf("读取消息错误: %v", err)
break
}
// 后续将消息推送给所有用户
}
}
管理客户端与广播消息
聊天室的关键是维护所有活跃连接，并在收到新消息时通知所有人。可以用全局变量保存客户端集合：
type Client struct {
conn *websocket.Conn
}
var clients = make(map[*Client]bool)
var broadcast = make(chan string)
func handleMessages() {
for {
msg := <-broadcast
for client := range clients {
err := client.conn.WriteJSON(msg)
if err != nil {
client.conn.Close()
delete(clients, client)
}
}
}
}
在main函数中启动消息广播协程：
go handleMessages()
当新连接建立时，将其加入clients，并将收到的消息发送到broadcast通道。
前端页面接入WebSocket
写一个简单的HTML页面，通过JavaScript建立WebSocket连接：
const ws = new WebSocket("ws://localhost:8080/ws");
ws.onmessage = function(event) {
const chatBox = document.getElementById("chat");
const msg = document.createElement("div");
msg.textContent = event.data;
chatBox.appendChild(msg);
};
function send() {
const input = document.getElementById("input");
ws.send(input.value);
input.value = "";
}
配合一个基础表单，用户就能输入并发送消息。
启动服务与测试
在main函数中注册路由并启动服务器：
func main() {
http.HandleFunc("/ws", handleConnections)
http.Handle("/", http.FileServer(http.Dir("./static")))
log.Println("服务器运行在 :8080")
log.Fatal(http.ListenAndServe(":8080", nil))
}
把HTML文件放在static目录下，运行程序后访问http://localhost:8080即可打开聊天页面。多个浏览器窗口间能实时收发消息。
基本上就这些。不复杂但容易忽略的是错误处理和连接清理——断开的客户端要及时从clients中移除，避免写入失效连接导致panic。可以加锁保护map操作，或改用sync.Map提升并发安全。功能扩展如私聊、房间、用户名等，可在结构体中添加字段并设计消息格式支持。
