mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
56 lines
974 B
Go
56 lines
974 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)
|
|
}
|
|
runtime.SetBlockProfileRate(1)
|
|
runtime.SetMutexProfileFraction(1)
|
|
pprof.StartCPUProfile(f)
|
|
OnInterrupt(func() {
|
|
pprof.StopCPUProfile()
|
|
|
|
// write block pprof
|
|
blockF, err := os.Create(cpuProfile+".block")
|
|
if err != nil {
|
|
return
|
|
}
|
|
p := pprof.Lookup("block")
|
|
p.WriteTo(blockF,0)
|
|
blockF.Close()
|
|
|
|
// write mutex pprof
|
|
mutexF, err := os.Create(cpuProfile+".mutex")
|
|
if err != nil {
|
|
return
|
|
}
|
|
p = pprof.Lookup("mutex")
|
|
p.WriteTo(mutexF,0)
|
|
mutexF.Close()
|
|
|
|
})
|
|
}
|
|
if memProfile != "" {
|
|
runtime.MemProfileRate = 1
|
|
f, err := os.Create(memProfile)
|
|
if err != nil {
|
|
glog.Fatal(err)
|
|
}
|
|
OnInterrupt(func() {
|
|
pprof.WriteHeapProfile(f)
|
|
f.Close()
|
|
})
|
|
}
|
|
|
|
}
|