要构建高性能的websocket服务，使用golang中的gorilla/websocket库是常见做法。1. 安装库：go get github.com/gorilla/websocket；2. 在http handler中通过upgrader升级连接并设置缓冲区与跨域策略；3. 为每个连接维护读写循环，分别在独立协程中处理消息收发；4. 优化性能时调整缓冲区大小、使用channel串行化写入操作以避免并发问题；5. 设置心跳机制保持长连接活跃状态；6. 注意资源回收和连接管理以支撑高并发场景。合理配置后即可高效支持实时通信业务。
构建高性能的 WebSocket 服务在 Golang 中是一个常见的需求，尤其是在实时通信、聊天系统、在线协作等场景下。而 gorilla/websocket 是 Go 社区中使用最广泛、性能稳定的 WebSocket 库之一。它不仅简单易用，而且具备良好的扩展性，适合用来搭建高效的服务。
如何开始使用 gorilla/websocket？
要使用 gorilla/websocket，首先需要安装这个库：
go get github.com/gorilla/websocket
然后，在你的 HTTP handler 中升级连接到 WebSocket 协议。关键点在于使用 Upgrader 结构体来控制升级过程，比如设置跨域策略、缓冲大小等。
立即学习“go语言免费学习笔记（深入）”；
一个简单的升级示例：
var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
CheckOrigin: func(r *http.Request) bool {
return true // 允许所有跨域请求，生产环境建议配置具体域名
},
}
func wsHandler(w http.ResponseWriter, r *http.Request) {
conn, _ := upgrader.Upgrade(w, r, nil)
// 处理连接逻辑
}这一步完成后，你就可以通过 conn 进行消息的读写操作了。
如何处理 WebSocket 的消息收发？
WebSocket 本质上是双向通信，所以你需要为每个连接维护一个读写循环。
通常的做法是：
- 启动一个 goroutine 用于读取消息
- 另一个 goroutine 或主协程用于发送消息
例如：
func handleConnection(conn *websocket.Conn) {
go func() {
for {
messageType, p, err := conn.ReadMessage()
if err != nil {
log.Println("read error:", err)
break
}
fmt.Printf("Received: %s\n", p)
// 可以在这里处理业务逻辑
}
}()
// 发送消息示例
ticker := time.NewTicker(time.Second * 5)
defer ticker.Stop()
for {
select {
case <-ticker.C:
err := conn.WriteMessage(websocket.TextMessage, []byte("heartbeat"))
if err != nil {
log.Println("write error:", err)
return
}
}
}
}这里有几个细节需要注意：
-
ReadMessage和WriteMessage都会阻塞当前 goroutine，所以必须分别放在不同的协程中处理 - 写入时要考虑并发问题，可以加锁或者使用带缓冲的通道（channel）做中间队列
- 设置合理的缓冲区大小，避免内存浪费或频繁分配
如何优化性能和资源管理？
虽然 gorilla/websocket 本身已经很轻量高效，但实际部署时还需要做一些调优工作，特别是高并发场景下。
调整缓冲区大小
默认的缓冲区可能太小，影响吞吐量。你可以根据业务特点调整 ReadBufferSize 和 WriteBufferSize，例如设为 8KB 或更大：
upgrader.ReadBufferSize = 8192 upgrader.WriteBufferSize = 8192
控制并发写入
多个 goroutine 同时调用 WriteMessage 会导致 panic，因为 WebSocket 不是并发安全的。推荐做法是使用一个 channel 来串行化写入操作：
type Client struct {
conn *websocket.Conn
send chan []byte
}
func (c *Client) writePump() {
for message := range c.send {
err := c.conn.WriteMessage(websocket.TextMessage, message)
if err != nil {
close(c.send)
return
}
}
}这样即使有多个地方想发送消息，也可以统一推送到 channel，由一个写协程负责发送。
设置心跳机制
长时间空闲连接容易被中间代理关闭，因此建议开启心跳机制。可以通过定时发送 ping 消息来保持活跃状态：
conn.SetReadDeadline(time.Now().Add(60 * time.Second))
conn.SetPongHandler(func(appData string) error {
conn.SetReadDeadline(time.Now().Add(60 * time.Second)) // 收到 pong 后重置超时
return nil
})同时定期发送 ping：
ticker := time.NewTicker(30 * time.Second)
for {
select {
case <-ticker.C:
conn.WriteControl(websocket.PingMessage, []byte{}, time.Now().Add(10*time.Second))
}
}总结
用 gorilla/websocket 构建高性能服务并不复杂，但要注意连接管理、并发控制、心跳机制这些关键点。只要合理设置参数并做好资源回收，就能支撑起大规模的实时通信场景。基本上就这些，剩下的就是根据具体业务逻辑去扩展了。
