深入解析wrk2的Lua脚本API：自定义你的性能测试

2026-01-14 18:30:19作者：江焘钦

wrk2是一个基于wrk的HTTP性能测试工具，专门设计用于产生恒定吞吐量负载并提供精确的延迟记录。作为wrk2的核心功能，Lua脚本API让开发者能够完全自定义测试逻辑，从请求生成到响应处理都能灵活控制。对于想要深入理解系统性能瓶颈的开发者和运维人员来说，掌握wrk2的Lua脚本API是进行高效性能测试的关键技能。

🎯 wrk2 Lua脚本API基础架构

wrk2提供了四个主要的Lua函数钩子，构成了完整的测试生命周期：

init函数：线程初始化时调用，用于设置测试环境
request函数：生成每个HTTP请求
response函数：处理服务器响应（可选）
done函数：测试结束时汇总结果（可选）

🚀 核心API函数详解

init(args) - 初始化配置

init函数在测试开始时执行，适合进行一次性设置操作。通过这个函数，你可以：

预生成多个请求模板
建立数据库连接池
加载测试数据文件
配置认证信息

request() - 请求生成器

这是最关键的API函数，负责为每个请求返回HTTP消息字符串。wrk2提供了便捷的wrk.format()方法来构建请求。

response(status, headers, body) - 响应处理器

当需要根据服务器响应动态调整测试策略时，可以使用response函数进行：

状态码验证和统计
响应内容解析
动态修改后续请求
错误处理和重试逻辑

done(summary, latency, requests) - 结果汇总

测试结束时调用，用于自定义结果展示和数据分析：

function done(summary, latency, requests)
   -- 自定义统计逻辑
end

📊 实战示例：高级测试场景

负载均衡测试

使用scripts/addr.lua脚本，你可以为每个线程分配不同的服务器地址，模拟真实的负载均衡环境：

function setup(thread)
   thread.addr = addrs[math.random(#addrs)]
end

HTTP流水线测试

scripts/pipeline.lua展示了如何实现HTTP流水线测试：

init = function(args)
   local r = {}
   r[1] = wrk.format(nil, "/?foo")
   r[2] = wrk.format(nil, "/?bar")
   r[3] = wrk.format(nil, "/?baz")
   req = table.concat(r)
end

🎨 延迟统计与性能分析

wrk2使用HdrHistogram进行精确的延迟记录，提供了丰富的统计接口：

latency.min - 最小延迟值
latency.max - 最大延迟值
latency.mean - 平均延迟值
latency:percentile(99.0) - 99%百分位延迟
latency[i] - 原始样本值

上图清晰地展示了wrk2在不同请求速率和Coordinated Omission机制下的延迟分布差异。启用CO机制后，高百分位延迟显著降低，特别是在16K RPS的高负载场景下效果尤为明显。

🔧 自定义报告生成

通过done函数，你可以创建完全自定义的测试报告：

function done(summary, latency, requests)
   io.write("========================\n")
   io.write("性能测试报告\n")
   io.write("========================\n")
   io.write(string.format("总请求数: %d\n", summary.requests))
   io.write(string.format("测试时长: %.2f秒\n", summary.duration / 1000000))
   io.write(string.format("平均延迟: %.2fms\n", latency.mean / 1000))
end