Golang如何做并发压力测试使用gomock编写并发基准测试-Golang-PHP中文网

Golang如何做并发压力测试使用gomock编写并发基准测试

P粉602998670

发布： 2025-07-25 09:53:02

原创

819人浏览过

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

Golang如何做并发压力测试使用gomock编写并发基准测试

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

什么是并发基准测试？

Go 的基准测试（Benchmark）默认是串行执行的，但你可以通过手动启动多个 goroutine 来模拟并发请求。这种做法非常适合测试函数在多协程下的性能表现，比如处理 HTTP 请求、数据库操作、缓存访问等。

并发测试的关键在于：控制并发数量、记录响应时间、避免资源竞争。

立即学习“go语言免费学习笔记（深入）”；

如何使用 gomock 模拟依赖？

在实际项目中，某些依赖项（如数据库、外部服务）可能不稳定或难以构造数据。这时就可以使用 gomock 对这些依赖进行模拟，保证测试的稳定性和可重复性。

白瓜面试

白瓜面试 - AI面试助手,辅助笔试面试神器

查看详情

步骤如下：

使用 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)
            }
        }
    })
}

登录后复制