mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-12-25 18:27:54 +08:00
35 lines
544 B
Go
35 lines
544 B
Go
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()
|
|
})
|
|
}
|
|
|
|
}
|