-
Notifications
You must be signed in to change notification settings - Fork 661
/
profiler.go
132 lines (111 loc) · 3.05 KB
/
profiler.go
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
// Copyright (C) 2019-2023, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.
package profiler
import (
"errors"
"os"
"path/filepath"
"runtime"
"runtime/pprof"
"github.com/ava-labs/avalanchego/utils/perms"
)
const (
// Name of file that CPU profile is written to when StartCPUProfiler called
cpuProfileFile = "cpu.profile"
// Name of file that memory profile is written to when MemoryProfile called
memProfileFile = "mem.profile"
// Name of file that lock profile is written to
lockProfileFile = "lock.profile"
)
var (
errCPUProfilerRunning = errors.New("cpu profiler already running")
errCPUProfilerNotRunning = errors.New("cpu profiler doesn't exist")
)
// Profiler provides helper methods for measuring the current performance of
// this process
type Profiler interface {
// StartCPUProfiler starts measuring the cpu utilization of this process
StartCPUProfiler() error
// StopCPUProfiler stops measuring the cpu utilization of this process
StopCPUProfiler() error
// MemoryProfile dumps the current memory utilization of this process
MemoryProfile() error
// LockProfile dumps the current lock statistics of this process
LockProfile() error
}
type profiler struct {
dir,
cpuProfileName,
memProfileName,
lockProfileName string
cpuProfileFile *os.File
}
func New(dir string) Profiler {
return new(dir)
}
func new(dir string) *profiler {
return &profiler{
dir: dir,
cpuProfileName: filepath.Join(dir, cpuProfileFile),
memProfileName: filepath.Join(dir, memProfileFile),
lockProfileName: filepath.Join(dir, lockProfileFile),
}
}
func (p *profiler) StartCPUProfiler() error {
if p.cpuProfileFile != nil {
return errCPUProfilerRunning
}
if err := os.MkdirAll(p.dir, perms.ReadWriteExecute); err != nil {
return err
}
file, err := perms.Create(p.cpuProfileName, perms.ReadWrite)
if err != nil {
return err
}
if err := pprof.StartCPUProfile(file); err != nil {
_ = file.Close() // Return the original error
return err
}
runtime.SetMutexProfileFraction(1)
p.cpuProfileFile = file
return nil
}
func (p *profiler) StopCPUProfiler() error {
if p.cpuProfileFile == nil {
return errCPUProfilerNotRunning
}
pprof.StopCPUProfile()
err := p.cpuProfileFile.Close()
p.cpuProfileFile = nil
return err
}
func (p *profiler) MemoryProfile() error {
if err := os.MkdirAll(p.dir, perms.ReadWriteExecute); err != nil {
return err
}
file, err := perms.Create(p.memProfileName, perms.ReadWrite)
if err != nil {
return err
}
runtime.GC() // get up-to-date statistics
if err := pprof.WriteHeapProfile(file); err != nil {
_ = file.Close() // Return the original error
return err
}
return file.Close()
}
func (p *profiler) LockProfile() error {
if err := os.MkdirAll(p.dir, perms.ReadWriteExecute); err != nil {
return err
}
file, err := perms.Create(p.lockProfileName, perms.ReadWrite)
if err != nil {
return err
}
profile := pprof.Lookup("mutex")
if err := profile.WriteTo(file, 1); err != nil {
_ = file.Close() // Return the original error
return err
}
return file.Close()
}