Merge pull request #2087 from mrpalide/fix/handle-pprof-issue

fix pprof of ar

Author: mrpalide

cmd/address-resolver/commands/root.go

@@ -119,12 +119,18 @@ skywire svc ar --addr ":9093" --redis "redis://localhost:6379" --sk $(tail -n1 a
 		if pprofAddr != "" {
 			pprofMux := http.NewServeMux()
 
+			// Register the index (which links to everything else)
 			pprofMux.HandleFunc("/debug/pprof/", pprof.Index)
 			pprofMux.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
 			pprofMux.HandleFunc("/debug/pprof/profile", pprof.Profile)
 			pprofMux.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
 			pprofMux.HandleFunc("/debug/pprof/trace", pprof.Trace)
 
+			// Register profile handlers using pprof.Handler
+			for _, profile := range []string{"heap", "goroutine", "threadcreate", "block", "mutex", "allocs"} {
+				pprofMux.Handle("/debug/pprof/"+profile, pprof.Handler(profile))
+			}
+
 			go func() {
 				logger.Infof("Starting pprof server on %s", pprofAddr)
 				server := &http.Server{
@@ -139,6 +145,8 @@ skywire svc ar --addr ":9093" --redis "redis://localhost:6379" --sk $(tail -n1 a
 					logger.Errorf("pprof server failed: %v", err)
 				}
 			}()
+
+			time.Sleep(100 * time.Millisecond)
 		}
 
 		ctx, cancel := cmdutil.SignalContext(context.Background(), logger)