用 Go 快速写一个 HTTP 压测工具

一、核心思路

压测工具的本质很简单：

1. 控制并发数（goroutine 池）

2. 控制总量或时长

3. 统计结果（QPS、成功率、延迟）

结构如下：

main
 ├─ 启动 N 个 worker（goroutine）
 ├─ 每个 worker 发 M 个请求
 ├─ 结果写入 channel
 └─ 主线程汇总统计

二、最小可用版（~60 行）

package main

import (
	"fmt"
	"net/http"
	"sync"
	"time"
)

func main() {
	url := "http://127.0.0.1:8080"
	concurrency := 50
	total := 1000

	perWorker := total / concurrency
	results := make(chan time.Duration, total)

	var wg sync.WaitGroup
	start := time.Now()

	for i := 0; i < concurrency; i++ {
		wg.Add(1)
		go func() {
			defer wg.Done()
			client := http.Client{Timeout: 10 * time.Second}
			for j := 0; j < perWorker; j++ {
				t := time.Now()
				resp, err := client.Get(url)
				if err == nil {
					resp.Body.Close()
				}
				results <- time.Since(t)
			}
		}()
	}

	wg.Wait()
	close(results)

	var count int
	var totalLatency time.Duration

	for d := range results {
		count++
		totalLatency += d
	}

	fmt.Printf("请求数: %d\n", count)
	fmt.Printf("QPS: %.2f\n", float64(count)/time.Since(start).Seconds())
	fmt.Printf("平均耗时: %v\n", totalLatency/time.Duration(count))
}

运行：

go run main.go

输出示例：

请求数: 1000
QPS: 4123.56
平均耗时: 12.34ms