Go语言中通过testing包的b.RunParallel方法可对并发函数进行基准测试，评估高并发下的性能。示例对比原子操作与互斥锁的并发递增操作，结果显示原子操作因无阻塞特性通常性能更优，合理使用RunParallel有助于真实模拟并发场景并优化技术选型。
在Go语言中，对并发函数进行基准测试可以帮助我们了解程序在高并发场景下的性能表现。通过testing包中的Benchmark函数，我们可以轻松地编写并行测试用例，评估并发函数的吞吐量、响应时间和资源消耗。
使用Benchmark进行并行测试
Go的testing包提供了b.RunParallel方法，专门用于模拟高并发场景。它会启动多个goroutine，并将迭代分配给这些goroutine并行执行，从而更真实地反映并发环境下的性能。
示例：测试一个并发安全的计数器
package main
import (
"sync"
"sync/atomic"
"testing"
)
var counter int64
func increment() {
atomic.AddInt64(&counter, 1)
}
func incrementWithMutex(mu *sync.Mutex) {
mu.Lock()
counter++
mu.Unlock()
}
并发函数的基准测试示例
下面是对原子操作和互斥锁两种方式的并发递增操作进行基准测试的完整示例：
立即学习“go语言免费学习笔记（深入）”；
```go
func BenchmarkAtomicInc(b *testing.B) {
counter = 0
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
increment()
}
})
// 验证结果正确性（可选）
if counter != int64(b.N) {
b.Fatalf("expected %d, got %d", b.N, counter)
}
}
func BenchmarkMutexInc(b *testing.B) {
var mu sync.Mutex
counter = 0
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
incrementWithMutex(&mu)
}
})
if counter != int64(b.N) {
b.Fatalf("expected %d, got %d", b.N, counter)
}
}说明：
-
b.RunParallel自动创建多个goroutine（默认GOMAXPROCS个）来运行循环。 -
pb.Next()控制每个goroutine的迭代次数，总迭代数为b.N。 - 适合测试并发安全的操作，如原子操作、互斥锁、channel通信等。
自定义并发度的测试方法
有时需要指定特定的并发级别，可以通过手动启动goroutine实现：
func BenchmarkCustomWorkers(b *testing.B) {
counter = 0
const workers = 10
var wg sync.WaitGroup
var mu sync.Mutex
b.ResetTimer()
for i := 0; i < b.N; i++ {
wg.Add(workers)
for w := 0; w < workers; w++ {
go func() {
mu.Lock()
counter++
mu.Unlock()
wg.Done()
}()
}
wg.Wait()
}
}注意：这种写法会在每次迭代中等待所有worker完成，实际压测时需根据需求调整逻辑。
运行与分析结果
执行基准测试：
go test -bench=.
输出示例：
BenchmarkAtomicInc-8 10000000 150 ns/op BenchmarkMutexInc-8 5000000 250 ns/op
从结果可以看出，在高并发下原子操作通常比互斥锁性能更好，因为其无阻塞特性减少了竞争开销。
基本上就这些。合理使用RunParallel能有效评估并发函数的真实性能，帮助我们做出更优的技术选择。
