Added mux handler for pprof debug endpoint

accuknox · May 17, 2023 · 5f0ee5a · 5f0ee5a
1 parent 268c443
commit 5f0ee5a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/main.go b/src/main.go
@@ -1,7 +1,6 @@
 package main
 
 import (
-	"fmt"
 	"math/rand"
 	"net"
 	"os"
@@ -44,7 +43,9 @@ func init() {
 		// Server for pprof
 		go func() {
 			log.Info().Msgf("pprof enabled on :6060\n")
-			fmt.Println(http.ListenAndServe(":6060", nil))
+			r := http.NewServeMux()
+			r.Handle("/debug/", http.DefaultServeMux)
+			http.ListenAndServe(":6060", r)
 		}()
 	}