From 7c2d5f723add95d97d664fa6a576c4dbd4b374f4 Mon Sep 17 00:00:00 2001 From: Johan Siebens Date: Mon, 12 Feb 2024 13:09:45 +0100 Subject: [PATCH] feat: add pprof endpoints --- internal/server/echo.go | 4 ++++ internal/server/server.go | 2 ++ 2 files changed, 6 insertions(+) diff --git a/internal/server/echo.go b/internal/server/echo.go index 91ed0ab..e0b2236 100644 --- a/internal/server/echo.go +++ b/internal/server/echo.go @@ -9,6 +9,10 @@ import ( "time" ) +func handler(h http.Handler) echo.HandlerFunc { + return echo.WrapHandler(h) +} + func EchoErrorHandler() echo.MiddlewareFunc { return func(next echo.HandlerFunc) echo.HandlerFunc { return func(c echo.Context) error { diff --git a/internal/server/server.go b/internal/server/server.go index bbd6980..05e057f 100644 --- a/internal/server/server.go +++ b/internal/server/server.go @@ -15,6 +15,7 @@ import ( "github.com/jsiebens/ionscale/internal/service" "github.com/jsiebens/ionscale/internal/templates" "github.com/labstack/echo-contrib/echoprometheus" + "github.com/labstack/echo-contrib/pprof" "github.com/labstack/echo/v4" "github.com/labstack/echo/v4/middleware" certmagicsql "github.com/travisjeffery/certmagic-sqlstorage" @@ -108,6 +109,7 @@ func Start(ctx context.Context, c *config.Config) error { metricsHandler := echo.New() metricsHandler.GET("/metrics", echoprometheus.NewHandler()) + pprof.Register(metricsHandler) createPeerHandler := func(machinePublicKey key.MachinePublic) http.Handler { registrationHandlers := handlers.NewRegistrationHandlers(machinePublicKey, c, sessionManager, repository)