Hey 代码艺术不到 1000 行 Go 代码如何设计一个生产级 HTTP 压测 CLI 工具【免费下载链接】heyHTTP load generator, ApacheBench (ab) replacement项目地址: https://gitcode.com/GitHub_Trending/he/heyhey是一款轻量级 HTTP 压测工具HTTP load generator常被用作 ApacheBenchab的现代替代品。它的全部核心逻辑不到 1000 行 Go 代码却能完成并发压测、QPS 限速、分阶段延迟分析、直方图统计等生产级功能——是学习「如何用小体量代码写出可靠 CLI 工具」的绝佳范本。一、项目全景5 个文件撑起一个压测工具先建立整体认知。整个仓库只有少量源码文件职责切分得非常干净模块文件职责入口层hey.go约 290 行解析命令行参数组装配置启动压测执行层requester/requester.goWorker 协程池、请求发送、结果收集统计层requester/report.go汇总指标、分位数、直方图展示层requester/print.go模板渲染 Summary / CSV 报告适配层requester/now_other.go、requester/now_windows.go跨平台时间基准差异可以看到一个经典的三层架构参数 → 执行 → 报告。想读懂任何 CLI 工具源码先找这条主线hey 就是最好的练习场。二、设计一参数到结构体的「单一配置对象」模式在 hey.go 中所有命令行参数-n请求数、-c并发数、-qQPS 限速、-z压测时长等全部先用标准库flag声明再在main函数里一次性收敛成一个Work结构体见 hey.goWork{ Request, N, C, QPS, Timeout, H2, ProxyAddr, Output, ... }这样做的好处是入口层只负责翻译业务层只依赖一个结构体。Work可以被随意构造和注入测试时不需要模拟命令行——requester/requester_test.go 里所有测试都是直接Work{N: 20, C: 2}起服务开跑代码极其简洁。还有一个小巧思重复出现的-H请求头通过自定义headerSlice类型实现flag.Value接口hey.go让flag包原生支持「同一参数写多次」无需引入任何第三方 CLI 库。三、设计二Channel 生产者-消费者天然背压并发压测最容易踩的坑是「协程池 全局切片」带来的锁竞争。hey 的选择是 Go 最地道的方案每个 Worker 每完成一个请求把结果*result投递到带缓冲的resultschannelrequester/requester.go独立 Reporter 协程for range轮询该 channel边收边累加统计requester/report.go缓冲大小取min(并发数×1000, 100万)用容量控制背压防止慢消费拖垮生产端requester/requester.go。生产者Worker与消费者Reporter完全解耦Worker 不需要关心统计逻辑Reporter 也不需要加任何锁。没有 mutex 的并发程序天然少了一大类 bug。四、设计三httptrace 把一次请求拆成 5 段普通压测工具只告诉你「平均 120ms」。而 hey 借助标准库httptrace.ClientTrace的回调requester/requester.go把每次请求拆成五个可独立测量的阶段DNS 解析DNS-lookupTCP/TLS 建连DNSdialup请求写出req write等待首字节resp wait响应读取resp read最终报告中每个阶段都有平均/最快/最慢三档统计requester/report.go。这意味着当用户抱怨「接口慢」时能直接定位是 DNS 慢、建连慢还是服务端慢——这是它相比 ab 最有价值的差异化能力。五、设计四优雅停机的两条路径 压测工具必须随时能停。hey 在 hey.go 里用两个 goroutine 覆盖了两种停止场景用户按 CtrlCsignal.Notify监听os.Interrupt触发w.Stop()达到-z指定时长time.Sleep到点后自动Stop()。Stop()向stopCh发送C个停止信号Worker 在每轮循环开头用select非阻塞检查停止位requester/requester.go实现「跑完当前请求即退出」的平滑收尾最后Finish()关闭 results channel 并等待 Reporter 完成统计。停止不是中断而是让所有组件有序地走完收尾流程——这是生产级与玩具级工具的分水岭。六、设计五模板驱动输出一份数据两种报告报告渲染放在 requester/print.goSummary 报告和 CSV 导出都是text/template模板字符串统计好的Report快照直接喂给模板执行。内置模板里甚至实现了纯文本直方图——用■字符按桶频率拉出条形图requester/print.go在终端里直观呈现延迟分布。要新增 JSON 输出加一个模板即可统计层零改动。这正是「展示层独立」的红利。七、设计六零第三方依赖 构建标签的跨平台翻一下 go.mod核心仅依赖golang.org/x/net为 HTTP/2 支持没有 gin、没有 cobra、没有任何 CLI 框架。跨平台差异则由构建标签解决now_other.go用time.Sincenow_windows.go单独处理 Windows 计时基准同一now()函数在不同系统编译出不同实现。发布环节同样精简Makefile 三行交叉编译产出 Linux/macOS/Windows 二进制配 Dockerfile 供容器场景使用。八、上手体验安装与常用命令 想亲手感受这份代码的艺术克隆仓库后交叉编译即可git clone https://gitcode.com/GitHub_Trending/he/hey make release常用命令一览详见 README.md# 默认 200 请求 / 50 并发 hey https://your-api.example.com # 1000 请求、100 并发 hey -n 1000 -c 100 https://your-api.example.com # 持续压测 30 秒 hey -z 30s -c 50 https://your-api.example.com # 每 Worker 限速 10 QPS hey -q 10 -c 5 -z 30s https://your-api.example.com # 导出 CSV 供后续分析 hey -o csv -n 500 https://your-api.example.com跑完后你会看到总耗时、Requests/sec、延迟直方图、P10~P99 分位分布以及前文提到的五段式延迟明细——全部来自那不到 1000 行代码。九、总结从 hey 身上提炼的 5 条 CLI 设计原则把这篇文章浓缩成可复用的清单写你自己的 Go CLI 工具时可以直接套用入口只翻译不写业务参数统一收敛进一个配置结构体业务层对结构体编程hey.go用 channel 代替锁生产-消费解耦Worker 无锁、可水平扩展requester/requester.go测量要拆细利用httptrace等钩子做分阶段计时报告才有诊断价值停止要优雅信号 时长双触发每轮循环检查停止位收尾再出报告输出走模板一份数据快照喂多个模板终端报告与 CSV 导出共享同一数据源requester/print.go。代码少不等于简陋。hey 用不到 1000 行代码证明清晰的职责分层、地道的并发原语和克制的依赖选择就是最强大的生产级设计。下次写压测工具前不妨先通读一遍 requester/ 目录——这大概是你今年能读到最短、却最值得精读的 Go 源码之一。【免费下载链接】heyHTTP load generator, ApacheBench (ab) replacement项目地址: https://gitcode.com/GitHub_Trending/he/hey创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考