概念
GitHub地址:https://github.com/wg/wrk
安装环境:Contos7
cd /usr/local/src
yum install git -y
git clone https://github.com/wg/wrk.git
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‘
wrk -t1 -c400 —timeout 5s -d 100s -s ‘/usr/local/src/test.lua’ -H “Authorization: ABCDEFGH123OPQ” —latency ‘http://localhost:8080/index.html‘
参数说明
返回结果
每次压测完成后,下一次压测前,建议先检查一下之前的连接是否全部断开,然后再重新压测,命令如下:
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
为什么要使用这个脚本呢?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压测的结果有一些差距,这个问题需要再进一步深入研究一下。