使用sync.WaitGroup确保所有协程完成,结合互斥锁保护共享变量,验证并发操作后结果符合预期。

测试并发函数时,保证结果正确的核心在于控制并发行为的可预测性,并验证最终状态是否符合预期。Golang 提供了多种机制来帮助我们写出可靠的并发测试。
使用 sync.WaitGroup 等待所有协程完成
在并发测试中,必须确保所有 goroutine 都执行完毕后再检查结果,否则可能因提前判断导致误报。
通过 sync.WaitGroup 可以协调多个 goroutine 的执行完成:
func TestConcurrentIncrement(t *testing.T) {var counter int
var mu sync.Mutex
var wg sync.WaitGroup
numGoroutines := 10
incrementTimes := 100
for i := 0; i wg.Add(1)
go func() {
defer wg.Done()
for j := 0; j mu.Lock()
counter++
mu.Unlock()
}
}()
}
wg.Wait()
if counter != numGoroutines*incrementTimes {
t.Errorf("expected %d, got %d", numGoroutines*incrementTimes, counter)
}
}
使用 t.Parallel() 并行运行测试用例
将独立的测试标记为并行,可以提高测试效率,同时暴露竞态问题。
立即学习“go语言免费学习笔记(深入)”;
调用 t.Parallel() 后,该测试会与其他并行测试同时运行:
func TestSomethingParallel(t *testing.T) {t.Parallel()
// 执行并发相关逻辑测试
}
这样可以让多个测试共享 CPU 资源,更容易触发调度器切换,从而发现潜在并发 bug。
启用竞态检测(Race Detector)
Golang 内建的竞态检测工具是保证并发正确性的关键手段。
运行测试时加上 -race 标志:
go test -race ./...它会在程序运行时监控内存访问,一旦发现多个 goroutine 同时读写同一变量且无同步措施,就会报错。这是捕捉数据竞争最有效的办法。
使用 channel 控制执行顺序或收集结果
在某些场景下,用 channel 替代锁更清晰安全。测试时可通过 channel 收集输出,确保结果一致性。
func TestConcurrentWithChannel(t *testing.T) {resultCh := make(chan int, 10)
for i := 0; i go func(val int) {
resultCh }(i)
}
var results []int
for i := 0; i results = append(results, }
if len(results) != 10 {
t.Fatal("missing results")
}
// 可进一步校验值是否合理
}
基本上就这些。关键是让测试覆盖典型并发路径,配合 WaitGroup、互斥锁、channel 和 race 检测,就能有效保障并发函数的正确性。










