1 files changed, 38 insertions, 0 deletions

diff --git a/echo/middleware/pprof.go b/echo/middleware/pprof.go
new file mode 100644
index 0000000..cc69505
--- /dev/null
+++ b/echo/middleware/pprof.go
@@ -0,0 +1,38 @@
+package middleware
+
+import (
+        "net/http"
+        "net/http/pprof"
+
+        "github.com/labstack/echo/v4"
+)
+
+func RegisterPprof(e *echo.Echo, prefixOptions ...string) {
+        prefix := "/debug/pprof"
+        if len(prefixOptions) > 0 {
+                prefix = prefixOptions[0]
+        }
+
+        prefixRouter := e.Group(prefix)
+        {
+                prefixRouter.GET("/", pprofAdapter(pprof.Index))
+                prefixRouter.GET("/allocs", pprofAdapter(pprof.Handler("allocs").ServeHTTP))
+                prefixRouter.GET("/block", pprofAdapter(pprof.Handler("block").ServeHTTP))
+                prefixRouter.GET("/cmdline", pprofAdapter(pprof.Cmdline))
+                prefixRouter.GET("/goroutine", pprofAdapter(pprof.Handler("goroutine").ServeHTTP))
+                prefixRouter.GET("/heap", pprofAdapter(pprof.Handler("heap").ServeHTTP))
+                prefixRouter.GET("/mutex", pprofAdapter(pprof.Handler("mutex").ServeHTTP))
+                prefixRouter.GET("/profile", pprofAdapter(pprof.Profile))
+                prefixRouter.POST("/symbol", pprofAdapter(pprof.Symbol))
+                prefixRouter.GET("/symbol", pprofAdapter(pprof.Symbol))
+                prefixRouter.GET("/threadcreate", pprofAdapter(pprof.Handler("threadcreate").ServeHTTP))
+                prefixRouter.GET("/trace", pprofAdapter(pprof.Trace))
+        }
+}
+
+func pprofAdapter(h http.HandlerFunc) echo.HandlerFunc {
+        return func(c echo.Context) error {
+                h.ServeHTTP(c.Response().Writer, c.Request())
+                return nil
+        }
+}