Go基准测试的常见函数

上面讲了基准测试的方法及常用参数的详细说明，但是为了获取更准确的测试信息，了解代码的运行性能，我们常常会配合一些函数来获得更精准的数据。在本文中，我们将介绍testing包中的几个常用的基准测试函数，以及它们的作用和用法。

一、ResetTimer

ResetTimer函数用于重置基准测试的计时器。它会将已经消耗的时间和内存分配次数清零，重新开始计时和统计。这个函数通常用于在基准测试函数中排除一些初始化或准备工作的时间和内存开销，使得基准测试的结果更加准确和公平。

例如，我们想要测试一个排序函数的性能，我们可以先生成一个随机的切片作为输入，然后调用ResetTimer函数，再调用排序函数：

func BenchmarkSort(b *testing.B) {
    // 生成一个随机的切片
    slice := make([]int, 1000)
    for i := range slice {
        slice[i] = rand.Intn(1000)
    }
    // 重置计时器
    b.ResetTimer()
    // 循环执行排序函数
    for i := 0; i < b.N; i++ {
        sort.Ints(slice)
    }
}

二、StopTimer & StartTimer

StopTimer函数用于暂停基准测试的计时器。它会停止计时和统计内存分配次数。StartTimer函数用于恢复基准测试的计时器。它会继续计时和统计内存分配次数。这两个函数通常用于在基准测试函数中排除一些不属于被测代码本身的时间和内存开销，例如打印日志、清理资源等。

例如，我们想要测试一个网络请求的性能，我们可以先创建一个客户端和一个请求对象，然后调用StopTimer函数，再发送请求并接收响应，然后调用StartTimer函数，最后输出结果：

func BenchmarkRequest(b *testing.B) {
   // 创建一个客户端和一个请求对象
	client := &http.Client{}
	req, err := http.NewRequest("GET", "https://www.baidu.com", nil)
	if err != nil {
		b.Fatal(err)
	}
	// 暂停计时器
	b.StopTimer()
	// 循环发送请求并接收响应
	for i := 0; i < b.N; i++ {
		// 恢复计时器
		b.StartTimer()
		resp, err := client.Do(req)
		if err != nil {
			b.Fatal(err)
		}
		// 关闭响应体
		resp.Body.Close()
		// 暂停计时器
		b.StopTimer()
	}
}

三、ReportAllocs

ReportAllocs函数用于开启基准测试的内存分配报告。它会在基准测试的结果中显示每次迭代分配了多少字节和多少对象。这个函数通常用于在基准测试函数中检测代码的内存效率和垃圾回收压力。

例如，我们想要测试一个字符串拼接的性能，我们可以先创建一个初始字符串和一个要拼接的字符串，然后调用ReportAllocs函数，再循环执行字符串拼接操作，最后输出结果：

func BenchmarkConcat(b *testing.B) {
    // 创建一个初始字符串和一个要拼接的字符串
    s1 := "Hello"
    s2 := "World"
    // 开启内存分配报告
    b.ReportAllocs()
    // 循环执行字符串拼接操作
    for i := 0; i < b.N; i++ {
        s1 += s2
    }
}

四、SetBytes

SetBytes函数用于设置基准测试的字节处理量。它会在基准测试的结果中显示每秒处理了多少字节。这个函数通常用于在基准测试函数中衡量代码的吞吐量和效率。

例如，我们想要测试一个压缩函数的性能，我们可以先创建一个随机的字节切片作为输入，然后调用SetBytes函数，再循环执行压缩函数，最后输出结果：

func BenchmarkCompress(b *testing.B) {
	// 创建一个随机的字节切片
	data := make([]byte, 1024)
	rand.Read(data)
	// 设置字节处理量
	b.SetBytes(int64(len(data)))
	// 循环执行压缩函数
	for i := 0; i < b.N; i++ {
		compress(data) 
		// compress略
	}
}

好了，掌握了的上面的函数，相信你对于golang的基准测试已经游刃有余了，下面的章节我们将会从实际编码出发，开始逐个要点讲解golang高性能编程的技巧，当然我也会带领你了解必要的原理知识，加深对技巧的认知，而不是八股文式的学习。