seaweedfs/weed/util/grace/pprof.go

35 lines
544 B
Go
Raw Normal View History

package grace
2017-06-22 08:33:58 +00:00
import (
"os"
"runtime"
2017-06-22 08:33:58 +00:00
"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
2017-06-22 08:33:58 +00:00
f, err := os.Create(memProfile)
if err != nil {
glog.Fatal(err)
}
OnInterrupt(func() {
pprof.WriteHeapProfile(f)
f.Close()
})
}
}