17370845950

用Golang实现WebSocket服务需借助gorilla/websocket库，通过upgrader.Upgrade()升级HTTP连接，用goroutine读写消息，维护线程安全的连接池，妥善处理错误与心跳，确保连接生命周期管理健壮。

用 Golang 实现一个简单的 WebSocket 服务并不复杂，核心是借助 gorilla/websocket 这个广泛使用的第三方库——它稳定、文档清晰、社区支持好，比标准库更直接易用。

安装依赖并初始化 WebSocket 连接处理

先通过 go mod 管理依赖：

go mod init example.com/ws-server
go get github.com/gorilla/websocket

然后定义 HTTP handler，把升级 WebSocket 请求的逻辑封装进去。关键点是：不能直接用 http.ResponseWriter 写响应，必须用 upgrader.Upgrade() 把连接“升格”为 WebSocket 连接。

常见写法示例：

• 定义全局 upgrader，设置允许跨域（开发时常用）：
  var upgrader = websocket.Upgrader{CheckOrigin: func(r *http.Request) bool { return true }}
• 在 handler 中调用 upgrader.Upgrade(w, r, nil)，成功后得到 *websocket.Conn 实例，后续收发消息都靠它

接收和广播消息（基础通信逻辑）

每个连接对应一个 goroutine，持续读取消息；服务端可维护一个连接池（比如用 map[*websocket.Conn]bool），配合互斥锁安全增删。

典型结构：

• 客户端发来消息 → 调用 conn.ReadMessage() 获取 messageType, data []byte
• 简单回传：直接 conn.WriteMessage(messageType, data)
• 广播给所有人：遍历连接池，对每个 conn 调用 WriteMessage（注意加锁读 map，且需检查 conn 是否已关闭）
• 建议统一用 TextMessage 类型，避免二进制解析复杂度

处理连接生命周期与错误退出

WebSocket 连接不是永久的，网络抖动、页面关闭、心跳超时都会导致断连。不能忽略错误，否则 goroutine 泄漏。

关键做法：

• ReadMessage 和 WriteMessage 都可能返回 error，最常见的是 websocket.CloseMessage 或 io.EOF，此时应主动 conn.Close() 并从连接池中移除
• 给每个连接启动两个 goroutine：一个读，一个写（尤其广播场景下，避免写阻塞读）
• 可选加 ping/pong 心跳：设置 conn.SetPingHandler() 和 conn.SetPongHandler()，定期调用 conn.WriteMessage(websocket.PingMessage, nil)

前端简单测试连接（验证通路）

不用框架，纯 HTML + JS 就能快速验证服务是否跑通：

只要控制台打出 "Received: Hello from browser"，说明握手成功、双向通信正常。后续可扩展 JSON 消息格式、用户标识、房间分组等。

基本上就这些。不复杂但容易忽略错误处理和并发安全——把连接管理、读写分离、异常清理做扎实，一个健壮的基础 WebSocket 服务就出来了。