0

0

使用Golang进行WebSocket长连接的并发压力测试

P粉602998670

P粉602998670

发布时间:2026-02-23 18:12:38

|

794人浏览过

|

来源于php中文网

原创

大量连接卡在handshake因http.transport默认配置宽松导致tcp连接未及时释放,需显式设置dialer超时、禁用http/2、限制本地端口范围、避免复用client实例。

使用golang进行websocket长连接的并发压力测试

gorilla/websocket 模拟高并发连接时,为什么大量连接卡在 handshake?

因为默认的 http.Transport 对连接复用和超时控制太宽松,底层 TCP 连接没及时释放,dialer 会阻塞在 DNS 解析或 TLS 握手阶段。尤其在短时发起数千连接时,本地端口耗尽、TIME_WAIT 积压、DNS 缓存未生效都会放大问题。

  • 显式设置 Dialer.TimeoutDialer.KeepAlive,比如 10 * time.Second30 * time.Second
  • 禁用 HTTP/2:http.DefaultTransport.(*http.Transport).TLSNextProto = make(map[string]func(string, *tls.Conn) http.RoundTripper),避免握手协商开销
  • net.Dialer 控制本地端口范围(如 LocalAddr: &net.TCPAddr{Port: 0})避免 bind 失败
  • 别复用同一个 http.Client 实例做千级并发 dial —— 每个 goroutine 应该用独立 Dialer 实例,否则锁争用严重

websocket.WriteMessage 频繁报 write tcp: use of closed network connection 怎么办?

不是网络问题,是连接被主动关闭后,还有 goroutine 在往已关闭的 *websocket.Conn 写数据。常见于:心跳检测失败后只关读、没同步关写;或 panic 后 defer 没覆盖全部路径。

  • 写操作前加 if conn == nil || conn.IsClosed() { return }IsClosed()gorilla/websocket v1.5+ 提供的非阻塞判断
  • 写逻辑必须包裹在 select 中监听 conn.CloseNotify() 或自定义 done channel,一收到关闭信号立刻退出写循环
  • 避免在 WriteMessage 后立刻调用 Close() —— 改用 WriteControl(websocket.CloseMessage, ...) 主动发 close 帧,再等对方响应
  • 测试时用 conn.SetWriteDeadline(time.Now().Add(500 * time.Millisecond)) 暴露写阻塞问题,比默认无限等待更易定位

压测中 CPU 突增但 QPS 上不去,runtime/pprof 显示大量 websocket.preparedMessage 占用堆内存

这是典型的消息预分配滥用:每次调用 WriteMessage 都会拷贝 payload 并封装成 preparedMessage,如果反复发送相同小消息(如心跳 ping),对象分配频率极高,GC 压力陡增。

稿定PPT
稿定PPT

海量PPT模版资源库

下载
  • 对固定内容(如 "ping")提前构建 websocket.PreparedMessage 实例,复用而非每次都 new
  • conn.WriteControl() 发送 ping/pong —— 它不走 message 编码流程,零分配
  • 关闭 WriteBufferPool(设为 nil)反而更稳:压测场景下 pool 的 lock 争用常比分配还重
  • 检查是否误将大结构体 JSON 序列化后直传给 WriteMessage —— 应先序列化到 []byte 再传,避免闭包捕获导致逃逸

如何让单机压出 10w+ WebSocket 连接而不 OOM 或卡死?

关键不在 Go 并发模型多强,而在 OS 层和连接生命周期管理是否精细。Golang 能跑满连接数,但系统限制和连接“假活”才是瓶颈。

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

  • 调高系统参数:ulimit -n 1000000sysctl -w net.core.somaxconn=65535、关闭 net.ipv4.tcp_tw_reuse(压测环境可开,但需确认服务端也配好)
  • 每个连接必须带独立 context,并在 goroutine 启动时就监听 cancel —— 别等 write/read 报错才退出
  • 禁用 SetReadDeadline 的绝对时间模式,改用 SetReadLimit + 心跳计数器,避免大量 timer 对象堆积
  • 连接建立后立即发一次 ping 并等待 pong,失败则立刻 close —— 防止“半开连接”占着 fd 不放

真正难的不是启动连接,是让它们按预期断开。很多压测脚本卡在 8w 连接不动,其实是 2w 连接早已静默断连却没被回收,fd 被占着,新连接 bind 失败。盯着 /proc/<pid>/fd</pid>ss -s 才能看清真相。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
golang如何定义变量
golang如何定义变量

golang定义变量的方法:1、声明变量并赋予初始值“var age int =值”;2、声明变量但不赋初始值“var age int”;3、使用短变量声明“age :=值”等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

207

2024.02.23

golang有哪些数据转换方法
golang有哪些数据转换方法

golang数据转换方法:1、类型转换操作符;2、类型断言;3、字符串和数字之间的转换;4、JSON序列化和反序列化;5、使用标准库进行数据转换;6、使用第三方库进行数据转换;7、自定义数据转换函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

242

2024.02.23

golang常用库有哪些
golang常用库有哪些

golang常用库有:1、标准库;2、字符串处理库;3、网络库;4、加密库;5、压缩库;6、xml和json解析库;7、日期和时间库;8、数据库操作库;9、文件操作库;10、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

349

2024.02.23

golang和python的区别是什么
golang和python的区别是什么

golang和python的区别是:1、golang是一种编译型语言,而python是一种解释型语言;2、golang天生支持并发编程,而python对并发与并行的支持相对较弱等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

212

2024.03.05

golang是免费的吗
golang是免费的吗

golang是免费的。golang是google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的开源编程语言,采用bsd开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

404

2024.05.21

golang结构体相关大全
golang结构体相关大全

本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

365

2025.06.09

golang相关判断方法
golang相关判断方法

本专题整合了golang相关判断方法,想了解更详细的相关内容,请阅读下面的文章。

200

2025.06.10

golang数组使用方法
golang数组使用方法

本专题整合了golang数组用法,想了解更多的相关内容,请阅读专题下面的文章。

1071

2025.06.17

pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法
pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法

本专题系统整理pixiv网页版官网入口及登录访问方式,涵盖官网登录页面直达路径、在线阅读入口及快速进入方法说明,帮助用户高效找到pixiv官方网站,实现便捷、安全的网页端浏览与账号登录体验。

1127

2026.02.13

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Go 教程
Go 教程

共32课时 | 5.5万人学习

Go语言实战之 GraphQL
Go语言实战之 GraphQL

共10课时 | 0.9万人学习

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号