fix: enabling ppof endpoint over DEBUG_MEMORY env

This commit is contained in:
Aditya
2025-09-03 19:42:56 +05:30
parent 9f3070b65c
commit cbe8f908bd
+1 -1
View File
@@ -4772,7 +4772,7 @@ func runWebserver(listener net.Listener) {
if strings.ToLower(os.Getenv("SHUFFLE_SWARM_CONFIG")) == "run" || strings.ToLower(os.Getenv("SHUFFLE_APP_REPLICAS")) == "" { if strings.ToLower(os.Getenv("SHUFFLE_SWARM_CONFIG")) == "run" || strings.ToLower(os.Getenv("SHUFFLE_APP_REPLICAS")) == "" {
// go AutoScaleApps(ctx, dockercli, maxExecutionsPerMinute) // go AutoScaleApps(ctx, dockercli, maxExecutionsPerMinute)
} }
if strings.ToLower(os.Getenv("SHUFFLE_DEBUG_MEMORY")) == "true" { if (strings.ToLower(os.Getenv("SHUFFLE_DEBUG_MEMORY")) == "true" || strings.ToLower(os.Getenv("DEBUG_MEMORY")) == "true") {
r.HandleFunc("/debug/pprof/", pprof.Index) r.HandleFunc("/debug/pprof/", pprof.Index)
r.HandleFunc("/debug/pprof/heap", pprof.Handler("heap").ServeHTTP) r.HandleFunc("/debug/pprof/heap", pprof.Handler("heap").ServeHTTP)
r.HandleFunc("/debug/pprof/profile", pprof.Profile) r.HandleFunc("/debug/pprof/profile", pprof.Profile)