17370845950

本文介绍一种可复现、可断言的测试方法，用于在 go 单元测试中精确限制并验证 goroutine 的并发执行数量，避免竞态与资源超限，适用于限流、工作池等场景。

在 Go 单元测试中直接“计数”正在运行的 goroutine 数量（如通过 runtime.NumGo

routine()）既不可靠也不推荐——该值包含运行时维护的系统 goroutine，且无法区分目标逻辑与干扰项。更稳健的做法是：主动控制并发上限，并在受控 mock 行为中实时观测并发状态。

核心思路是：

• 使用带缓冲的 channel 作为并发令牌（limiter := make(chan bool, limit)），实现“最多 limit 个 goroutine 同时执行”；
• 封装待测函数调用逻辑，在进入和退出时显式增减并发计数器；
• 利用 sync.WaitGroup 等待全部完成，并在计数超标时立即标记失败（或 panic）以快速反馈。

以下是一个可直接用于 *_test.go 的完整测试示例：

func TestGoroutineConcurrencyLimit(t *testing.T) {
    const (
        count  = 10
        limit  = 3
    )

    var (
        wg            sync.WaitGroup
        concurrentCnt int
        mu            sync.Mutex
        failed        bool
    )
    wg.Add(count)

    // Mock worker: 模拟实际业务逻辑，但加入并发安全的计数与断言
    mockWorker := func() {
        defer func() {
            mu.Lock()
            concurrentCnt--
            mu.Unlock()
            wg.Done()
        }()

        mu.Lock()
        concurrentCnt++
        if concurrentCnt > limit {
            failed = true // 立即捕获超限，无需等待全部结束
        }
        mu.Unlock()

        time.Sleep(50 * time.Millisecond) // 模拟耗时操作
    }

    // spawn 函数：确保最多 limit 个 goroutine 并发执行
    spawn := func(fn func(), total, maxConcurrent int) {
        limiter := make(chan struct{}, maxConcurrent)
        for i := 0; i < total; i++ {
            limiter <- struct{}{} // 获取令牌
            go func() {
                defer func() { <-limiter }() // 归还令牌
                fn()
            }()
        }
    }

    spawn(mockWorker, count, limit)
    wg.Wait()

    if failed {
        t.Fatalf("concurrency limit %d violated: observed >%d goroutines running simultaneously", limit, limit)
    }
    t.Logf("✅ Passed: exactly %d goroutines ran concurrently (limit=%d)", limit, limit)
}

⚠️ 注意事项：

• 所有对共享变量（如 concurrentCnt, failed）的操作必须加锁（sync.Mutex）或使用原子操作（atomic.Int32），否则测试本身会因数据竞争而不可靠；
• 避免依赖 time.Sleep 做同步——它不稳定且拖慢测试；应优先使用 channel、sync.WaitGroup 或 sync.Once；
• 若被测函数本身已含 goroutine 调度逻辑（如启动 goroutine 池），建议将其抽象为可注入的 func() 参数，便于在测试中替换为可控 mock；
• 生产代码中推荐使用 golang.org/x/sync/semaphore 或 errgroup.Group 替代手写限流逻辑，更健壮且经过充分测试。

通过这种结构化、可观测、可断言的方式，你不仅能验证“是否启动了指定数量的 goroutine”，更能精准保障“任何时候都未超过预期并发上限”，真正实现对并发行为的确定性测试。