45 lines
508 B
Go
45 lines
508 B
Go
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()
|
|
|
|
}
|