testing包提供了对Go包的自动测试支持。这是和go test 命令相呼应的功能, go test 命令会自动执行所以符合格式

当带着 -bench=“.” ( 参数必须有!)来执行*go test命令的时候性能测试程序就会被顺序执行。符合下面格式的函数被认为是一个性能测试程序,

执行 go test -bench=”.” 后 结果 :

// 表示测试全部通过

源码包位置:src/testing/benchmark.go

testing目录下的文件有

testing.T

判定失败接口

打印信息接口

综合接口产生:


testing.B

首先 , testing.B 拥有testing.T 的全部接口。

下面带着三个问题去阅读源码:

  • b.N是如何自动调整的?

  • 内存统计是如何实现的?

  • SetBytes()其使用场景是什么?


B定义了性能测试的数据结构,我们提取其比较重要的一些成员进行分析:

启动计时:B.StartTimer()

StartTimer()负责启动计时并初始化内存相关计数,测试执行时会自动调用,一般不需要用户启动。

StartTimer()负责启动计时,并记录当前内存分配情况,不管是否有“-benchmem”参数,内存都会被统计,参数只决定是否要在结果中输出。


停止计时:B.StopTimer()

StopTimer()负责停止计时,并累加相应的统计值。

需要注意的是,StopTimer()并不一定是测试结束,一个测试中有可能有多个统计阶段,所以其统计值是累加的。


重置计时:B.ResetTimer()

ResetTimer()用于重置计时器,相应的也会把其他统计值也重置。

ResetTimer()比较常用,典型使用场景是一个测试中,初始化部分耗时较长,初始化后再开始计时。


设置处理字节数:B.SetBytes(n int64)

// SetBytes records the number of bytes processed in a single operation.

// If this is called, the benchmark will report ns/op and MB/s.

这是一个比较含糊的函数,通过其函数说明很难明白其作用。

其实它是用来设置单次迭代处理的字节数,一旦设置了这个字节数,那么输出报告中将会呈现“xxx MB/s”的信息,用来表示待测函数处理字节的性能。待测函数每次处理多少字节数只有用户清楚,所以需要用户设置。


报告内存信息:

ReportAllocs() 用于设置是否打印内存统计信息,与命令行参数“-benchmem”一致,但本方法只作用于单个测试函数。

性能测试是如何启动的

性能测试要经过多次迭代,每次迭代可能会有不同的b.N值,每次迭代执行测试函数一次,跟据此次迭代的测试结果来分析要不要继续下一次迭代。


我们先看一下每次迭代时所用到的方法,runN():

该方法指定b.N的值,执行一次测试函数。


与T.Run()类似,B.Run()也用于启动一个子测试,实际上用户编写的任何一个测试都是使用Run()方法启动的,我们看下B.Run()的伪代码:

所有的测试都是先使用run1()方法执行一次测试,run1()方法中实际上调用了runN(1),执行一次后再决定要不要继续迭代。


测试结果实际上以最后一次迭代的数据为准,当然,最后一次迭代往往意味着b.N更大,测试准确性相对更高。


B.N是如何调整的?

B.launch()方法里最终决定B.N的值。我们看下伪代码:

不考虑程序出错,而且用户没有主动停止测试的场景下,每个性能测试至少要执行b.benchTime长的秒数,默认为1s。先执行一遍的意义在于看用户代码执行一次要花费多长时间,如果时间较短,那么b.N值要足够大才可以测得更精确,如果时间较长,b.N值相应的会减少,否则会影响测试效率。


最终的b.N会被定格在某个10的指数级,是为了方便阅读测试报告。


内存是如何统计的?

我们知道在测试开始时,会把当前内存值记入到b.startAllocs和b.startBytes中,测试结束时,会用最终内存值与开始时的内存值相减,得到净增加的内存值,并记入到b.netAllocs和b.netBytes中。


每个测试结束,会把结果保存到BenchmarkResult对象里,该对象里保存了输出报告所必需的统计信息:

其中MemAllocs和MemBytes分别对应b.netAllocs和b.netBytes。

那么最终统计时只需要把净增加值除以b.N即可得到每次新增多少内存了。

每个操作内存对象新增值:

每个操作内存字节数新增值:

golang benchmark源码分析-LMLPHP

golang benchmark源码分析-LMLPHP


本文分享自微信公众号 - golang算法架构leetcode技术php(golangLeetcode)。
如有侵权,请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”,欢迎正在阅读的你也加入,一起分享。

07-02 19:14