Golang并行测试实践：RunParallel方法详解

RunParallel是Go中用于并行基准测试的核心方法，它通过启动多个goroutine并利用sync.WaitGroup同步，使测试能真实模拟高并发场景下的性能表现；其使用pb.Next()控制迭代，确保总执行次数为b.N且由多个goroutine分摊，默认并发数为GOMAXPROCS，可通过SetParallelism调整；在对比sync.Map与加锁map的性能时，测试显示在读写混合场景下后者可能更优，说明RunParallel能有效揭示不同并发策略的实际开销；使用时需保证并发安全、避免共享状态污染、合理设计读写比例，并结合-race检测数据竞争，从而准确评估并发组件性能。

在 Go 语言中进行并行基准测试是评估代码并发性能的关键手段。testing.B 提供了 RunParallel 方法，专门用于模拟高并发场景下的性能表现，尤其适合测试并发安全的数据结构、缓存、数据库连接池等。

下面详细介绍如何使用 RunParallel 进行有效的并行基准测试，并结合实际示例说明。

什么是 RunParallel？

RunParallel 是 *testing.B 的一个方法，它会启动多个 goroutine 并并行执行测试逻辑。它内部使用 sync.WaitGroup 控制并发，自动协调多个 goroutine 的执行，并确保在所有 goroutine 完成后才结束计时。

它的基本用法如下：

func BenchmarkXXX(b *testing.B) {
    b.RunParallel(func(pb *testing.PB) {
        for pb.Next() {
            // 被测代码
        }
    })
}

• pb.Next() 返回 true 直到迭代完成，类似 for i := 0; i < b.N; i++。
• 每个 goroutine 都会独立调用 Next()，因此总执行次数为 b.N，但由多个 goroutine 分摊。
• 默认情况下，RunParallel 使用 GOMAXPROCS 个 goroutine（通常是 CPU 核心数）。

实际示例：并发访问 map 的性能测试

Go 的原生 map 不是并发安全的，我们通常使用 sync.Map 或加锁的 map + sync.RWMutex。下面我们对比两者在并发读写场景下的性能。

1. 使用 sync.Map

func BenchmarkSyncMap(b *testing.B) {
    var m sync.Map

    b.RunParallel(func(pb *testing.PB) {
        for pb.Next() {
            key := rand.Intn(1000)
            m.Store(key, key)
            m.Load(key)
        }
    })
}

2. 使用 map + sync.RWMutex

func BenchmarkMutexMap(b *testing.B) {
    var mu sync.RWMutex
    m := make(map[int]int)

    b.RunParallel(func(pb *testing.PB) {
        for pb.Next() {
            key := rand.Intn(1000)
            mu.Lock()
            m[key] = key
            mu.Unlock()

            mu.RLock()
            _ = m[key]
            mu.RUnlock()
        }
    })
}

运行基准测试：

go test -bench=Benchmark -run=^$ -count=3

输出可能类似：

BenchmarkSyncMap-8        1000000           1200 ns/op
BenchmarkMutexMap-8       2000000            800 ns/op

可以看到，在这个简单读写混合场景中，加锁的 map 反而比 sync.Map 更快。这是因为 sync.Map 在高竞争或频繁写入时开销较大，而 RunParallel 帮助我们真实还原了并发竞争场景。

如何控制并发度？

默认使用 GOMAXPROCS 个 goroutine，但你可以通过设置 GOMAXPROCS 或使用环境变量调整。例如：

GOMAXPROCS=4 go test -bench=BenchmarkSyncMap

如果你想手动控制 goroutine 数量，可以结合 b.SetParallelism()（Go 1.18+）：

b.SetParallelism(2) // 使用 2 * GOMAXPROCS 个 goroutine
b.RunParallel(...)

比如 SetParallelism(1) 是默认行为，SetParallelism(4) 会启动更多 goroutine，适合模拟更高并发。

注意事项和最佳实践

• ✅ 确保测试逻辑是并发安全的：如果你测的是非并发安全结构（如普通 map），必须加锁，否则会触发 data race。
• ✅ 避免共享状态污染：每个 goroutine 应尽量独立，避免相互影响。例如，可以按 goroutine 分配不同 key 范围。
• ✅ 使用 pb.Next() 控制迭代：不要在 RunParallel 中直接用 for i := 0; i < b.N; i++，否则每个 goroutine 都会跑 b.N 次，总次数爆炸。
• ✅ 结合 -race 检测数据竞争：

go test -bench=BenchmarkSyncMap -run=^$ -race

• ✅ 合理设计测试场景：比如读多写少、写多读少，可以调整 Store 和 Load 的比例来模拟真实业务。

小技巧：模拟读写比例

比如 90% 读，10% 写：

b.RunParallel(func(pb *testing.PB) {
    for pb.Next() {
        key := rand.Intn(1000)
        if rand.Float32() < 0.9 {
            m.Load(key)
        } else {
            m.Store(key, key)
        }
    }
})

这样更贴近缓存类场景。

基本上就这些。RunParallel 是 Go 并行基准测试的核心工具，用好它能帮你发现并发瓶颈、验证锁策略、评估 sync.Map 是否真的更优。关键在于模拟真实并发场景，而不是只测单线程性能。