tricrypt/profilers/cpu.go

package profilers

import (
	"os"
	"runtime/pprof"
)

func GetCPUProfiler() func() {
	f, err := os.Create("cpu.prof")
	if err != nil {
		panic(err)
	}
	defer f.Close()
	if err := pprof.StartCPUProfile(f); err != nil {
		panic(err)
	}
	return pprof.StopCPUProfile
}