讲师中心 微信公众号

工具推荐

视频效率加速
Laper
Laper

剧本策划到成片协作

updream

一键生成高质量推广成片

Seko
Seko

AI短剧一键生成成片

AniShort
AniShort

AI短剧协同制作平台

Go语言实现WebSocket高并发服务器 WebSocket底层原理剖析【案例】

秋宇酱_6171

秋宇酱_6171

发布时间:2026-06-02 15:37:19

|

404人浏览过

|

来源于php中文网

原创

Go的goroutine不会因WebSocket连接数暴涨而崩溃,根本原因是其极低开销(初始栈仅2KB)和M:N调度模型,万级连接对应万级goroutine而非OS线程;但必须拆分为独立读/写协程,避免ReadMessage/WriteMessage阻塞导致协程挂起。

go语言实现websocket高并发服务器 websocket底层原理剖析【案例】

为什么 goroutine 不会因 WebSocket 连接数暴涨而崩溃

Go 的并发模型不是靠“线程多”撑住高连接,而是靠 goroutine 的极低开销(初始栈仅 2KB)和 Go 调度器的 M:N 协程映射。一个稳定运行的 WebSocket 服务端,每连接只需 1–2 个 goroutine(读 + 写),万级连接对应的是万级 goroutine,而非万级 OS 线程——这是它不卡死的根本原因。

但前提是:不能在单个 goroutine 中阻塞读写。常见错误是把 conn.ReadMessage() 和 conn.WriteMessage() 放在同一个循环里,一旦写阻塞(比如客户端网络卡顿),整个协程就挂住,后续消息无法读取,最终堆积触发 iowait 上升。

  • 每个连接必须拆成至少两个独立 goroutine:一个专读(ReadMessage),一个专写(WriteMessage 或带缓冲的 writePump)
  • 读协程收到 io.EOF 或 websocket.CloseMessage 后应主动退出,而不是 panic 或忽略
  • 写协程需用带缓冲的 chan []byte 接收消息,避免直接调用 WriteMessage 阻塞读协程

如何让 Upgrader 兼容 SockJS、Spring WebFlux 等非标服务端

gorilla/websocket.Upgrader 默认校验 Origin 头、拒绝空子协议、严格比对 Sec-WebSocket-Accept,导致与 SockJS 封装层或某些 Spring Boot 配置握手失败。这不是 bug,是 RFC 合规性设计,但生产环境必须妥协。

关键修复点是重写 CheckOrigin 并允许空或宽松匹配:

立即学习“go语言免费学习笔记(深入)”;

Go语言(Golang)1.26.0
Go语言(Golang)1.26.0

Go语言(Golang)1.26.0版本官方下载,版本号 1.26.0,适合旧项目维护、兼容性测试和指定版本开发环境搭建。

下载
var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool {
        // 允许任意 Origin(开发/内网可信场景)
        return true
        // 或更安全:只放行已知域名
        // origin := r.Header.Get("Origin")
        // return origin == "https://myapp.com" || origin == ""
    },
    // 若服务端要求子协议但未声明,可设为空切片避免校验失败
    Subprotocols: []string{},
}
  • 永远不要在生产环境无条件返回 true,至少加白名单判断
  • 若仍握手失败,检查服务端是否发送了非法的 Sec-WebSocket-Extensions 头,此时需用自定义 http.RoundTripper 过滤响应头(客户端侧)
  • Subprotocols 设为 []string{} 可绕过子协议协商失败,但若服务端强依赖某协议(如 graphql-ws),则必须显式传入

心跳机制必须由服务端主动发 ping,不能只依赖 TCP KeepAlive

TCP 层的 KeepAlive 默认 2 小时才探测一次,远慢于 WebSocket 应用层心跳(通常 30–45 秒)。如果只靠系统级保活,网络中断后连接会滞留数小时,netstat 显示 ESTABLISHED 实为“半死”状态,持续消耗 fd 和内存。

正确做法是在服务端 Upgrader 后立即启动 pongHandler,并定时 WriteControl 发送 ping:

conn.SetPingHandler(func(appData string) error {
    return conn.WriteControl(websocket.PongMessage, []byte(appData), time.Now().Add(10*time.Second))
})
<p>// 在连接建立后的主循环中
ticker := time.NewTicker(30 <em> time.Second)
go func() {
defer ticker.Stop()
for range ticker.C {
if err := conn.WriteControl(websocket.PingMessage, nil, time.Now().Add(10</em>time.Second)); err != nil {
return // 连接已断,退出
}
}
}()
  • SetPingHandler 必须在 Upgrade 成功后立刻设置,否则首次 pong 可能丢失
  • WriteControl 是线程安全的,但不可并发调用;多个 goroutine 需通过 channel 或 mutex 同步
  • 客户端若不回 pong,服务端不会自动关连接——必须结合 SetReadDeadline 检测超时读,否则 “假在线” 会越积越多

消息广播性能瓶颈不在 channel,而在锁和内存拷贝

当需要向数千客户端广播同一条消息(如行情快照),用 for range clients { conn.WriteMessage(...) } 是最慢的。问题不在于 channel 传递,而在于:每次 WriteMessage 都触发一次 syscall write、一次 TLS 加密(若启用)、一次内存 copy(底层 bufio.Writer 缓冲区填充)。

优化路径有两条:

  • 对全量广播场景,改用 conn.WriteMessage 前先做一次 json.Marshal,复用字节切片,避免每个连接重复序列化
  • 对高频小消息(如心跳、状态变更),启用 conn.EnableWriteCompression(true),但注意:压缩只对 >1024 字节有效,且服务端需支持 permessage-deflate 扩展
  • 终极方案是引入消息分片 + 批量写:将待发消息暂存到 ring buffer,由专用 writer goroutine 统一 flush,减少系统调用次数

真正容易被忽略的是:广播时没做连接有效性检查。遍历 clients map 时若不加锁,或未在写前调用 conn.SetWriteDeadline,极易因某个慢连接拖垮整批广播——这比 channel 容量不足更致命。

热门AI工具

更多
Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

相关专题

更多
C语言变量命名
C语言变量命名

c语言变量名规则是:1、变量名以英文字母开头;2、变量名中的字母是区分大小写的;3、变量名不能是关键字;4、变量名中不能包含空格、标点符号和类型说明符。php中文网还提供c语言变量的相关下载、相关课程等内容,供大家免费下载使用。

3849

2023.06.20

c语言入门自学零基础
c语言入门自学零基础

C语言是当代人学习及生活中的必备基础知识,应用十分广泛,本专题为大家c语言入门自学零基础的相关文章,以及相关课程,感兴趣的朋友千万不要错过了。

2508

2023.07.25

c语言运算符的优先级顺序
c语言运算符的优先级顺序

c语言运算符的优先级顺序是括号运算符 > 一元运算符 > 算术运算符 > 移位运算符 > 关系运算符 > 位运算符 > 逻辑运算符 > 赋值运算符 > 逗号运算符。本专题为大家提供c语言运算符相关的各种文章、以及下载和课程。

1440

2023.08.02

c语言数据结构
c语言数据结构

数据结构是指将数据按照一定的方式组织和存储的方法。它是计算机科学中的重要概念,用来描述和解决实际问题中的数据组织和处理问题。数据结构可以分为线性结构和非线性结构。线性结构包括数组、链表、堆栈和队列等,而非线性结构包括树和图等。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

1378

2023.08.09

c语言random函数用法
c语言random函数用法

c语言random函数用法:1、random.random,随机生成(0,1)之间的浮点数;2、random.randint,随机生成在范围之内的整数,两个参数分别表示上限和下限;3、random.randrange,在指定范围内,按指定基数递增的集合中获得一个随机数;4、random.choice,从序列中随机抽选一个数;5、random.shuffle,随机排序。

1456

2023.09.05

c语言const用法
c语言const用法

const是关键字,可以用于声明常量、函数参数中的const修饰符、const修饰函数返回值、const修饰指针。详细介绍:1、声明常量,const关键字可用于声明常量,常量的值在程序运行期间不可修改,常量可以是基本数据类型,如整数、浮点数、字符等,也可是自定义的数据类型;2、函数参数中的const修饰符,const关键字可用于函数的参数中,表示该参数在函数内部不可修改等等。

2438

2023.09.20

c语言get函数的用法
c语言get函数的用法

get函数是一个用于从输入流中获取字符的函数。可以从键盘、文件或其他输入设备中读取字符,并将其存储在指定的变量中。本文介绍了get函数的用法以及一些相关的注意事项。希望这篇文章能够帮助你更好地理解和使用get函数 。

4100

2023.09.20

c数组初始化的方法
c数组初始化的方法

c语言数组初始化的方法有直接赋值法、不完全初始化法、省略数组长度法和二维数组初始化法。详细介绍:1、直接赋值法,这种方法可以直接将数组的值进行初始化;2、不完全初始化法,。这种方法可以在一定程度上节省内存空间;3、省略数组长度法,这种方法可以让编译器自动计算数组的长度;4、二维数组初始化法等等。

19635

2023.09.22

Go-Zero API开发使用教程
Go-Zero API开发使用教程

本专题聚焦Go-Zero的API开发,拆解.api描述文件的语法结构、路由与分组定义、参数绑定与校验规则,说明统一响应体、JWT鉴权、跨域与接口文档生成的落地方式,并给出网关聚合与接口测试的实践建议。

0

2026.11.12

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn