如何使用Golang实现简单Websocket聊天_处理客户端连接和消息

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 监听注册、注销、广播三类事件
• 新连接进来时，往 register channel 发送该连接指针；断开时走 unregister；消息则塞进 broadcast

消息收发与广播逻辑

每个客户端应有专属 goroutine 处理读写，避免阻塞：

• 读协程：循环调用 conn.ReadMessage()，收到后直接发给 hub.broadcast；出错时触发注销流程
• 写协程（可选但推荐）：为每个 client 开一个 send channel 和对应 goroutine，从 channel 取消息再调用 conn.WriteMessage()，这样广播时不会因某个慢连接卡住全体
• 广播实现：hub 的主循环从 broadcast 拿到消息后，遍历 clients，对每个连接发一次；若 WriteMessage 报错，就删掉该连接并关闭其 send channel

基础消息格式建议

纯文本够用时，直接用 string 或 []byte；想带用户昵称或类型标识，可用 JSON：

{ "type": "message", "from": "Alice", "content": "Hello!" }

服务端用 json.Unmarshal 解析，前端按字段渲染。注意限制单条消息大小（如 maxMessageSize = 512），防止恶意长包拖垮服务。