为什么选择WebSocket？

在传统HTTP协议中，客户端必须主动发起请求才能获取数据。但对于实时聊天室、在线协作编辑、股票行情推送等场景，我们需要更高效的双向通信机制。WebSocket协议应运而生，它在单个TCP连接上提供全双工通信通道，延迟仅为HTTP轮询的1/5。

Golang WebSocket实现方案

1. 核心库选择

推荐使用经过生产验证的gorilla/websocket库：
go go get github.com/gorilla/websocket

2. 建立WebSocket服务端

go
package main

import (
"net/http"
"github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
CheckOrigin: func(r *http.Request) bool {
return true // 生产环境应验证Origin
},
}

func handler(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println("升级WebSocket失败:", err)
return
}
defer conn.Close()

for {
    // 读取客户端消息
    messageType, p, err := conn.ReadMessage()
    if err != nil {
        log.Println("读取错误:", err)
        break
    }

    // 处理消息（示例：原样返回）
    if err := conn.WriteMessage(messageType, p); err != nil {
        log.Println("写入错误:", err)
        break
    }
}

}

func main() {
http.HandleFunc("/ws", handler)
http.ListenAndServe(":8080", nil)
}

3. 客户端实现

对应HTML客户端代码：html

高级实践技巧

连接管理优化

实际项目中需要管理大量连接：go
type Client struct {
conn *websocket.Conn
send chan []byte
}

var clients = make(map[*Client]bool)
var broadcast = make(chan []byte)

func (c *Client) readPump() {
defer func() {
delete(clients, c)
c.conn.Close()
}()

for {
    _, message, err := c.conn.ReadMessage()
    if err != nil {
        break
    }
    broadcast <- message
}

}

func (c *Client) writePump() {
defer c.conn.Close()
for message := range c.send {
if err := c.conn.WriteMessage(websocket.TextMessage, message); err != nil {
break
}
}
}

性能调优建议

1. 心跳机制：定期Ping/Pong保持连接活性
2. 消息压缩：对文本消息启用websocket.Compressor
3. 连接池：复用TCP连接减少握手开销
4. 协议升级：考虑使用WSS(WebSocket Secure)

常见问题解决方案

Q：如何处理连接中断？
go conn.SetCloseHandler(func(code int, text string) error { log.Printf("连接关闭: %d %s", code, text) return nil })

Q：如何控制消息大小？
go upgrader.ReadBufferSize = 2048 // 限制单条消息最大2KB

Q：怎样实现广播功能？
go func handleBroadcast() { for { msg := <-broadcast for client := range clients { select { case client.send <- msg: default: close(client.send) delete(clients, client) } } } }

生产环境注意事项

1. 安全防护：

   
   
   
   • 必须验证Origin头
   • 启用WSS加密通信
   • 限制单IP连接数

2. 监控指标：
   go prometheus.MustRegister(wsConnections) // 在连接建立/关闭时更新指标

3. 优雅退出：
   go // 捕获中断信号 signal.Notify(interrupt, os.Interrupt) <-interrupt // 通知所有客户端连接即将关闭