讲师中心 微信公众号
AI工具推荐 视频效率加速

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

大静君_6998

大静君_6998

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

|

814人浏览过

|

来源于php中文网

原创

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

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

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

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

  • 显式设置 Dialer.Timeout 和 Dialer.KeepAlive,比如 10 * time.Second 和 30 * 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 压力陡增。

Golang Uber Dig
Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器,提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象,支持命名依赖等

下载
  • 对固定内容(如 "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 1000000、sysctl -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工具

更多
火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

花生AI
花生AI Hot

一款AI视频创作工具,主要用于B站推出的AI视频创作工具,适合需要提升相关任务效率的用户。

WorkBuddy

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

DeepSeek

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

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

PixTV
PixTV Hot

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

VibeKnow
VibeKnow Hot

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

豆包大模型

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

创客贴AI

创客贴AI是一款AI图像与设计工具,AI辅助的智能在线设计工具。

相关专题

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

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

539

2024.02.23

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

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

616

2024.02.23

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

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

1236

2024.02.23

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

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

791

2024.03.05

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

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

1806

2024.05.21

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

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

5174

2025.06.09

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

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

2394

2025.06.10

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

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

4266

2025.06.17

DBeaver怎么导入数据
DBeaver怎么导入数据

DBeaver数据导入专题,整理Excel、CSV、SQL、TXT等常见文件的数据导入方法,涵盖目标数据库选择、数据表创建、字段映射、数据类型设置及导入参数配置等操作,帮助用户将本地文件中的数据写入数据库,并了解批量导入、重复数据处理和导入失败等常见问题的解决方法。

0

2026.11.05

热门下载

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

精品课程

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

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