echo: allow trace without debugecho/v0.16.2

1 files changed, 5 insertions, 4 deletions

diff --git a/echo/echo_default.go b/echo/echo_default.go
index b6d4fa2..dadeb90 100644
--- a/echo/echo_default.go
+++ b/echo/echo_default.go
@@ -72,6 +72,7 @@ type EchoConfig struct {
         DisableHttp3          bool
         PrometheusConfig      *prometheus.PrometheusConfig
         CORSConfig            *middleware.CORSConfig
+        EnableTrace           bool
 }
 
 type EchoWrapper struct {
@@ -92,10 +93,6 @@ func NewEchoWrapper(ctx context.Context, debug bool) (*EchoWrapper, error) {
         e := echo.New()
         e.Debug = debug
 
-        if e.Debug {
-                glmw.RegisterPprof(e)
-        }
-
         e.Logger.SetLevel(log.INFO)
         if debug {
                 e.Logger.SetLevel(log.DEBUG)
@@ -166,6 +163,10 @@ func (w *EchoWrapper) AddErrorHandler(h ContentErrorHandler, mime ...string) {
 }
 
 func (w *EchoWrapper) Configure(c EchoConfig) (err error) {
+        if w.Debug || c.EnableTrace {
+                glmw.RegisterPprof(w.Echo)
+        }
+
         w.configureAutocert(&c)
 
         if err := w.configureIpExtractor(&c); err != nil {