go语言工具_benchmark

go语言工具_benchmark

基准测试

基准测试主要通过测试CPU和内存的效率问题，来评估被测试代码的性能，进而找到更好的解决方案。

编写一个简单的基准测试

func BenchmarkXXXXXX(b *testing.B) {
	b.ResetTimer()
	defer b.StopTimer()
	for idx := 0; idx < b.N; idx++ {
		//TODO 写入想测试的函数
	}
}

写基准测试的时候，有几个注意事项：

1、文件名必须以_test.go结尾。
2、基准测试的函数必须以Benchmark开头。
3、基准测试函数必须接受一个指向Benchmark类型的指针作为唯一参数。
4、该函数不能有返回值。
5、被测试的代码应该放入在for循环中。

这里简单测试一下字符串相加吧。

var s string
for i := 0; i < 100; i++ {
	s += strconv.Itoa(i)
}

将这段代码复制到刚刚的TODO的位置。然后保存文件。在cmd中输入以下命令。

go test -bench=. -benchmem

重点看

BenchmarkXXXXXX-4         150320              7733 ns/op            9776 B/op         99 allocs/op

1、BenchmarkXXXXXX为测试的函数名。
2、函数名后的-4，代表运行时对应的GOMAXPROCS的值。
3、150320 表示运行for循环的次数，也就是代码被测试的次数。
4、7733 ns/op表示每次执行需要的时间。（ns是纳秒的缩写，op是operation操作的意思）
5、9776B/op表示每次操作分配的字节数。
6、99 allocs/op表示每次操作分配内存的次数。

我们知道在go语言中，string是只读的，每次操作时都需要去分配内存，如果直接相加效率会很低，字符串拼接有其他集中方法，一起用Benchmark测试一下。
分别为：
fmt.Sprintf、strings.Builder、bytes.Buffer

package mybench

import (
	"bytes"
	"fmt"
	"strconv"
	"strings"
	"testing"
)

const numbers = 100

func BenchmarkSprintf(b *testing.B) {
	b.ResetTimer()
	defer b.StopTimer()
	for idx := 0; idx < b.N; idx++ {
		var s string
		for i := 0; i < numbers; i++ {
			s = fmt.Sprintf("%v%v", s, i)
		}
	}
}

func BenchmarkStringBuilder(b *testing.B) {
	b.ResetTimer()
	defer b.StopTimer()
	for idx := 0; idx < b.N; idx++ {
		var bulider strings.Builder
		for i := 0; i < numbers; i++ {
			bulider.WriteString(strconv.Itoa(i))
		}
		_ = bulider.String()
	}
}

func BenchmarkBytesBuf(b *testing.B) {
	b.ResetTimer()
	defer b.StopTimer()
	for idx := 0; idx < b.N; idx++ {
		var buf bytes.Buffer
		for i := 0; i < numbers; i++ {
			buf.WriteString(strconv.Itoa(i))
		}
		_ = buf.String()
	}
}

func BenchmarkStringAdd(b *testing.B) {
	b.ResetTimer()
	defer b.StopTimer()
	for idx := 0; idx < b.N; idx++ {
		var s string
		for i := 0; i < numbers; i++ {
			s += strconv.Itoa(i)
		}
	}
}

测试结果

很显然StringBuilder性能更好。

小结：遇到某个需求有好几种方式去实现时，可以使用Benchmark去测试一下性能，选取最优解去实现。字符串拼接尽量使用StringBuilder。