Go语言实现简易聊天室教程

本文深入讲解了如何用 Go 语言构建一个安全、可运行的 Web 聊天室，强调必须基于 net/http 搭配 WebSocket 协议（而非裸 TCP）实现前后端实时通信，并详解 gorilla/websocket 的正确集成方式——包括路径注册、Origin 校验绕过、响应头避免冲突等关键细节；同时直击并发痛点，指出直接使用 map 存储客户端连接的致命风险，提出以 channel 驱动的 Hub 结构体实现线程安全的连接管理，让初学者避开常见坑，快速落地一个健壮、生产就绪的实时聊天服务。

为什么用 net/http 而不是 net 直接写 TCP 服务

因为浏览器无法直连裸 TCP，而聊天室需要前端实时交互。用 http 提供静态页面 + WebSocket 协议升级是最小可行路径。Go 标准库不带 WebSocket 支持，必须引入第三方库（如 gorilla/websocket），它能复用 http.ServeMux，避免额外端口和 TLS 配置复杂度。

常见错误：直接用 net.Listen("tcp", ":8080") 启动监听，结果前端 new WebSocket("ws://...") 报 net::ERR_CONNECTION_REFUSED——根本原因是没走 HTTP 握手，无法完成 WebSocket 协议升级。

• 必须用 http.HandleFunc 注册一个路径（如 "/ws"），并在 handler 中调用 upgrader.Upgrade()
• gorilla/websocket 的 Upgrader.CheckOrigin 默认拒绝非同源请求，开发时需显式设为 func(r *http.Request) bool { return true }，否则浏览器报 403
• 不要在 upgrade 前写任何响应（如 w.WriteHeader()），否则会触发 http: multiple response.WriteHeader calls

map[*websocket.Conn]bool 作为客户端集合的安全隐患

多个 goroutine 并发读写同一个 map 会 panic：fatal error: concurrent map read and map write。聊天室中，readPump（读消息）、writePump（广播）、client.unregister（断开）都可能同时操作该 map。

解决方式不是加锁包一层 sync.Mutex，而是改用 sync.Map 或更推荐的——用 channel 控制注册/注销，让所有变更都通过一个专属 goroutine 串行处理。例如：

type Hub struct {
	clients    map[*websocket.Conn]bool
	broadcast  chan []byte
	register   chan *websocket.Conn
	unregister chan *websocket.Conn
}

func (h *Hub) run() {
	for {
		select {
		case client := 
<p>注意：<code>sync.Map</code> 适合读多写少，但聊天室的注册/注销频率不高，用 channel 更清晰、无锁且天然支持优雅退出。</p>

<h3><code>readPump</code> 和 <code>writePump</code> 必须成对启动，且不能共用连接</h3>
<p>每个 WebSocket 连接需启动两个独立 goroutine：<code>readPump</code> 负责从连接读消息并转发给 hub，<code>writePump</code> 负责从 hub 接收广播消息并写回连接。若只启一个，会出现单向通信或连接卡死。</p>
<p>关键细节：</p>

• readPump 中调用 conn.ReadMessage() 是阻塞的，需配合 conn.SetReadDeadline() 防僵死；超时后应主动关闭连接，否则 writePump 会持续尝试写入已失效的 conn，触发 use of closed network connection
• writePump 必须用 for { select { case message := 模式消费 channel，不能每次广播都新开 goroutine，否则 goroutine 泄漏
• 不能在 readPump 里直接调用 conn.WriteMessage() 回复用户——WebSocket 不允许并发写，会 panic：concurrent write to websocket connection

前端 WebSocket 连接地址写错导致反复重连

浏览器控制台看到 WebSocket connection to 'ws://localhost:8080/ws' failed:，大概率是协议或路径不匹配。Go 后端用 http.ListenAndServe(":8080", nil) 时，WebSocket 地址必须严格对应：

• 本地开发：ws://localhost:8080/ws（注意是 ws，不是 http；路径必须与 http.HandleFunc("/ws", ...) 一致）
• 部署到 Nginx 后：wss://yourdomain.com/ws，此时 Nginx 必须配置 proxy_set_header Upgrade $http_upgrade; 和 proxy_set_header Connection "upgrade";，否则握手失败，降级为 HTTP 200
• 若后端监听的是 127.0.0.1:8080 而非 :8080，外部机器访问会失败，因为绑定到了回环地址

真正容易被忽略的是：当使用 http.Server{Addr: ":8080"}.Serve() 时，日志不会提示监听成功，但实际已启动；如果误以为没起来而反复刷新页面，前端会因连接拒绝快速重试，触发浏览器限流机制，表现为“突然连不上”，其实是后端根本没跑。

好了，本文到此结束，带大家了解了《Go语言实现简易聊天室教程》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！