高性能测试工具WRK的高级使用方法

WRK

github地址：https://github.com/wg/wrk
简介：基于事件机制的高性能http压力测试工具，除了能针对单个url进行测试外，最重要的就是能够构造不同的url，不同的参数进行测试。或者发送携带body的POST请求，在有些场景下不要太帅。

说明

wrk高级的地方就是使用了luajit，在不同阶段抛出来几个回调函数，只需要针对这些函数操作即可构造想要请求。典型的回调函数有以下几个

1. function setup(thread) 在创建线程时调用
2. function init(args) 在线程启动时调用
3. function request() 每个请求都会调用，特别说明，此函数调用频繁，所以不要在这里写入耗时的操作，如果复杂的请求，在init函数就构造好，这里直接引用。
4. function delay()
5. function response(status, headers, body) ，频繁调用。

举例

1 发送get 请求

如果想构造不同的请求比如/1?arg=1, /2?arg=2, /3?arg=3……，并且发送GET请求，创建文件为http_get.lua , 代码如下

counter = 0
num = 0
request = function()
   path = "/" .. counter .. "?arg=" .. num
   counter = counter + 1
   num = num + 1
   return wrk.format("", path)
end

执行命令为：

wrk -c 4 -t  4 -d 10s -s http_get.lua  http://cool.test.net/v1.f4v

结果为：

  4 threads and 4 connections
  Thread Stats   Avg      Stdev     Max   +/- Stdev
    Latency   476.52us  306.27us   5.74ms   94.84%
    Req/Sec     2.12k   171.21     2.39k    65.12%
  9085 requests in 1.10s, 1.49MB read
Requests/sec:   8256.69
Transfer/sec:      1.35MB

2 发送POST请求

文件名 http_post.lua

counter = 0
request = function()
   path = "/" .. counter
   counter = counter + 1
   headers = {TH= "httpheader"}
   body = "hello world " .. counter
   return wrk.format("POST", path, headers, body)
end

3 文件中指定uri

如果要测试的url在文件中呢，那么在init阶段把uri都读取到table中，在request中直接引用即可， 具体如下
uri文件为： uri.txt

/1.txt
/2.txt
/3.txt
/4.txt
/5.txt

文件名 http_file_uri.lua，代码

urimap = {}
counter = 0
function init(args)
    for line in io.lines("uri.txt") do
       print(line)
       urimap[counter] = line
       counter = counter + 1
   end
   counter = 0
end

request = function()
   counter = counter + 1
   path = urimap[counter%table.getn(urimap) + 1]
   return wrk.format(nil, path)
end

效果如下：

  Thread Stats   Avg      Stdev     Max   +/- Stdev
    Latency   484.00us  335.83us   8.44ms   95.13%
    Req/Sec     2.14k   192.78     2.59k    71.50%
  85400 requests in 10.00s, 14.00MB read
Requests/sec:   8538.60
Transfer/sec:      1.40MB

服务器端收到请求