wrk（1）- 详细使用

小菠萝测试笔记

发布于 2021-06-17 20:05:02

1.2K00

代码可运行

文章被收录于专栏：自动化、性能测试自动化、性能测试

运行总次数：0

代码可运行

背景

公司对 http 接口都是直接用 wrk 进行压测，简单快捷
刚好有个压测任务分到我
而且中台跟 Lua 也有关系，刚好 wrk 也支持 Lua 脚本，所以学起来，再用起来

介绍

wrk 是一个类似 ab(apache bench)、jmeter 的压力测试工具，官方称它为：现代的 HTTP 基准测试工具
用 C 编写的 HTTP 协议压测工具
底层基于 epoll 和 kqueue 实现，使用了多线程和多路复用 IO（非阻塞 IO），利用异步的事件驱动框架，通过很少的线程就可以压出很大的并发量
降低测试工具本身性能开销对测试结果准确性的影响
支持使用 LuaJIT 脚本，可以执行 HTTP 请求生成、响应处理和自定义报告

它的定位

轻量级性能测试工具
仅支持 HTTP 协议
仅支持单机压测，多机器压测需要每个机器都手动执行一次 wrk 命令
不可取代 Jmeter、LR 等专业性能工具

架构&简单源码解析

在 wrk 里面，每个线程都有自己独立的 Lua 虚拟机和 Event Loop
通过命令行参数 -c 指定的连接数，会平均分给所有线程，每个新建的 socket，都会调用 fcntl 将其设置为 NONBLOCK，即非阻塞，然后托管给 Event Loop
直接使用 redis 的 Event Loop 实现，适配了不同操作系统的实现
启动的时候，每个线程都会新建一个 Lua State，并调用 luaL_dofile 加载命令行参数 -s 指定的 lua 脚本文件
如果没有自定义的 lua 脚本，wrk 默认发送的是 HTTP 1.1 GET 请求，用长连接

语法格式

Usage: wrk <options> <url>
  Options:
    -c, --connections <N>  Connections to keep open
    -d, --duration    <T>  Duration of test
    -t, --threads     <N>  Number of threads to use

    -s, --script      <S>  Load Lua script file
    -H, --header      <H>  Add header to request
        --latency          Print latency statistics
        --timeout     <T>  Socket/request timeout
    -v, --version          Print version details

  Numeric arguments may include a SI unit (1k, 1M, 1G)
  Time arguments may include a time unit (2s, 2m, 2h)

参数说明

-c：与服务器保持的 http 连接数
-d：压测持续运行时间，可以是 2s、2m、2h
-t：启动的线程数
-s：指定 lua 脚本
-H：自定义 http header 请求头，例如："User-Agent: benchmark-wrk"
--latency：打印延迟统计数据
--time：http 超时时间，如果在此时间内未收到响应，则当做超时

数字参数：可以使用 1k、1M、1G 单位

-t

一般是 CPU 核数，最大不要超过 CPUx2 核数，否则会带来额外的上下文切换，将线程数设置为 CPU 核数主要是为了 WRK 能最大化利用 CPU，使结果更准确（截取网上，暂时没看到官方推荐）
和并发数没有直接关系
查看 Linux CPU 总核数：

grep processor /proc/cpuinfo |wc -l

-c

连接数（connection）可以理解为并发数
一般在测试过程中，这个值需要使用者不断向上调试，直至 QPS 达到一个临界点，便可认为此时的并发数为系统所能承受的最大并发量
实际上，wrk 会为每个线程分配（c/t）个 socket 连接
每个连接会先执行请求动作，然后等待直到收到响应后才会再发送请求，所以每个时间点的并发数大致等于连接数（connection）

官方 Tips

运行 wrk 的机器必须有足够数量的临时端口可用，关闭的 socket 必须快速回收
仅更改 HTTP 方法、路径、添加请求头或正文的用户脚本不会对性能产生影响
每个请求的操作，特别是构建新的 HTTP 请求，以及 response() 的使用将必然减少可以生成的负载量

简单栗子

启动 2 个线程，保持 5 个 http 连接打开的状态下，持续压测 10s 的基准测试

wrk -t2 -c5 -d10s https://httpbin.org/get

qps 是 14.85

启动 16 个线程，保持 400 个 http 连接打开的状态下，持续压测 5s 的基准测试，并打印延迟统计数据

wrk -t16 -c400 -d5s --latency https://httpbin.org/get

qps 是 578.7

结果解析

Running 5s test @ https://httpbin.org/get   压测时间5s
  16 threads and 400 connections   共16个测试线程，400个连接，和上参数设置一样
  Thread Stats   Avg      Stdev     Max   +/- Stdev
                平均值    标准差     最大值  正负标准差的范围，越大表示值和平均值不会差很多，离散也不大，表示 Avg 相对可信
    Latency   311.74ms  211.97ms   1.57s    90.79%
    延迟
    Req/Sec    47.86     31.17   170.00     66.79%
    每个线程每秒的完成的请求数
  Latency Distribution 延迟分布
     50%  234.77ms
     75%  244.43ms
     90%  402.99ms
     99%    1.26s     99% 的请求在 1.26s 内完成
  2938 requests in 5.08s, 1.21MB read  5.08 s内共处理完成了 2938 个请求，读取了 1.21MB 数据
  Socket errors: connect 166, read 0, write 0, timeout 1   Socket 成功连接 166个，超时 1 个
Requests/sec:    578.70  平均每秒处理完成 578.7 个请求，QPS=578.7
Transfer/sec:    243.05KB  平均每秒读取数据 243.05KB