压力测试工具Wrk介绍
压力测试工具Wrk介绍
概念
GitHub地址：https://github.com/wg/wrk
安装方式
安装环境：Contos7
- 切换到用户级的源码目录
cd /usr/local/src
- 安装git，因为要从GitHub下载，如果已经安装，忽略此步，其他机器请使用其他命令如apt-get
yum install git -y
- 从GitHub中clone源码
git clone https://github.com/wg/wrk.git
- 完成以后，进入wrk目录
cd wrk
- 然后执行编译操作，编译过程会比较久
make
- 创建一个软连接，方便其他目录下使用
ln -s /usr/local/src/wrk/wrk /usr/local/bin
基本用法
- 一般用法
wrk -t1 -c400 —timeout 5s -d 100s —latency ‘http://localhost:8080/index.html‘
- 添加Header和Lua脚本（Lua脚本后面再详细说）
wrk -t1 -c400 —timeout 5s -d 100s -s ‘/usr/local/src/test.lua’ -H “Authorization: ABCDEFGH123OPQ” —latency ‘http://localhost:8080/index.html‘
参数说明
- -c：总的连接数（每个线程处理的连接数=总连接数/线程数）
- -d：测试的持续时间，如2s(2second)，2m(2minute)，2h(hour)
- -t：需要执行的线程总数
- -s：执行Lua脚本，这里写lua脚本的路径和名称，后面会给出案例
- -H：需要添加的头信息，注意header的语法，举例，-H “token: abcdef”，说明一下，token，冒号，空格，abcdefg（不要忘记空格，否则会报错的）。
- —timeout：超时的时间
- —latency：显示延迟统计信息
返回结果
- Latency：响应时间
- Req/Sec：每个线程每秒钟的执行的连接数
- Avg：平均
- Max：最大
- Stdev：标准差
- +/- Stdev： 正负一个标准差占比
- Requests/sec：每秒请求数（也就是QPS），这是一项压力测试的性能指标，通过这个参数可以看出吞吐量
- Latency Distribution，如果命名中添加了—latency就会出现相关信息
注意检查连接状态
每次压测完成后，下一次压测前，建议先检查一下之前的连接是否全部断开，然后再重新压测，命令如下：
ss -sn
返回结果如下，timewait 0/0代表目前没有等待，说明所有的连接已全部断开
TTCP: 13 (estab 7, closed 2, orphaned 0, synrecv 0, timewait 1/0), ports 14
Transport Total IP IPv6
* 180 - -
RAW 0 0 0
UDP 4 4 0
TCP 11 11 0
INET 15 15 0
FRAG 0 0 0
注意事项
- 使用wrk无法看到response，不能确定执行的接口返回的值是否复合预期，所以建议先基准测试，但是wrk不输出response信息，那就想办法输出一下response信息，确定结果符合预期，再压测，具体参见后面的lua脚本
- 一般线程数是系统CPU核数的2~4倍，可以根据业务需要，自行调整
Lua脚本的相关介绍
为什么要使用这个脚本呢？header太长了，可以写在脚本里，还有一个上面介绍的“一般用法”只能执行get请求，对于post或者put的这种接口很是无奈，所以要扩展一下，废话不多说了，直接开始。
详细的lua脚本可以参考https://github.com/wg/wrk/tree/master/scripts的文件。
Post请求，如创建test_post.lua文件
awrk.method = "POST"
wrk.body = "foo=bar&baz=quux"
wrk.headers["Content-Type"] = "application/x-www-form-urlencoded"
使用的时候在wrk通过-s引入这个lua文件即可。
lua还提供了几个函数
setup 函数 、init 函数 、delay函数、request函数 、response函数、done函数。
谈一谈如何做基准测试，首先用到的是response函数，这个就是返回响应数据的，然后可以结合done函数使用，done函数的作用就是在所有请求执行完成后调用，一般用于统计结果。
既然要执行基准测试，那就一个线程，一个连接就可以了，不说done函数了，主要说一下我如何利用response函数做基准测试的。
function response(status, headers, body)
print(body)
end
就把上述的代码写到lua文件中，然后引用lua文件就可以了，无论返回的结果是json或者是html，都会被打印出来，先说一下我遇到的情况，执行了一个线程，一个连接，但是wrk返回结果显示很多个请求（至于原因还没整明白），然后打印出了很多信息，虽然是基准测试，没有什么影响，但是内容太多，排查信息很麻烦，既然输出了多次结果，那就说明执行了多次response函数，这就好办了，让他执行一次response函数就好了。
a=1
function response(status, headers, body)
if(a==1)
then
a=2
print(body)
end
end
简单说一下改造后的代码，先定义一个a=1，第一次调用response函数的时候判断，如果a的值为1，那就打印出返回的body，然后把a的状态改变为2，下一次再调用response函数的时候，a的值为2了，就不会再执行print函数，只会打印出一次结果。压测的时候如果lua中不含头信息和post数据，直接不引用即可，如果有这些信息，那就把response函数注释就OK了。
-
语法注意，lua的每一个功能块的结尾都需要end结尾，一个方法需要，甚至一个if语句块或者for循环的语句块都需要有end结尾。
-
再谈一谈扩展性，上面的内容介绍的都差不多了，可以把接口lua脚本都准备好，写一个shell脚本，利用变量执行不同并发的压测，利用ss -sn命令令、正则表达式、控制语句等自动执行压测。
补充说明：最近压测发现Wrk、ab、JMeter压测的结果有一些差距，这个问题需要再进一步深入研究一下。