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

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

P粉602998670

发布： 2025-12-19 16:37:14

原创

812人浏览过

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

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

用 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() 防止僵死连接占用内存

不能靠全局 map 直接读写，必须加并发保护。推荐结构：

定义 Hub 结构体，含 clients map[*websocket.Conn]bool、broadcast chan []byte、register 和 unregister 两个 channel
启动一个独立 goroutine 运行 hub.run()，用 select 监听注册、注销、广播三类事件
新连接进来时，往 register channel 发送该连接指针；断开时走 unregister；消息则塞进 broadcast

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

会译·对照式翻译

会译是一款AI智能翻译浏览器插件，支持多语种对照式翻译

读协程：循环调用 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），防止恶意长包拖垮服务。

以上就是如何使用Golang实现简单Websocket聊天_处理客户端连接和消息的详细内容，更多请关注php中文网其它相关文章！

大家都在看：

Go html/template：在 HTML 中安全地嵌入 JSON 数据前端资源异步加载与性能优化 Revel框架静态文件加载异常排查与解决 Revel 框架静态文件加载异常排查与解决指南如何在Golang中实现静态资源管理