add pprof (#24)

cmd/httpserver/main.go

@@ -44,6 +44,11 @@ var flags []cli.Flag = []cli.Flag{
 		Value: "your-project",
 		Usage: "add 'service' tag to logs",
 	},
+	&cli.BoolFlag{
+		Name:  "pprof",
+		Value: false,
+		Usage: "enable pprof debug endpoint",
+	},
 	&cli.Int64Flag{
 		Name:  "drain-seconds",
 		Value: 45,
@@ -63,6 +68,7 @@ func main() {
 			logDebug := cCtx.Bool("log-debug")
 			logUID := cCtx.Bool("log-uid")
 			logService := cCtx.String("log-service")
+			enablePprof := cCtx.Bool("pprof")
 			drainDuration := time.Duration(cCtx.Int64("drain-seconds")) * time.Second
 
 			log := common.SetupLogger(&common.LoggingOpts{
@@ -81,6 +87,7 @@ func main() {
 				ListenAddr:  listenAddr,
 				MetricsAddr: metricsAddr,
 				Log:         log,
+				EnablePprof: enablePprof,
 
 				DrainDuration:            drainDuration,
 				GracefulShutdownDuration: 30 * time.Second,

httpserver/server.go

@@ -11,12 +11,14 @@ import (
 	"github.com/flashbots/go-template/metrics"
 	"github.com/flashbots/go-utils/httplogger"
 	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/chi/v5/middleware"
 	"go.uber.org/atomic"
 )
 
 type HTTPServerConfig struct {
 	ListenAddr  string
 	MetricsAddr string
+	EnablePprof bool
 	Log         *slog.Logger
 
 	DrainDuration            time.Duration
@@ -55,6 +57,11 @@ func New(cfg *HTTPServerConfig) (srv *Server, err error) {
 	mux.With(srv.httpLogger).Get("/drain", srv.handleDrain)
 	mux.With(srv.httpLogger).Get("/undrain", srv.handleUndrain)
 
+	if cfg.EnablePprof {
+		srv.log.Info("pprof API enabled")
+		mux.Mount("/debug", middleware.Profiler())
+	}
+
 	srv.srv = &http.Server{
 		Addr:         cfg.ListenAddr,
 		Handler:      mux,