blob: 94bcdb8b3730df66446af11cbbccbd0a75526da0 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
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()
})
}
}
|
