Golang如何测试多协程执行结果-Golang-PHP中文网

Golang如何测试多协程执行结果

P粉602998670

发布： 2025-10-12 09:03:01

原创

744人浏览过

使用sync.WaitGroup和channel可有效测试Go多协程，确保协程完成后再验证结果，结合锁或通道避免数据竞争，并通过go test -race检测竞态条件，保证并发安全。

golang如何测试多协程执行结果

在Go语言中，测试多协程执行结果的关键是确保并发逻辑正确、数据竞争可控，并能准确验证最终状态。直接启动多个goroutine后如果不加同步，测试函数可能会在协程完成前就结束。以下是几种常用且有效的测试方法。

使用 sync.WaitGroup 等待所有协程完成

当需要并发执行多个任务并等待它们全部完成时，sync.WaitGroup 是最常用的同步工具。

在测试中，你可以为每个协程调用 Add(1)，并在每个协程结束时调用 Done()，主协程通过 Wait() 阻塞直到所有任务完成。

示例代码：

func TestMultipleGoroutines(t *testing.T) {
    var wg sync.WaitGroup
    results := make([]int, 10)
    mu := sync.Mutex{} // 保护切片写入
<pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">for i := 0; i < 10; i++ {
    wg.Add(1)
    go func(idx int) {
        defer wg.Done()
        // 模拟一些工作
        time.Sleep(time.Millisecond * 10)
        mu.Lock()
        results[idx] = idx * 2
        mu.Unlock()
    }(i)
}

wg.Wait() // 等待所有协程完成

// 验证结果
for i := 0; i < 10; i++ {
    if results[i] != i*2 {
        t.Errorf("Expected %d, got %d", i*2, results[i])
    }
}

登录后复制

}

使用 channel 接收协程结果

channel 是 Go 中协程通信的推荐方式。你可以让每个协程将结果发送到一个 channel，主协程接收所有结果并进行验证。

立即学习“go语言免费学习笔记（深入）”；

这种方式更符合 Go 的“通过通信共享内存”理念，也能避免显式使用锁。

示例代码：

func TestGoroutinesWithChannel(t *testing.T) {
    resultCh := make(chan int, 10)
<pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">for i := 0; i < 10; i++ {
    go func(idx int) {
        // 模拟处理
        time.Sleep(time.Millisecond * 5)
        resultCh <- idx * 2
    }(i)
}

// 收集所有结果
results := make([]int, 0, 10)
for i := 0; i < 10; i++ {
    results = append(results, <-resultCh)
}

// 排序以方便验证（因为并发顺序不确定）
sort.Ints(results)
expected := []int{0, 2, 4, 6, 8, 10, 12, 14, 16, 18}
for i, v := range results {
    if v != expected[i] {
        t.Errorf("Index %d: expected %d, got %d", i, expected[i], v)
    }
}

登录后复制

}