Hey 源码结构全景图HTTP 压测工具 hey.go 与 requester 包的完整架构导览【免费下载链接】heyHTTP load generator, ApacheBench (ab) replacement项目地址: https://gitcode.com/GitHub_Trending/he/heyhey 是一款用 Go 语言编写的轻量级 HTTP 负载生成器load generator常被视为 ApacheBenchab的现代替代品。整个项目只有两个入口文件加一个 requester/ 包代码量极小却是学习并发压测工具如何设计的绝佳样本。本文带你从 hey.go 走到 requester/requester.go看清每一个模块的分工。一、仓库文件地图先看全景再拆细节整个仓库非常扁平核心就 8 个文件文件职责hey.go程序入口命令行参数解析、请求构造、启动压测hey_test.go入口层的单元测试requester/requester.go压测引擎核心Work结构体与 worker 并发调度requester/report.go结果聚合延迟统计、百分位、直方图requester/print.go输出渲染Summary / CSV 两种模板requester/now_other.go非 Windows 平台的高精度计时requester/now_windows.goWindows 平台计时QueryPerformanceCounterrequester/requester_test.gorequester 包的测试周边工程文件还有 go.mod仅依赖golang.org/x/net一个模块、Makefile跨平台构建 release 产物和 Dockerfile。这种入口 单包的结构让 hey 可以编译成零依赖的单个二进制文件。二、入口层hey.go 如何把命令行变成压测任务hey.go 约 290 行承担了全部胶水工作逻辑可以拆成四步 1. 命令行参数定义文件开头用 Go 标准库flag声明了所有参数hey.go#L41-L67-n请求总数默认 200、-c并发 worker 数默认 50-q每 worker 的 QPS 限流、-z压测持续时长-m/-H/-d/-D控制方法、请求头、请求体-h2开启 HTTP/2-cpus控制 CPU 核数2. 参数校验与请求构造main()函数hey.go#L106-L253先做合法性校验如-n不能小于-c随后把参数翻译成标准库的http.Request正则解析自定义头、Basic Auth拼装 User-Agent自动带上hey/0.0.1标识。3. 装配 Work 并启动关键在 hey.go#L224-L252把所有配置塞进requester.Work结构体调用w.Init()和w.Run()。同时它注册了两个停止信号——CtrlC通过os/signal捕获中断信号触发w.Stop()-z定时停止后台 goroutine 睡满指定时长后自动调用w.Stop()也就是说hey.go 自己不执行任何 HTTP 请求它只负责配置 启停控制真正的引擎在 requester 包里。三、引擎层requester 包的并发模型requester 包是 hey 的心脏四个文件各司其职。Work 结构体一次压测的全部状态requester/requester.go 中的Workrequester.go#L49-L100持有压测配置N/C/QPS/Timeout/H2 等以及两个关键通道results chan *result结果管道缓冲上限 100 万条stopCh chan struct{}优雅停机信号生命周期只有四个方法Init()→Run()→外部可触发Stop()→Finish()。runWorkersC 个并发 worker 怎么跑requester.go#L235-L264 的runWorkers()展示了 hey 最核心的并发设计构建一个共享的http.Transport——MaxIdleConnsPerHost取并发数 C上限 500让 TCP 连接在 worker 间复用若开启-h2用http2.ConfigureTransport升级传输层启动 C 个 goroutine每个执行runWorker(client, N/C)用sync.WaitGroup等待全部完成。每个 worker 内部requester.go#L208-L233用time.Ticker实现 QPS 限流并在每轮循环开头检查stopCh保证收到停止信号后能立刻退出。makeRequest一次请求的分段计时requester.go#L146-L206 是理解 hey 统计精度的关键。它通过httptrace.ClientTrace钩住请求生命周期的 6 个时间点把一次请求拆成独立耗时阶段含义DNSdialup建立 TCP 连接总耗时DNS纯 DNS 解析耗时req write写出请求耗时resp wait首字节延迟TTFBresp read读取完整响应耗时最终打包成一个resultrequester.go#L36-L47投递到results通道。这套生产者worker→ 管道channel→ 消费者reporter模型让压测与统计互不阻塞。四、统计层从原始结果到 Summary 报告report.go聚合器requester/report.go 中的runReporterreport.go#L84-L114在独立 goroutine 里循环消费results通道累计各阶段耗时之和、状态码分布、错误分布。snapshot()report.go#L143-L215在压测结束时对延迟切片排序产出响应时间直方图等宽分成 10 个桶histogram()report.go#L237-L268延迟分布P10 / P25 / P50 / P75 / P90 / P95 / P99 七个百分位latencies()report.go#L217-L235注意maxRes 1000000report.go#L31明细数据最多保留 100 万条超大压测下靠均值/直方图兜底防止内存爆炸。print.go两种输出模板requester/print.go 用 Go 的text/template定义了两套渲染模板print.go#L96-L127默认 Summary总量、最快/最慢/平均耗时、RPS、直方图、百分位、各阶段明细、状态码与错误分布CSV-o csv每个请求一行8 列指标方便导入表格工具二次分析文件头部的注释print.go#L15-L34对两种格式的字段解释得非常清楚读代码时可直接当作文档用。now_*.go跨平台高精度计时压测工具对计时精度极其敏感因此now()做了平台拆分now_other.go其他平台直接用time.Since以包级startTime为零点now_windows.goWindows 下调用QueryPerformanceCounter系统调用绕开time.Now()的微秒级精度瓶颈这是 Go 构建标签build tag做平台差异化的典型用法。五、一次压测的完整数据流把各模块串起来hey 的执行链路是hey.go 解析参数 → 构造http.Request→ 填充requester.WorkWork.Run()启动 reporter goroutine C 个 worker goroutinerequester.go#L119-L129每个 worker 循环调用makeRequest带分段计时结果写入results通道reporter 实时消费、累计worker 全部Done后Finish()关闭通道reporter 排序、分桶、算百分位finalize()触发模板渲染输出 Summary期间 CtrlC 或-z到期 →Stop()向stopCh发 C 个信号 → workers 优雅退出六、这套架构值得新手借鉴的 4 个设计点入口与引擎解耦hey.go 不含任何 HTTP 逻辑requester 包可以脱离命令行独立使用这也是 requester_test.go 能直接测引擎的原因channel 解耦生产/消费worker 只管发请求reporter 只管统计二者互不等待压测吞吐不受统计开销影响优雅停机stopCh信号 每轮循环检查避免发了 99% 的请求还停不下来模板化输出统计逻辑report与展示逻辑print 模板分离加一种输出格式只需加一个模板串七、如何上手阅读这份代码 建议的阅读顺序与 README.md 中的示例配合先在终端跑一遍hey -n 100 -c 10 https://example.com直观感受输出格式从 hey.go 的main()顺藤摸瓜到Work.Run()精读 requester/requester.go 的makeRequest与runWorkers再看 report.go 的snapshot()理解直方图与百分位是怎么算出来的如果想从源码构建clone 仓库后执行make release即可生成三大平台的二进制见 Makefile日常开发只需go build。hey 用不到 1500 行核心代码实现了生产级压测工具的完整能力——入口解析、并发引擎、分段计时、百分位统计、优雅停机——是学习 Go 并发与压测原理不可多得的精简范本。【免费下载链接】heyHTTP load generator, ApacheBench (ab) replacement项目地址: https://gitcode.com/GitHub_Trending/he/hey创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考