diff --git a/biz/util/ppro.go b/biz/util/ppro.go new file mode 100644 index 0000000..0b1e578 --- /dev/null +++ b/biz/util/ppro.go @@ -0,0 +1,44 @@ +package util + +import ( + "log" + "os" + "runtime/pprof" +) + +type Pprof struct { + fc *os.File + fm *os.File +} + +func NewProf() *Pprof { + fc, err := os.Create("./cpu.prof") + if err != nil { + log.Fatal(err) + } + + fm, err := os.Create("./men.prof") + if err != nil { + log.Fatal(err) + } + + p := &Pprof{ + fc: fc, + fm: fm, + } + p.Start() + return p +} + +func (p *Pprof) Start() { + pprof.StartCPUProfile(p.fc) + pprof.WriteHeapProfile(p.fm) +} + +func (p *Pprof) Close() { + pprof.StopCPUProfile() + + p.fc.Close() + p.fm.Close() + +}