读写分离可避免阻塞、减少协程切换与内存分配,提升 WebSocket 性能;Reader 与 Writer 协程各司其职,分别处理收发消息并独立控制超时与背压。

提升 Go WebSocket 性能的关键之一,是避免读写操作互相阻塞,并减少 goroutine 频繁切换和内存分配。读写分离 + 缓冲 channel 是实践中最常用、效果最直接的优化手段。
为什么需要读写分离
默认情况下,一个 WebSocket 连接共用一个网络连接,但读和写操作语义不同、频率不同、错误处理逻辑也不同。如果在同一个 goroutine 里同步读写,或共用一个 channel 收发消息,容易导致:
- 写操作被读超时卡住(比如对方不读,send 缓冲满)
- 读操作因写失败而中断(比如 write panic 后未 recover,整个连接崩溃)
- 消息顺序混乱或丢失(多个 goroutine 直接调用 conn.WriteMessage)
读写分离后,读协程只管收、写协程只管发,职责清晰,各自可独立控制超时、重连、背压。
如何实现读写分离结构
每个连接启动两个长期运行的 goroutine:
立即学习“go语言免费学习笔记(深入)”;
-
Reader:循环调用
conn.ReadMessage(),解析后发到client.in -
Writer:从
client.outchannel 拉消息,调用conn.WriteMessage()发送
中间用两个 channel 桥接:
· client.in:无缓冲或小缓冲,用于接收客户端上行消息(如 JSON 指令)· client.out:带缓冲(如 cap=64),用于下行广播或响应,缓解突发推送压力
缓冲 channel 的大小怎么选
缓冲不是越大越好,要平衡内存占用与丢包风险:
- 下行 channel(out)建议设为 32 ~ 128:覆盖大多数广播峰值,避免 writer 频繁阻塞;超过缓冲会丢弃旧消息或返回错误,需配合背压策略(如 drop-oldest 或 reject-new)
- 上行 channel(in)一般用 无缓冲或 1~8:读协程应快速消费,防止客户端发太快把 server 堆满;若业务允许延迟处理,可稍加大,但需配超时 select
示例初始化:
client := &Client{
in: make(chan []byte, 4),
out: make(chan []byte, 64),
}补充几个关键细节
- Writer 必须处理
write deadline和net.ErrClosed,遇到写失败应关闭 channel 并退出 goroutine - Reader 应捕获
websocket.CloseMessage,主动关闭 in/out channel,通知 writer 退出 - 不要在 handler 里直接往
client.out写,统一走封装方法(如client.Send(msg)),内部做 select default 防阻塞 - 大量连接时,考虑用
sync.Pool复用[]byte消息缓冲,减少 GC 压力
基本上就这些。不复杂但容易忽略——读写一混,压测时延迟毛刺和连接抖动立马出现。











