add -pprof option

sentriz · Dec 2, 2023 · 8163781 · 8163781
1 parent 32e0f30
commit 8163781
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/cmd/gonic/gonic.go b/cmd/gonic/gonic.go
@@ -9,6 +9,7 @@ import (
 	"fmt"
 	"log"
 	"net/http"
+	"net/http/pprof"
 	"os"
 	"os/signal"
 	"path"
@@ -87,6 +88,7 @@ func main() {
 	set.Var(&confMultiValueArtist, "multi-value-artist", "setting for mutli-valued track artist scanning (optional)")
 	set.Var(&confMultiValueAlbumArtist, "multi-value-album-artist", "setting for mutli-valued album artist scanning (optional)")
 
+	confPprof := set.Bool("pprof", false, "enable the /debug/pprof endpoint (optional)")
 	confExpvar := set.Bool("expvar", false, "enable the /debug/vars endpoint (optional)")
 
 	deprecatedConfGenreSplit := set.String("genre-split", "", "(deprecated, see multi-value settings)")
@@ -283,6 +285,14 @@ func main() {
 		}))
 	}
 
+	if *confPprof {
+		mux.HandleFunc("/debug/pprof/", pprof.Index)
+		mux.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
+		mux.HandleFunc("/debug/pprof/profile", pprof.Profile)
+		mux.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
+		mux.HandleFunc("/debug/pprof/trace", pprof.Trace)
+	}
+
 	errgrp, ctx := errgroup.WithContext(context.Background())
 
 	errgrp.Go(func() error {