Golang并行基准测试怎么做 使用RunParallel方法实践

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

Golang并行基准测试怎么做 使用RunParallel方法实践

在 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

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

是否真的更优。关键在于模拟真实并发场景,而不是只测单线程性能。

© 版权声明
THE END
喜欢就支持一下吧
点赞8 分享