Timeパッケージを使って簡易ベンチマークを取ってみる

個人的にメモ。
原始的だが time.Time パッケージを使って、終了時間から開始時間をSubする。

func BenchFuncSeconds(f func() error) error {
    st := time.Now()
    err := f()
    if err != nil {
        return err
    }
    ed := time.Now()
    log.Printf("Time %v", ed.Sub(st))
    return nil
}

例えば、下記の関数を計測するとする。

func GetHelloList(name string, count int) []string {
    var list []string
    for i:=0;i<count;i++ {
        list = append(list, fmt.Sprintf("Hello %s\n", name))
    }
    return list
}

BenchFuncSecondsの引数は(f func() errorなのでname string, count int)を引数に受け取り、[]stringを返り値に持つGetHelloListは一見計測できないように見える。

が、下記のように即時関数内func() errorで実行してやることで型の違いを気にせずに実行できるようになる。

func main() {
    _ = BenchFuncSeconds(func() error {
        _ = GetHelloList("Tarou", 99999)
        return nil
    })
}