并发基准测试是通过模拟多协程环境来评估函数性能的方法。go 的基准测试默认串行执行,但可通过启动多个 goroutine 实现并发测试;1. 使用 mockgen 生成接口 mock 类型;2. 创建 mock 实例并设定期望值;3. 将 mock 注入被测函数。编写时使用 runparallel 方法控制并发,pb.next() 控制迭代,共享计时输出平均耗时。注意避免资源竞争、合理设置 gomaxprocs、防止过度模拟、关注内存分配,以确保测试准确性和稳定性。

并发压力测试是验证程序在高并发场景下性能表现的重要手段。Golang 本身对并发支持很好,使用 testing 包中的基准测试功能,再结合 gomock 做依赖模拟,可以很方便地编写出高效的并发压力测试用例。

什么是并发基准测试?
Go 的基准测试(Benchmark)默认是串行执行的,但你可以通过手动启动多个 goroutine 来模拟并发请求。这种做法非常适合测试函数在多协程下的性能表现,比如处理 HTTP 请求、数据库操作、缓存访问等。
并发测试的关键在于:控制并发数量、记录响应时间、避免资源竞争。
立即学习“go语言免费学习笔记(深入)”;

如何使用 gomock 模拟依赖?
在实际项目中,某些依赖项(如数据库、外部服务)可能不稳定或难以构造数据。这时就可以使用 gomock 对这些依赖进行模拟,保证测试的稳定性和可重复性。
步骤如下:
- 使用
mockgen生成接口的 mock 类型 - 在测试中创建 mock 实例并设置期望值和返回值
- 将 mock 实例注入到被测函数中
例如:

// 创建 mock 对象
mockDB := NewMockDatabase(ctrl)
// 设置期望调用次数和返回值
mockDB.EXPECT().Query(gomock.Any()).Return("data", nil).AnyTimes()这样即使没有真实数据库连接,也能完成并发测试。
编写并发基准测试的方法
Go 的 testing.B 提供了 RunParallel 方法,用于运行并发基准测试。下面是一个简单的模板:
func BenchmarkConcurrentAccess(b *testing.B) {
ctrl := gomock.NewController(b)
defer ctrl.Finish()
// 初始化 mock 对象
mockService := NewMockService(ctrl)
mockService.EXPECT().DoSomething(gomock.Any()).Return(nil).AnyTimes()
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
// 模拟并发请求
err := doWork(mockService)
if err != nil {
b.Fatal(err)
}
}
})
}关键点说明:
-
RunParallel会自动调整并发 goroutine 数量 -
pb.Next()控制迭代次数 - 所有 goroutine 共享一次计时,最终输出平均耗时
注意事项与常见问题
在并发测试中,有一些细节容易被忽略,但也会影响测试结果准确性:
避免共享状态导致的竞争
如果多个 goroutine 修改了相同的变量,记得加锁或者使用原子操作。合理设置 GOMAXPROCS(如果有必要)
Go 1.5+ 默认使用所有 CPU 核心,但在一些旧版本中可能需要手动设置。不要过度模拟,保持测试真实性
虽然 gomock 很强大,但如果模拟逻辑过于复杂,反而可能掩盖真实问题。关注内存分配和 GC 压力
并发下频繁创建对象可能导致内存激增,使用-benchmem查看内存分配情况。
基本上就这些。并发压力测试不复杂,但细节容易忽略,特别是结合 mock 和并发的时候,要确保模拟逻辑不影响测试目标。










