How to pass arguments to the LUA script

[suni1mp]

How to pass argument to LUA script

./wrk -d 20 -p 5 -s ./luascripts/wrkscript_get.lua arg1,arg2 -R 4 http://URL

[steers]

As per the scripting API documentation, the init function takes a parameter args, an array whose values are populated by arguments provided to the wrk binary after --, e.g.

$ wrk -t1 -c1 -d10 -R10 -s /tmp/args.lua http://localhost:1337 -- arg1 arg2

Try experimenting using an init function like the following:

function init(args) 
     for i, v in ipairs(args) do
         print(i,v)
     end 
 end 

Which, when invoked as listed above, should output the following:

1       arg1
2       arg2

You can also use environment variables, e.g.

$ env foo=1 bar=2 wrk -t1 -c1 -d10 -R10 -s /tmp/args.lua http://localhost:1337

Which can be accessed in Lua using os.getenv(), e.g.

local environment =  {
   foo = tonumber(os.getenv("foo")),
   bar = tonumber(os.getenv("bar")),
}