2. Go 并发原语基础：goroutine、channel、sync.WaitGroup 的基本使用与性能特性

在 OpenAPI 压测场景中，每一个 HTTP 请求都可以抽象为一个独立的并发任务。Go 语言提供的 goroutine、channel 和 sync.WaitGroup 是构建高性能压测引擎的三大基石。本章从使用语法出发，深入分析三者的性能特性与适用边界，为后续的聚合建模与压测框架设计打下基础。

2.1 goroutine：轻量级并发单元

goroutine 是 Go 运行时管理的用户态线程，初始栈空间仅 2KB~4KB，创建和销毁的开销远低于操作系统线程。在压测场景中，每个虚拟用户（VU）通常对应一个 goroutine。

基本使用

package main

import (
    "fmt"
    "time"
)

func sendRequest(id int) {
    // 模拟 HTTP 请求耗时
    time.Sleep(100 * time.Millisecond)
    fmt.Printf("请求 %d 完成\n", id)
}

func main() {
    for i := 0; i < 10; i++ {
        go sendRequest(i) // 启动 goroutine
    }
    // 等待所有 goroutine 完成（此处仅为演示，实际应使用 WaitGroup）
    time.Sleep(2 * time.Second)
}

2.2 channel：goroutine 间的安全通信

channel 是 Go 语言提供的 CSP（Communicating Sequential Processes）模型实现，用于在 goroutine 之间传递数据。在压测场景中，channel 常用于任务分发、结果收集和流量控制。

基本使用

func worker(id int, jobs <-chan int, results chan<- int) {
    for job := range jobs {
        // 模拟请求处理
        results <- job * 2
    }
}

func main() {
    const numJobs = 100
    jobs := make(chan int, numJobs)
    results := make(chan int, numJobs)

    // 启动 5 个 worker goroutine
    for w := 1; w <= 5; w++ {
        go worker(w, jobs, results)
    }

    // 发送任务
    for j := 1; j <= numJobs; j++ {
        jobs <- j
    }
    close(jobs)

    // 收集结果
    for r := 1; r <= numJobs; r++ {
        <-results
    }
}

channel 类型与性能对比

2.3 sync.WaitGroup：goroutine 同步协调

sync.WaitGroup 用于等待一组 goroutine 全部完成。在压测场景中，它确保所有虚拟用户执行完毕后再进行统计汇总。

基本使用

func main() {
    var wg sync.WaitGroup
    const concurrency = 100

    for i := 0; i < concurrency; i++ {
        wg.Add(1) // 计数器 +1
        go func(id int) {
            defer wg.Done() // 计数器 -1
            // 执行压测请求
            fmt.Printf("VU %d 完成\n", id)
        }(i)
    }

    wg.Wait() // 阻塞直到计数器归零
    fmt.Println("所有 VU 执行完毕")
}

性能特性分析

• 原子操作：Add/Done 内部使用 atomic 包实现，无锁竞争，性能极高。
• 阻塞开销：Wait() 调用会阻塞当前 goroutine，但不会阻塞操作系统线程，GMP 模型会复用 M 执行其他 P 上的 goroutine。
• 复用限制：WaitGroup 在 Wait() 返回后不能直接复用，必须重新声明或通过 Reset() 重置（Go 1.21+ 支持）。

2.4 三原语协同工作流

下图展示了一个典型的压测引擎中，goroutine、channel 和 sync.WaitGroup 如何协同工作：

2.5 性能基准对比

以下基准测试对比了不同并发原语组合在模拟压测场景下的吞吐表现（测试环境：8 核 CPU，Go 1.22，1000 次请求，每次请求模拟 10ms 延迟）：

2.6 常见性能陷阱与最佳实践

• goroutine 泄漏：channel 未关闭导致接收方永久阻塞，或 WaitGroup 计数器未归零。压测中务必使用 defer wg.Done() 和 close(ch) 确保资源释放。
• channel 缓冲区过大：缓冲区过大可能导致内存占用飙升，且掩盖了背压信号。建议缓冲区大小不超过并发数的 2 倍。
• WaitGroup.Add 位置错误：Add 必须在 go 语句之前调用，否则可能出现 Wait 提前返回。推荐模式：wg.Add(1); go func() { defer wg.Done(); ... }()。
• select + default 空转：在压测循环中使用 select 的 default 分支会导致 CPU 空转，应使用带超时的 select 或 time.Ticker 控制频率。