Extra pprof endpoints (#1031)

This PR adds extra `pprof` endpoints that we can handle. So we can profile: ``` # CPU usage go tool pprof -http=: http://localhost:9999/debug/pprof/profile # Heap usage go tool pprof -http=: http://localhost:9999/debug/pprof/heap # Concurrency profiling # It requires changing the constant `main.EnableConcurrencyProfiling` go tool pprof -http=: http://localhost:9999/debug/pprof/mutex go tool pprof -http=: http://localhost:9999/debug/pprof/block ``` --------- Signed-off-by: Rafał Strzaliński <[email protected]> Co-authored-by: Przemyslaw Delewski <[email protected]>
QuesmaOrg · Nov 22, 2024 · 32e99dc · 32e99dc
1 parent 968ed31
commit 32e99dc
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/quesma/main.go b/quesma/main.go
@@ -28,6 +28,7 @@ import (
 	"quesma/table_resolver"
 	"quesma/telemetry"
 	"quesma/tracing"
+	"runtime"
 	"syscall"
 	"time"
 )
@@ -41,7 +42,15 @@ const banner = `
                       \__>           \/     \/      \/     \/ 
 `
 
+const EnableConcurrencyProfiling = false
+
 func main() {
+
+	if EnableConcurrencyProfiling {
+		runtime.SetBlockProfileRate(1)
+		runtime.SetMutexProfileFraction(1)
+	}
+
 	println(banner)
 	fmt.Printf("Quesma build info: version=[%s], build hash=[%s], build date=[%s]\n",
 		buildinfo.Version, buildinfo.BuildHash, buildinfo.BuildDate)

diff --git a/quesma/quesma/ui/console_routes.go b/quesma/quesma/ui/console_routes.go
@@ -306,6 +306,9 @@ func (qmc *QuesmaManagementConsole) initPprof(router *mux.Router) {
 	router.HandleFunc("/debug/pprof/", pprof.Index)
 	router.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
 	router.HandleFunc("/debug/pprof/profile", pprof.Profile)
+	router.HandleFunc("/debug/pprof/heap", pprof.Handler("heap").ServeHTTP)
+	router.HandleFunc("/debug/pprof/block", pprof.Handler("block").ServeHTTP)
+	router.HandleFunc("/debug/pprof/mutex", pprof.Handler("mutex").ServeHTTP)
 	router.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
 	router.HandleFunc("/debug/pprof/trace", pprof.Trace)
 }