Go的BenchMark的使用 | 青训营笔记

115 阅读1分钟

这是我参与「第五届青训营 」伴学笔记创作活动的第 5 天

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指明运行多少个基准测试。

运行结果如下

image.png

其中:

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