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

雨点打透心脏的1/2处 2022-12-18 14:54 415阅读 0赞

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脚本的使用和方法请求的实例，不明白的小伙伴可以留言哦！