Golang实现简单聊天室_Go语言并发通信实战项目

应使用 net/http 而非 net 直接写 TCP 服务,因浏览器无法直连裸 TCP;需通过 HTTP 提供静态页并用 WebSocket 协议升级实现前端实时交互,借助 gorilla/websocket 复用 http.ServeMux,避免额外端口与 TLS 配置复杂度。

为什么用 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/websocketUpgrader.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 := <-h.register:
			h.clients[client] = true
		case client := <-h.unregister:
			if _, ok := h.clients[client]; ok {
				delete(h.clients, client)
				client.Close()
			}
		case message := <-h.broadcast:
			for client := range h.clients {
				client.WriteMessage(websocket.TextMessage, message)
			}
		}
	}
}

注意:sync.Map 适合读多写少,但聊天室的注册/注销频率不高,用 channel 更清晰、无锁且天然支持优雅退出。

readPumpwritePump 必须成对启动,且不能共用连接

每个 WebSocket 连接需启动两个独立 goroutine:readPump 负责从连接读消息并转发给 hub,writePump 负责从 hub 接收广播消息并写回连接。若只启一个,会出现单向通信或连接卡死。

关键细节:

  • 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() 时,日志不会提示监听成功,但实际已启动;如果误以为没起来而反复刷新页面,前端会因连接拒绝快速重试,触发浏览器限流机制,表现为“突然连不上”,其实是后端根本没跑。