wrk(压测工具)源码笔记

最新推荐文章于 2024-08-01 14:25:42 发布

带你环游world

最新推荐文章于 2024-08-01 14:25:42 发布

阅读量5.4k

点赞数 1

分类专栏： linux 文章标签： wrk源码 ae

本文链接：https://blog.csdn.net/codingwithme/article/details/52251451

版权

本文详细探讨了wrk压力测试工具的源码，重点关注了Redis的AE事件循环机制，包括AE_FILE_EVENTS和AE_TIME_EVENT，并解释了AE_EVENT_LOOP的实现原理。此外，还分析了wrk的统计数据计算和Socket连接处理，指出wrk在多线程环境下如何确保线程安全。通过对wrk的深入理解，有助于提升网络编程和性能测试的技能。

摘要由CSDN通过智能技术生成

wrk是一款比较常用的压测工具，支持多核，多线程，同时支持Lua脚本对返回结果和统计数据进行定制。详见github: https://github.com/wg/wrk

wrk源码部分结合了很多优秀的github开源项目，例如Redis数据库中的事件循环ae，用于nginx、nodejs、joyent的解析http请求相应的模块http-parser，以及Mike Pall所写的及时lua脚本编译器LuaJIT–lua Just in Time。这次我着重追踪了下wrk代码本身、以及ae时间循环的部分，至于http-parser以及luaJIT由于不涉及主要逻辑，暂时没有深究。抛砖引玉，一起探讨哈~

AE(Redis EventLoop)

ae是Redis内部使用的事件循环，由于我没有仔细研读过Redis的源码，所以对ae本身并不熟悉。只是ae的实现，和接触过的node的实现类似(Event-Driven Design)，结合之前实践时对Redis的零星记忆(例如Redis单线程、命令串式执行等)，大概明白其在wrk中的作用。
ae的event-loop监听的事件分为两种，一种是AE_FILE_EVENTS，另一种是AE_TIME_EVENTS。

AE_FILE_EVENTS

AE_FILE_EVENTS在Linux下是利用epoll相关机制实现的。例如：

aeCreateEventLoop(); // 调用epoll_create
aeCreateFileEvent(); // 调用epoll_ctl
aeProcessEvents(); // 调用epoll_wait

以wrk的调用为例:
main()->(每个thread)aeCreateEventLoop()