feat: add support for profiling router

Author: efectn

config.go

@@ -65,6 +65,10 @@ type Config struct {
 	// If not set, it defaults to "warn".
 	LogLevel string `mapstructure:"logLevel"`
 
+	// EnableProfiler enables the pprof profiler on the server.
+	// This is not recommended for production use.
+	EnableProfiler bool `mapstructure:"enableProfiler"`
+
 	// ReloadFunc is called when a reload is done via http api.
 	ReloadFunc func()
 

redirector.go

@@ -1,16 +1,18 @@
 package redirector
 
 import (
+	"net/http"
+
 	"github.com/armbian/redirector/middleware"
-	"github.com/chi-middleware/logrus-logger"
+	logger "github.com/chi-middleware/logrus-logger"
 	"github.com/go-chi/chi/v5"
+	cm "github.com/go-chi/chi/v5/middleware"
 	lru "github.com/hashicorp/golang-lru"
 	"github.com/oschwald/maxminddb-golang"
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/prometheus/client_golang/prometheus/promauto"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
 	log "github.com/sirupsen/logrus"
-	"net/http"
 )
 
 var (
@@ -116,6 +118,11 @@ func (r *Redirector) Start() http.Handler {
 	router.Get("/geoip", r.geoIPHandler)
 	router.Get("/metrics", promhttp.Handler().ServeHTTP)
 
+	if r.config.EnableProfiler {
+		log.Warn("Enabling pprof profiler endpoints")
+		router.Mount("/debug", cm.Profiler())
+	}
+
 	router.NotFound(r.redirectHandler)
 
 	if r.config.BindAddress != "" {