概要
この記事は、CyberAgent PTA Advent Calendar 2022の12日目の記事です。
広告プロダクトチームでサーバーサイドエンジニアをやっています。
Golangの広告プロダクト実装において、システムパフォーマンスも意識しながら実装をしていたため、今回はGoにベンチマークテストについて書きます。
Go Benchmark
Goのtestingパッケージには、ベンチマークを行う機能が提供されています。
func BenchmarkXxx(*testing.B)
のように、関数の接頭辞にBenchmarkを付けることで、ベンチマークテスト用の関数として見なされます。
go testに-benchフラグを付けることで実行可能です。
ベンチマーク用関数のサンプルは下記で、
func BenchmarkRandInt(b *testing.B) {
for i := 0; i < b.N; i++ {
rand.Int()
}
}
b.N回実行されるようになっていて、
下記のような実行結果が出力されます。
BenchmarkRandInt-8 68453040 17.8 ns/op
ループあたり17.8nsの速度で、68453040回実行されたことを意味しています。
オプション
いくつかのオプションが用意されていて、より詳細な情報を出力することもできます。
-benchmemはメモリの情報を出力できるので、おすすめ。
-cpu, -benchtime, -cpuprofileなどいくつかあります。
詳細はこちら: https://pkg.go.dev/cmd/go#hdr-Testing_flags
ベンチマークテストしてみる
import (
"fmt"
"strconv"
"testing"
)
func BenchmarkA(b *testing.B) {
for i := 0; i < b.N; i++ {
testA(i)
}
}
func BenchmarkB(b *testing.B) {
for i := 0; i < b.N; i++ {
testB(i)
}
}
func testA(num int) string {
return strconv.Itoa(num)
}
func testB(num int) string {
return fmt.Sprintf("%d", num)
}
strconv.Itoaとfmt.Sprintfを比較するテストを書いてみました。
go test -bench . -benchmemで実行します。
BenchmarkA-8 47877226 24.79 ns/op 7 B/op 0 allocs/op
BenchmarkB-8 15777586 76.22 ns/op 16 B/op 1 allocs/op
strconv.Itoaは、ループ当たり24.79nsの速度で、47877226回実行でき、
fmt.Sprintfは、ループ当たり76.22nsの速度で、15777586回実行できたことが分かりました!