如何在Go基准测试中排除数据准备的时间与内存分配？

要只统计Set方法的执行时间和内存分配，核心是把数据准备逻辑和哈希表初始化放在基准测试计时器启动之前，用Go测试包的b.ResetTimer()方法隔离无关开销。具体操作如下：

正确的基准测试写法示例 #

func BenchmarkHashTable_Set(b *testing.B) {
    // 1. 提前完成所有数据准备和哈希表初始化
    ht := NewHashTable()
    // 预先生成所有要测试的key和value，避免循环内重复创建
    keys := make([]string, b.N)
    values := make([][]byte, b.N)
    for i := 0; i < b.N; i++ {
        keys[i] = fmt.Sprintf("key-%d", i)
        values[i] = []byte(fmt.Sprintf("val-%d", i))
    }

    // 2. 重置计时器，彻底排除之前的准备开销
    b.ResetTimer()

    // 3. 只在循环内执行要测试的Set方法
    for i := 0; i < b.N; i++ {
        ht.Set(keys[i], values[i])
    }

    // 可选：防止编译器因"无实际使用"优化掉哈希表和Set调用
    b.KeepAlive(ht)
}

关键原理说明 #

• b.ResetTimer()：会重置基准测试的时间统计和内存分配计数，只有调用之后的代码会被纳入性能统计范围。
• 预生成测试数据：把key、value的创建逻辑移到计时器重置前，避免循环内的字符串拼接、字节数组转换等操作干扰Set方法的真实性能。
• b.KeepAlive()：如果你的Set方法没有返回值，或者哈希表后续没有被使用，编译器可能会优化掉整个调用逻辑，用这个方法可以强制保留哈希表实例，确保测试真实有效。

避坑提醒 #

• 不要把哈希表的初始化（NewHashTable()）放在循环内，否则每次循环都会创建新哈希表，完全偏离测试Set方法的初衷。
• 如果需要测试"空哈希表首次Set"的性能，那初始化可以保留在计时器之后，但这种场景属于特殊需求，大部分情况还是要把初始化放在准备阶段。

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