【青训营】Go的BenchMark的使用

本文内容总结于 字节跳动青年训练营 第五届后端组

Go自带了一些性能测试工具，其中BenchMark是较为重要的一个。
我们以计算斐波那契数列的示例来展示BenchMark的使用

package Benchmark

import "testing"

func Fib(n int) int {
   if n < 2 {
      return n
   }
   return Fib(n-1) + Fib(n-2)
}

// from fib_test.go
func BenchmarkFib(b *testing.B) {
   // 运行Fib 函数b.N次
   for n := 0; n < b.N; n++ {
      Fib(10)
   }
}

其中func BenchmarkFib()为测试函数，传入的参数为b *testing.B，当运行Benchmark基准测试的时候，BenchMark会抓取参数为 *testing.B的函数进行测试

1. 基准测试代码文件必须是_test.go结尾，和单元测试一样；
2. 基准测试的函数以Benchmark开头；
3. 参数须为 *testing.B；
4. 基准测试函数不能有返回值；
5. b.N是基准测试指定的循环次数，不能由用户指定

然后使用命令行输入:

go test -bench=. -benchmem -count=3

其中go test用于启动测试，使用-bench指明启动的是基准测试，并且-bench=也用于指明测试的是什么函数，-bench=.的意思是测试所有Bench函数。 -benchmem用于指明基准测试的内存相关设置，现在暂时不使用。-count=3指明运行多少个基准测试。

运行结果如下

其中：

• BenchmarkFib-8 为测试的函数名
• 第二列表示基准测试进行了多少次测试，这就是b.N的值
• 第三列表示每次执行花费了多少时间
• 第四列表示每次执行需要多少内存
• 第五列表示每次执行申请了多少次内存