Golang WebSocket聊天服务核心是安全升级连接、并发管理客户端、统一广播消息;使用gorilla/websocket+goroutine+channel实现,含Hub结构体、注册/注销/广播channel、读写超时与跨域控制。

用 Golang 实现一个能处理多客户端连接和消息广播的简单 WebSocket 聊天服务,核心在于三件事:安全升级 HTTP 连接、并发管理客户端、统一转发消息。不需要复杂框架,靠 gorilla/websocket + goroutine + channel 就能跑起来。
升级连接并建立客户端会话
每个 WebSocket 连接始于一个 HTTP 请求,需用 Upgrader 升级。关键点是:
-
跨域控制:开发阶段设
CheckOrigin: func(r *http.Request) bool { return true };上线前务必改成只允许指定域名 -
连接生命周期:
defer conn.Close()放在 handler 开头,确保异常退出时也能释放资源 -
读写超时设置:调用
conn.SetReadDeadline()和conn.SetWriteDeadline()防止僵死连接占用内存
用 Hub 管理所有在线用户
不能靠全局 map 直接读写,必须加并发保护。推荐结构:
- 定义
Hub结构体,含clients map[*websocket.Conn]bool、broadcast chan []byte、register和unregister两个 channel - 启动一个独立 goroutine 运行
hub.run(),用select监听注册、注销、广播三类事件 - 新连接进来时,往
registerchannel 发送该连接指针;断开时走unregister;消息则塞进broadcast
消息收发与广播逻辑
每个客户端应有专属 goroutine 处理读写,避免阻塞:
立即学习“go语言免费学习笔记(深入)”;
-
读协程:循环调用
conn.ReadMessage(),收到后直接发给hub.broadcast;出错时触发注销流程 -
写协程(可选但推荐):为每个 client 开一个
sendchannel 和对应 goroutine,从 channel 取消息再调用conn.WriteMessage(),这样广播时不会因某个慢连接卡住全体 -
广播实现:hub 的主循环从
broadcast拿到消息后,遍历clients,对每个连接发一次;若WriteMessage报错,就删掉该连接并关闭其 send channel
基础消息格式建议
纯文本够用时,直接用 string 或 []byte;想带用户昵称或类型标识,可用 JSON:
{ "type": "message", "from": "Alice", "content": "Hello!" }
服务端用 json.Unmarshal 解析,前端按字段渲染。注意限制单条消息大小(如 maxMessageSize = 512),防止恶意长包拖垮服务。









