mirror of
https://github.com/chrislusf/seaweedfs
synced 2024-06-29 05:41:02 +02:00
34 lines
534 B
Go
34 lines
534 B
Go
package util
|
|
|
|
import (
|
|
"os"
|
|
"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)
|
|
defer pprof.StopCPUProfile()
|
|
OnInterrupt(func() {
|
|
pprof.StopCPUProfile()
|
|
})
|
|
}
|
|
if memProfile != "" {
|
|
f, err := os.Create(memProfile)
|
|
if err != nil {
|
|
glog.Fatal(err)
|
|
}
|
|
OnInterrupt(func() {
|
|
pprof.WriteHeapProfile(f)
|
|
f.Close()
|
|
})
|
|
}
|
|
|
|
}
|