Linux用wrk做HTTP压测教程

本文深入解析了为何 wrk 正在取代传统 ab 成为 Linux 下高并发 HTTP 压测的首选工具——它基于 epoll/kqueue 多路复用，以极低资源消耗实现数万 QPS，更关键的是原生支持 Lua 脚本，轻松应对登录态维持、动态 Token 轮换、随机路径生成等真实业务场景，彻底摆脱 ab 单线程模拟导致的 fd 耗尽、行为失真等痛点；文章不仅手把手教你绕过编译直装二进制版、避开 ulimit 和连接数配置陷阱，更通过可落地的脚本示例揭示了线程与连接分离设计的核心逻辑，助你从“能压”迈向“压得准、压得真”。

wrk 不是 Linux 自带工具，但它是比 ab 更现代、更真实、更适合高并发 HTTP 压测的选择——尤其当你需要模拟多连接、长连接、脚本化行为（比如带 Token 轮换、随机路径）时，ab 很快会力不从心。

为什么不用 ab 而选 wrk？

ab 是单线程事件模型，靠 fork 多进程模拟并发，实际压不出真实客户端行为；而 wrk 基于 epoll（Linux）或 kqueue（macOS），用少量线程 + 多路复用，轻松跑出数万 QPS，且内存占用低、CPU 利用率高。更重要的是：wrk 支持 Lua 脚本，能做登录态维持、动态 header、请求参数变异等 ab 根本做不到的事。

• 常见错误现象：ab -c 1000 -n 10000 http://api.example.com/ 报 socket: Too many open files 或响应时间突然飙升——这不是服务端瓶颈，是 ab 自身 fd 耗尽
• 使用场景：压测网关、微服务接口、JWT 鉴权 API、需保持连接池的 gRPC-HTTP 转发层
• 性能影响：wrk -t4 -c400 -d30s http://localhost:8080/health 比同等参数的 ab 多压出 3–5 倍有效请求量

怎么装 wrk（别用源码编译）

大多数发行版没进默认仓库，但二进制包够用，别折腾 GCC 和 OpenSSL 依赖：

• Ubuntu/Debian：sudo apt install -y curl && curl -L https://github.com/wg/wrk/releases/download/4.2.0/wrk_4.2.0_amd64.deb -o wrk.deb && sudo dpkg -i wrk.deb
• CentOS/RHEL：sudo yum install -y epel-release && sudo yum install -y wrk（EPEL 有打包，版本略旧但稳定）
• 验证：wrk --version 输出类似 wrk 4.2.0 [epoll] sizeof(struct kevent)=32 即可

基础压测命令和关键参数含义

wrk 默认只发 GET，不带任何 header，这点和 ab 类似，但参数语义更直白：

• -t 是线程数（不是“并发用户”），一般设为 CPU 核数，如 -t4
• -c 是总连接数（connection pool size），不是每线程连接数——-t4 -c400 表示 4 个线程共维护 400 个 TCP 连接
• -d 是持续时间，单位秒；-s 指定 Lua 脚本，-H 添加 header，如 -H "Authorization: Bearer xxx"
• 容易踩的坑：-c 设太高（如 10000）但服务端 net.core.somaxconn 或 ulimit -n 没调，客户端自己先卡住；建议先 ulimit -n 65535

示例：wrk -t4 -c200 -d10s -H "User-Agent: wrk-test" http://127.0.0.1:3000/api/v1/users

带登录态的真实压测怎么做？

纯 wrk -H 写死 Cookie 或 Token 只适用于静态鉴权，真实场景要先登录拿 token，再用 token 发业务请求——这必须用 Lua 脚本：

• 写一个 login.lua：用 wrk.method = "POST" 登录，解析响应 JSON 提取 token，存到 wrk.thread:set("token", token)
• 再写 auth-get.lua：在 init 阶段从 thread 取 token，拼到 header 里；每次 request() 都动态生成新请求
• 执行：wrk -t2 -c100 -d30s -s auth-get.lua http://api.example.com/
• 注意：Lua 脚本里不能 sleep、不能阻塞 IO；token 过期要自己加逻辑刷新，否则压测中途全 401

wrk 的真实门槛不在安装，而在理解「连接数」和「线程数」的分离设计，以及 Lua 脚本中 thread-local 状态管理。很多人卡在脚本里 token 没传进去、header 拼错、或忘了 wrk.headers["X-Header"] = value 的赋值语法——这些细节不试三次根本记不住。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于文章的相关知识，也可关注golang学习网公众号。