wrk 使用lua脚本进行压测接口请求

lua脚本是一种轻量小巧的脚本语言，用标准c语言编写，并以源代码形式开放，其设计目的是为了嵌入应用程序中，从而为程序提供灵活的扩展和定制功能。wrk工具嵌入了lua脚本语言，因此，在自定义压测场景时，可在wrk目录下使用lua定制压测场景。

一、lua声明周期

　共有三个阶段，启动阶段，运行阶段，结束阶段。wrk支持在这三个阶段对压测进行个性化。　　　

1、启动阶段

function setup(thread)

在脚本文件中实现setup方法，wrk就会在测试线程已经初始化但还没有启动的时候调用该方法。wrk会为每一个测试线程调用一次setup方法，并传入代表测试线程的对象thread作为参数。setup方法中可操作该thread对象，获取信息、存储信息、甚至关闭该线程。

thread.addr - get or set the thread's server address
thread:get(name) - get the value of a global in the thread's env
thread:set(name, value) - set the value of a global in the thread's env
thread:stop() - stop the thread

2、运行阶段

function init(args)  --由测试线程调用，只会在进入运行阶段时，调用一次。支持从启动wrk的命令中，获取命令行参数；
function delay()  --在每次发送request之前调用，如果需要delay，那么delay相应时间；
function request()  --用来生成请求；每一次请求都会调用该方法，所以注意不要在该方法中做耗时的操作；
function response(status, headers, body)  --在每次收到一个响应时调用；为提升性能，如果没有定义该方法，那么wrk不会解析headers和body；

3、结束阶段

function done(summary, latency, requests)  --在整个测试过程中只会调用一次，可从参数给定的对象中，获取压测结果，生成定制化的测试报告。

二、自定义脚本中可访问的变量和方法：

　　1、变量：wrk

wrk = {
    scheme  = "http",
    host    = "localhost",
    port    = nil,
    method  = "GET",
    path    = "/",
    headers = {},
    body    = nil,
    thread  = <userdata>,
  }

2、方法：wrk.fomat wrk.lookup wrk.connect             

function wrk.format(method, path, headers, body)  --根据参数和全局变量wrk，生成一个HTTP rquest string。
function wrk.lookup(host, service)  --给定host和service（port/well known service name），返回所有可用的服务器地址信息。
function wrk.connect(addr)  --测试与给定的服务器地址信息是否可以成功创建连接

三、lua脚本压测实例

       压测命令，如：wrk -t8 -c200 -d30s --latency  -s test.lua  http://www.nandao.com

　　test.lua是用lua写的压测脚本，如下是压测脚本的实例：

　　1、使用post方法压测

wrk.method = "POST"
wrk.headers["S-COOKIE2"]="a=2&b=Input&c=10.0&d=24***"
wrk.body = "recent_seven=20191127_32;20191128_111"
wrk.headers["Host"]="api.i.**.com"
function response(status,headers,body)
        if status ~= 200 then --将服务器返回状态码不是200的请求结果打印出来
                print(body)
        --      wrk.thread:stop()
        end
end

2、wrk读取文件，实现随机header-cookie 或其他参数

idArr = {}
falg = 0
wrk.method = "POST"
wrk.body = "a=1"
function init(args)
        for line in io.lines("integral/cookies.txt") do
                print(line)
                idArr[falg] = line
                falg = falg+1
        end
        falg = 0
end

--wrk.method = "POST"
--wrk.body = "a=1"
--wrk.path = "/v1/points/reading"

request = function()
        parms = idArr[math.random(0,4)] --随机传递文件中的参数
        --parms = idArr[falg%(table.getn(idArr)+1)] 循环传递文件中的参数
        wrk.headers["S-COOKIE2"] = parms
        falg = falg+1
        return wrk.format()
end

3、wrk 创建数组并初始化，拼接随机参数，进行get 请求，文件名：startget.lua，命令如下：wrk -t8 -c200 -d30s  -s startget.lua --latency  -s test.lua http://www.nandao.com

idArr = {};
function init(args)
        idArr[1] = "a";
        idArr[2] = "b";
        idArr[3] = "c";
        idArr[4] = "d";
end

request = function()
        parms = idArr[math.random(1,4)]
        path = "/test/points/reading?id="..parms
        return wrk.format("GET",path)
end

这就是wrk  压测过程中，lua脚本的使用和方法请求的实例，不明白的小伙伴可以留言哦！