fix pprof.

m1k1o · Nov 25, 2023 · ffeef5e · ffeef5e
1 parent cdd0e4f
commit ffeef5e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 39 deletions.
diff --git a/internal/server/debug.go b/internal/server/debug.go
diff --git a/internal/server/manager.go b/internal/server/manager.go
@@ -58,12 +58,6 @@ func New(ApiManager types.ApiManager, roomConfig *config.Room, config *config.Se
 		}))
 	}
 
-	// mount pprof endpoint
-	if config.PProf {
-		withPProf(router)
-		logger.Info().Msgf("with pprof endpoint at %s", pprofPath)
-	}
-
 	//
 	// admin page
 	//
@@ -240,6 +234,12 @@ func New(ApiManager types.ApiManager, roomConfig *config.Room, config *config.Se
 		})
 	}
 
+	// mount pprof endpoint
+	if config.PProf {
+		router.Mount("/debug", middleware.Profiler())
+		logger.Info().Msgf("with pprof endpoint")
+	}
+
 	// handle all remaining paths with proxy
 	router.Handle("/*", proxyHandler)