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 完成后才结束计时。
立即学习“go语言免费学习笔记(深入)”;
它的基本用法如下:
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
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
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
是否真的更优。关键在于模拟真实并发场景,而不是只测单线程性能。