go/go-micro
1
0
Fork 0
mirror of https://github.com/go-micro/go-micro.git synced 2025-06-24 22:26:54 +02:00
Code Issues Releases Activity

fixing mem.pprof issue: (#2198)

Browse Source 
parsing profile: concatenated profiles detected
fixing web service not start profile issue
fixing no default profile options issue
This commit is contained in:
Johnson C
2021-08-04 16:39:01 +08:00
committed by GitHub
parent c3107e6843
commit 3e0411a3f6
3 changed files with 39 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
debug/profile/pprof/pprof.go
View File

@ -7,7 +7,6 @@ import (
"runtime"
"runtime/pprof"
"sync"
"time"
"github.com/asim/go-micro/v3/debug/profile"
)
@ -17,7 +16,6 @@ type profiler struct {
sync.Mutex
running bool
exit chan bool
// where the cpu profile is written
cpuFile *os.File
@ -25,23 +23,6 @@ type profiler struct {
memFile *os.File
}
func (p *profiler) writeHeap(f *os.File) {
defer f.Close()
t := time.NewTicker(time.Second * 30)
defer t.Stop()
for {
select {
case <-t.C:
runtime.GC()
pprof.WriteHeapProfile(f)
case <-p.exit:
return
}
}
}
func (p *profiler) Start() error {
p.Lock()
defer p.Unlock()
@ -50,15 +31,12 @@ func (p *profiler) Start() error {
return nil
}
// create exit channel
p.exit = make(chan bool)
cpuFile := filepath.Join("/tmp", "cpu.pprof")
memFile := filepath.Join("/tmp", "mem.pprof")
cpuFile := filepath.Join(os.TempDir(), "cpu.pprof")
memFile := filepath.Join(os.TempDir(), "mem.pprof")
if len(p.opts.Name) > 0 {
cpuFile = filepath.Join("/tmp", p.opts.Name+".cpu.pprof")
memFile = filepath.Join("/tmp", p.opts.Name+".mem.pprof")
cpuFile = filepath.Join(os.TempDir(), p.opts.Name+".cpu.pprof")
memFile = filepath.Join(os.TempDir(), p.opts.Name+".mem.pprof")
}
f1, err := os.Create(cpuFile)
@ -76,9 +54,6 @@ func (p *profiler) Start() error {
return err
}
// write the heap periodically
go p.writeHeap(f2)
// set cpu file
p.cpuFile = f1
// set mem file
@ -93,18 +68,19 @@ func (p *profiler) Stop() error {
p.Lock()
defer p.Unlock()
select {
case <-p.exit:
return nil
default:
close(p.exit)
pprof.StopCPUProfile()
p.cpuFile.Close()
p.running = false
p.cpuFile = nil
p.memFile = nil
if !p.running {
return nil
}
pprof.StopCPUProfile()
p.cpuFile.Close()
runtime.GC()
pprof.WriteHeapProfile(p.memFile)
p.memFile.Close()
p.running = false
p.cpuFile = nil
p.memFile = nil
return nil
}
func (p *profiler) String() string {