6a979fb40d
* update staticcheck Don't fork staticcheck: use the upstream version. Remove unneeded SSA loading. * Cache go/analysis facts Don't load unneeded packages for go/analysis. Repeated run of go/analysis linters now 10x faster (2s vs 20s on this repo) than before.
85 lines
1.6 KiB
Go
85 lines
1.6 KiB
Go
package timeutils
|
|
|
|
import (
|
|
"fmt"
|
|
"sort"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/golangci/golangci-lint/pkg/logutils"
|
|
)
|
|
|
|
type Stopwatch struct {
|
|
name string
|
|
startedAt time.Time
|
|
stages map[string]time.Duration
|
|
log logutils.Log
|
|
|
|
sync.Mutex
|
|
}
|
|
|
|
func NewStopwatch(name string, log logutils.Log) *Stopwatch {
|
|
return &Stopwatch{
|
|
name: name,
|
|
startedAt: time.Now(),
|
|
stages: map[string]time.Duration{},
|
|
log: log,
|
|
}
|
|
}
|
|
|
|
type stageDuration struct {
|
|
name string
|
|
d time.Duration
|
|
}
|
|
|
|
func (s *Stopwatch) sprintStages() string {
|
|
if len(s.stages) == 0 {
|
|
return "no stages"
|
|
}
|
|
|
|
stageDurations := []stageDuration{}
|
|
for n, d := range s.stages {
|
|
stageDurations = append(stageDurations, stageDuration{
|
|
name: n,
|
|
d: d,
|
|
})
|
|
}
|
|
sort.Slice(stageDurations, func(i, j int) bool {
|
|
return stageDurations[i].d > stageDurations[j].d
|
|
})
|
|
stagesStrings := []string{}
|
|
for _, s := range stageDurations {
|
|
stagesStrings = append(stagesStrings, fmt.Sprintf("%s: %s", s.name, s.d))
|
|
}
|
|
|
|
return fmt.Sprintf("stages: %s", strings.Join(stagesStrings, ", "))
|
|
}
|
|
|
|
func (s *Stopwatch) Print() {
|
|
p := fmt.Sprintf("%s took %s", s.name, time.Since(s.startedAt))
|
|
if len(s.stages) == 0 {
|
|
s.log.Infof("%s", p)
|
|
return
|
|
}
|
|
|
|
s.log.Infof("%s with %s", p, s.sprintStages())
|
|
}
|
|
|
|
func (s *Stopwatch) PrintStages() {
|
|
var stagesDuration time.Duration
|
|
for _, s := range s.stages {
|
|
stagesDuration += s
|
|
}
|
|
s.log.Infof("%s took %s with %s", s.name, stagesDuration, s.sprintStages())
|
|
}
|
|
|
|
func (s *Stopwatch) TrackStage(name string, f func()) {
|
|
startedAt := time.Now()
|
|
f()
|
|
|
|
s.Lock()
|
|
s.stages[name] += time.Since(startedAt)
|
|
s.Unlock()
|
|
}
|