如何让b.RunParallel()实现与b.Run()一致的基准测试Setup/Teardown？

问题背景 #

用b.Run()做基准测试时，把初始化（Setup）和延迟清理（Teardown）放在b.Run()外面就能实现单次执行——因为调用b.Run()的基准测试函数本身只会以N=1跑一次，不会被计入测量，完美实现单次初始化和清理。但换成b.RunParallel()时，要是没找对写法，Setup和Cleanup会跟着基准测试的N值重复执行，资源反复创建销毁，效率很低。

之前试过几种方案都不行：

• testing.TestMain是包级全局操作，会影响所有测试，没法只针对单个基准测试；
• init函数也是全局初始化，同样不符合需求；
• b.ResetTimer只能排除初始化的计时，但如果资源还是在并行逻辑里重复创建，效率还是比不上b.Run()的单次初始化方式。

可行方案 #

核心就是利用基准测试函数本身只执行一次这个特性，把单次Setup和Teardown放在b.RunParallel()的外面，而不是并行执行的代码块里，再配合b.ResetTimer()排除初始化的耗时，既保证资源只创建一次，又不影响其他测试。

代码示例 #

func BenchmarkParallelSingleSetup(b *testing.B) {
    // 单次初始化：整个基准测试只执行一次
    sharedResource := initSharedResource()
    defer cleanupSharedResource(sharedResource) // 基准测试结束时统一清理

    // 重置计时器，不让初始化耗时影响基准测试结果
    b.ResetTimer()

    // 启动并行测试
    b.RunParallel(func(pb *testing.PB) {
        // 并行执行的逻辑，直接用外面初始化好的资源
        for pb.Next() {
            runBenchmarkTask(sharedResource)
        }
    })
}

// 初始化共享资源（比如数据库连接、缓存实例等）
func initSharedResource() *MyResource {
    return &MyResource{}
}

// 清理共享资源
func cleanupSharedResource(r *MyResource) {
    // 释放资源逻辑，比如关闭连接、删除临时文件
}

// 基准测试的核心任务
func runBenchmarkTask(r *MyResource) {
    // 这里写要测试的业务逻辑
}

关键细节 #

1. 基准测试函数本身只会执行一次，所以外层的initSharedResource()和defer cleanupSharedResource()只会跑一次，完全实现了针对当前基准测试的单次Setup/Teardown。
2. b.ResetTimer()把计时起点拉到并行测试开始前，确保基准测试的结果只反映核心逻辑的耗时，不包含初始化的时间。
3. 所有并行goroutine共享同一个初始化好的资源，避免了重复创建资源的开销，效率和b.Run()的方案一致。
4. 这个方案只作用于当前基准测试，不会影响包内其他测试用例，完全符合需求。

内容的提问来源于stack exchange，提问作者Sapan V