diff --git a/cmd/vaultik/main.go b/cmd/vaultik/main.go index 70e7c19..234de24 100644 --- a/cmd/vaultik/main.go +++ b/cmd/vaultik/main.go @@ -16,9 +16,11 @@ func main() { panic("could not create CPU profile: " + err.Error()) } defer func() { _ = f.Close() }() + if err := pprof.StartCPUProfile(f); err != nil { panic("could not start CPU profile: " + err.Error()) } + defer pprof.StopCPUProfile() } @@ -30,7 +32,9 @@ func main() { panic("could not create memory profile: " + err.Error()) } defer func() { _ = f.Close() }() + runtime.GC() // get up-to-date statistics + if err := pprof.WriteHeapProfile(f); err != nil { panic("could not write memory profile: " + err.Error()) }