add pprof middleware

supabase · Jun 28, 2023 · 34e662e · 34e662e
1 parent 4111d9a
commit 34e662e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/api/api.go b/internal/api/api.go
@@ -9,6 +9,7 @@ import (
 	"github.com/didip/tollbooth/v5"
 	"github.com/didip/tollbooth/v5/limiter"
 	"github.com/go-chi/chi"
+	chimiddleware "github.com/go-chi/chi/middleware"
 	"github.com/rs/cors"
 	"github.com/sebest/xff"
 	"github.com/sirupsen/logrus"
@@ -85,6 +86,9 @@ func NewAPIWithVersion(ctx context.Context, globalConfig *conf.GlobalConfigurati
 		r.UseBypass(api.databaseCleanup)
 	}
 
+	if globalConfig.Logging.Level == "debug" {
+		r.chi.Mount("/debug", chimiddleware.Profiler())
+	}
 	r.Get("/health", api.HealthCheck)
 
 	r.Route("/callback", func(r *router) {