mirrors/diun
1
0
Fork 0
mirror of https://github.com/crazy-max/diun.git synced 2025-12-21 21:33:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add profiler flag (#336)

Browse Source 
Co-authored-by: CrazyMax <crazy-max@users.noreply.github.com>
This commit is contained in:
CrazyMax
2021-04-25 18:04:07 +02:00
committed by GitHub
parent 4c6606d882
commit 0438ca485f
8 changed files with 50 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
cmd/main.go
View File

@@ -14,6 +14,7 @@ import (
"github.com/crazy-max/diun/v4/internal/config"
"github.com/crazy-max/diun/v4/internal/logging"
"github.com/crazy-max/diun/v4/internal/model"
"github.com/pkg/profile"
"github.com/rs/zerolog/log"
)
@@ -75,6 +76,31 @@ func main() {
}
log.Debug().Msg(cfg.String())
// Profiler
if len(cli.Profiler) > 0 {
profilePath := profile.ProfilePath(cfg.Db.Path)
switch cli.Profiler {
case "cpu":
defer profile.Start(profile.CPUProfile, profilePath).Stop()
case "mem":
defer profile.Start(profile.MemProfile, profilePath).Stop()
case "alloc":
defer profile.Start(profile.MemProfile, profile.MemProfileAllocs(), profilePath).Stop()
case "heap":
defer profile.Start(profile.MemProfile, profile.MemProfileHeap(), profilePath).Stop()
case "routines":
defer profile.Start(profile.GoroutineProfile, profilePath).Stop()
case "mutex":
defer profile.Start(profile.MutexProfile, profilePath).Stop()
case "threads":
defer profile.Start(profile.ThreadcreationProfile, profilePath).Stop()
case "block":
defer profile.Start(profile.BlockProfile, profilePath).Stop()
default:
log.Fatal().Msgf("Unknown profiler: %s", cli.Profiler)
}
}
// Init
if diun, err = app.New(meta, cli, cfg); err != nil {
log.Fatal().Err(err).Msgf("Cannot initialize %s", meta.Name)