WRK 的源码详解网上已经有一些了，但是都不是特别详细，最近有时间特意详读了一下，希望能给大家一些参考以及启发
WRK 介绍
wrk 是一个多线程，内部通过事件驱动模型 +IO 多路复用处理网络连接的一个针对 HTTP/HTTPS 的压测工具（不支持 windows 系统）
WRK 代码结构
wrk 源码在 src 目录下，其中比较重要的文件有 wrk，stats，net，http_parser
wrk 想阅读明白要先理解 AE，在第一个文章中，我会先介绍一下：AE
AE 介绍（Redis EventLoop）
AE 是 Redis 内置的一个高性能事件驱动器，主要分为文件事件和定时器事件
但也是由于 AE 这套规则只能在符合 POSIX 标准的系统中工作，所以 wrk 不能在 windows 系统中运行
API
创建/删除驱动器：
aeEventLoop *aeCreateEventLoop(int setsize);
void aeDeleteEventLoop(aeEventLoop *eventLoop);
setsize 参数表示了 eventloop 可以监听的网络事件 fd 个数，一般这个数字是最大并发数 + 一些预留空间
创建/删除文件 IO 事件：当套接字变得可读/可写时，执行
int aeCreateFileEvent(aeEventLoop *eventLoop, int fd, int mask, aeFileProc *proc, void *clientData);
void aeDeleteFileEvent(aeEventLoop *eventLoop, int fd, int mask);
eventLoop 为上面创建的驱动器，fd 为监听的文件句柄，注册事件类型分别有：AE_READABLE，AE_WRITABLE ，AE_BARRIER，mask 是这几个事件经过或运算后的掩码（例如如果想监听读写事件，则 mask=AE_READABLE|AE_WRITABLE，如果只想监听写事件，则 mask=AE_WRITABLE），proc 为执行的回调函数，clientData 为传给回调参数的值
创建/删除定时器事件：让一段程序在指定的时间之后执行一次。
long aeCreateTimeEvent(aeEventLoop *eventLoop, long long milliseconds, aeTimeProc *proc, void *clientData, aeEventFinalizerProc *finalizerProc);
int aeDeleteTimeEvent(aeEventLoop *eventLoop, long long id);
milliseconds 为超时时间，proc 为超时后的回调函数，finalizerProc 为定时器删除后执行的回调函数
处理事件
void aeMain(aeEventLoop *eventLoop);
// 事件循环
一般使用方式是创建驱动器，再根据需要添加文件 IO 事件或定时器事件，通过 aeMain 事件循环
关于 AE 中的 IO 复用
FeeBSD 或 Apple 使用的是 kqueue，linux 使用的是 epoll，sunos 使用的是 evport，则默认使用 select，从效率上来看是：evport > epoll > kqueue > select
源码中是这么标注的
#if defined(__FreeBSD__) || defined(__APPLE__)
#define HAVE_KQUEUE
#elif defined(__linux__)
#define HAVE_EPOLL
#elif defined (__sun)
#define HAVE_EVPORT
在 WRK 中是如何使用的
请求处理：
每个线程创建了一个独立的 aeEventLoop，线程内创建若干 connection（socket），并为每个 socket 句柄注册事件（aeCreateFileEvent），包括：AE_READABLE，AE_WRITABLE
关键代码
flags = AE_READABLE | AE_WRITABLE;
aeCreateFileEvent(loop, fd, flags, socket_connected, c);
统计结果：
每个线程注册定时器（aeCreateTimeEvent），每 100ms 执行一次统计请求和响应结果
关键代码
aeCreateTimeEvent(loop, RECORD_INTERVAL_MS, record_rate, thread, NULL);
第二章可跳转：WRK 源码详解 (第二章：处理连接)
