17370845950

Go语言处理WebSocket需用gorilla/websocket包，核心包括连接生命周期管理、消息类型区分（文本/二进制/控制帧）、并发安全读写（读写goroutine分离+写通道串行化）、结构化消息路由及超时/错误/心跳等基础配置。

Go语言处理WebSocket消息的核心在于理解连接生命周期、消息类型区分和并发安全的数据读写。标准库golang.org/x/net/websocket已弃用，推荐使用更活跃、性能更好、API更清晰的gorilla/websocket包。

建立连接并设置基础配置

服务端需注册HTTP handler，升级HTTP连接为WebSocket；客户端通过dialer发起连接。关键点是设置合理的读写超时、缓冲区大小和错误处理机制。

• 服务端用upgrader.Upgrade()将http.ResponseWriter转为*websocket.Conn
• 禁用默认ping/pong自动响应（CheckOrigin和SetPingHandler(nil)），便于自定义心跳逻辑
• 客户端连接时建议设置Dialer.TLSClientConfig（如需HTTPS）和Dialer.HandshakeTimeout

区分文本、二进制与控制帧消息

WebSocket协议定义了多种消息类型：文本（websocket.TextMessage）、二进制（websocket.BinaryMessage）、ping/pong（控制帧）、close帧。不区分类型直接读取容易导致解析失败或panic。

• 用conn.ReadMessage()返回(messageType, data []byte, error)，先判断messageType再解码
• 文本消息通常用json.Unmarshal(data, &struct{})；二进制消息建议用encoding/binary或Protocol Buffers解析
• 主动收到websocket.CloseMessage时应调用conn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))后关闭连接

安全读写：避免阻塞与竞态

WebSocket连接是全双工的，但ReadMessage和WriteMessage默认是阻塞的。多个goroutine同时写会导致panic，必须加锁或使用带缓冲的写通道统一调度。

• 每个连接建议启动两个goroutine：一个专用于读（处理业务逻辑+心跳检测），一个专用于写（从channel消费消息）
• 写操作统一走chan []byte，由writer goroutine串行调用conn.WriteMessage()，避免并发写冲突
• 读循环中检查net.ErrClosed或websocket.IsUnexpectedCloseError，及时退出并清理资源

结构化消息解析与路由分发

实际项目中，客户端发送的消息往往带type字段（如"login"、"chat"、"ping"），服务端需按类型分发到不同处理器，类似轻量级RPC。

• 定义统一消息结构体，例如：type Message struct { Type string `json:"type"` Data json.RawMessage `json:"data"` }
• 用map[string]func(*Conn, json.RawMessage) error注册处理器，根据msg.Type查表调用
• 对Data字段做二次解码前，先校验JSON合法性（json.Valid()）防止恶意输入触发panic

基本上就这些。Golang处理WebSocket不复杂但容易忽略超时、类型判断和并发写保护。用好gorilla/websocket + 明确的连接状态管理 + 消息路由设计，就能支撑万级长连接稳定运行。