Go中实现WebSocket服务器应使用gorilla/websocket库，核心是安全连接管理、分离读写goroutine、用sync.Map集中维护连接、启用心跳与deadline机制处理异常并及时清理资源。
在 Go 中实现 WebSocket 服务器，核心是使用 gorilla/websocket 库——它稳定、文档清晰、社区支持好，是目前最主流的选择。关键不在“能不能连”，而在于如何安全地管理连接、广播消息、处理异常和避免 goroutine 泄漏。
初始化 WebSocket 连接并升级 HTTP 请求
WebSocket 基于 HTTP 升级（Upgrade）机制，需在 HTTP handler 中显式调用 Upgrader.Upgrade()。注意设置允许跨域（开发阶段常用），并禁用默认的 Origin 检查（生产环境应自行校验）：
- 定义全局
Upgrader实例，复用可避免重复分配 - 调用
upgrader.Upgrade(w, r, nil)将响应升级为 WebSocket 连接 - 若请求头不含
Upgrade: websocket或握手失败，该方法会自动返回 400/403 等错误
为每个连接启动独立读写 goroutine
每个客户端连接应分离读、写逻辑，防止阻塞。典型做法是：一个 goroutine 负责 ReadMessage（接收客户端消息），另一个负责 WriteMessage（发送消息），中间通过 channel 通信：
- 读 goroutine 持续调用
conn.ReadMessage()，收到消息后转发到中心广播 channel 或路由到指定用户 - 写 goroutine 从 channel 接收待发送数据，调用
conn.WriteMessage()；需加锁或使用conn.SetWriteDeadline()防止写阻塞 - 务必用
defer conn.Close()并在退出前从连接池中移除该连接
集中管理连接与安全广播
用线程安全的 map（如 sync.Map）或带互斥锁的结构体存储活跃连接，键可为用户 ID 或随机 session ID：
立即学习“go语言免费学习笔记（深入）”；
- 新连接建立时，生成唯一标识、存入 map、触发登录事件（如通知其他用户）
- 广播消息时，遍历 map 中所有连接，对每个
*websocket.Conn调用WriteMessage；遇到写错误（如客户端断开）立即关闭并清理 - 禁止直接把原始 HTTP request body 或未校验的 JSON 字段原样广播，应解析、过滤、再序列化
处理常见异常与资源回收
WebSocket 连接极易因网络抖动、页面关闭、心跳超时而中断，必须主动检测并清理：
- 启用心跳：服务端定期调用
conn.WriteMessage(websocket.PingMessage, nil)，并设置conn.SetPingHandler()处理客户端 Ping - 设读写 deadline：用
conn.SetReadDeadline()和conn.SetWriteDeadline()配合定时器，超时即断连 - 捕获
websocket.IsUnexpectedCloseError()和io.EOF，它们代表正常断开；其余错误（如 net.OpError）视为异常，需记录日志
