From fb470068093c8b94ece8f94df7d5453b3f5915c8 Mon Sep 17 00:00:00 2001
From: Nathan Coad <nathan@thecoads.com>
Date: Fri, 27 Sep 2024 20:14:52 +1000
Subject: [PATCH] add debug endpoints

---
 main.go                 | 1 -
 server/router/router.go | 8 ++++++++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index 1c844dd..b0f039e 100644
--- a/main.go
+++ b/main.go
@@ -4,7 +4,6 @@ import (
 	"context"
 	"fmt"
 	"log/slog"
-	_ "net/http/pprof"
 	"os"
 	"runtime"
 	"time"
diff --git a/server/router/router.go b/server/router/router.go
index c9f5667..ff4a962 100644
--- a/server/router/router.go
+++ b/server/router/router.go
@@ -3,6 +3,7 @@ package router
 import (
 	"log/slog"
 	"net/http"
+	"net/http/pprof"
 	"vctp/db"
 	"vctp/dist"
 	"vctp/internal/secrets"
@@ -42,5 +43,12 @@ func New(logger *slog.Logger, database db.Database, buildTime string, sha1ver st
 	// endpoint for encrypting vcenter credential
 	mux.HandleFunc("/api/encrypt", h.EncryptData)
 
+	// Register pprof handlers
+	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)
+
 	return middleware.NewLoggingMiddleware(logger, mux)
 }