package grace import ( "os" "runtime" "runtime/pprof" "github.com/chrislusf/seaweedfs/weed/glog" ) func SetupProfiling(cpuProfile, memProfile string) { if cpuProfile != "" { f, err := os.Create(cpuProfile) if err != nil { glog.Fatal(err) } pprof.StartCPUProfile(f) OnInterrupt(func() { pprof.StopCPUProfile() }) } if memProfile != "" { runtime.MemProfileRate = 1 f, err := os.Create(memProfile) if err != nil { glog.Fatal(err) } OnInterrupt(func() { pprof.WriteHeapProfile(f) f.Close() }) } }