-
Notifications
You must be signed in to change notification settings - Fork 5k
/
stats.go
52 lines (44 loc) · 957 Bytes
/
stats.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
package stats
import (
"time"
)
// mock out time.Now() for unit tests
var now = time.Now
// TimingStats is a helper to breakdown the timing of an expensive function call
// Usage:
// ts := NewTimingStats()
// ts.AddCheckpoint("checkpoint-1")
// ...
// ts.AddCheckpoint("checkpoint-2")
// ...
// ts.AddCheckpoint("checkpoint-3")
// ts.Timings()
type TimingStats struct {
StartTime time.Time
checkpoints []tsCheckpoint
}
type tsCheckpoint struct {
name string
time time.Time
}
func NewTimingStats() *TimingStats {
return &TimingStats{
StartTime: now(),
}
}
func (t *TimingStats) AddCheckpoint(name string) {
cp := tsCheckpoint{
name: name,
time: now(),
}
t.checkpoints = append(t.checkpoints, cp)
}
func (t *TimingStats) Timings() map[string]time.Duration {
timings := make(map[string]time.Duration)
prev := t.StartTime
for _, cp := range t.checkpoints {
timings[cp.name] = cp.time.Sub(prev)
prev = cp.time
}
return timings
}