为何设置-benchtime=1x时Go Benchmark函数仍会多次执行？

Go标准库的testing包执行基准测试时，会默认先执行校准阶段：先用很小的b.N值（通常就是1）运行一次测试函数，目的是评估单轮测试的耗时、判断是否存在函数内联等情况，用来确认后续正式运行的参数是否合理。你指定的-benchtime=1x是正式运行阶段的次数，所以校准阶段跑1次，正式阶段跑1次，总共就执行了2次，自然第二次（正式阶段的那次）会因为DB值已经被改成1而校验失败。

方法1：遵循基准测试标准写法拆分逻辑 #

基准测试的规范写法是把一次性初始化逻辑放到循环外、b.ResetTimer()之前，核心被测逻辑放在b.N次循环中，如果你的测试目的是压测读写逻辑，可以调整结构：

func BenchmarkMyStuff(b *testing.B) {
    // 一次性初始化逻辑，仅执行1次
    x := GetValueFromDB()
    require.Equal(b, 0, x)
    // 重置计时器，排除初始化耗时对基准结果的影响
    b.ResetTimer()
    // 循环b.N次执行核心被测逻辑
    for i := 0; i < b.N; i++ {
        x++
        WriteValueToDB(x)
    }
}

如果你的场景要求每次运行都必须从初始值0开始，可以用下面两种方案。

方法2：跳过校准阶段（Go 1.18+支持） #

Go 1.18版本新增了-benchtime的特殊语法，0xN格式的参数表示跳过校准阶段，直接运行N次。你可以把运行命令调整为：

go test -cpuprofile cpu.prof -memprofile mem.prof -run=XXX -benchtime=0x1 -bench BenchmarkMyStuff path/to/this/file

这样就只会运行1次测试，不会触发校准阶段的额外执行。

方法3：新增数据重置逻辑 #

你可以通过注册清理函数，每次测试执行完后自动把DB值恢复为初始状态，兼容多轮运行的场景：

func BenchmarkMyStuff(b *testing.B) {
    // 注册清理函数，测试结束后自动重置DB值
    b.Cleanup(func() {
        WriteValueToDB(0)
    })
    require.Equal(b, 1, b.N)
    x := GetValueFromDB()
    require.Equal(b, 0, x)
    x++
    WriteValueToDB(x)
}

内容的提问来源于stack exchange，提问作者Saqib Ali