Merge pull request #43 from application-research/add-debug-endpoints

Add debug endpoints

api/debug_profiler.go

@@ -0,0 +1,34 @@
+package api
+
+import (
+	httpprof "net/http/pprof"
+	"runtime/pprof"
+	"time"
+
+	"github.com/application-research/edge-ur/core"
+	"github.com/labstack/echo/v4"
+)
+
+// ConfigureDebugRouter It configures the router to handle requests for node profiling
+func ConfigureDebugProfileRouter(profilerGroup *echo.Group, node *core.LightNode) {
+
+	profilerGroup.GET("/pprof/:prof", func(c echo.Context) error {
+		httpprof.Handler(c.Param("prof")).ServeHTTP(c.Response().Writer, c.Request())
+		return nil
+	})
+
+	profilerGroup.GET("/cpuprofile", func(c echo.Context) error {
+		if err := pprof.StartCPUProfile(c.Response()); err != nil {
+			return err
+		}
+
+		defer pprof.StopCPUProfile()
+
+		select {
+		case <-c.Request().Context().Done():
+			return c.Request().Context().Err()
+		case <-time.After(time.Second * 30):
+		}
+		return nil
+	})
+}

api/router.go

@@ -67,6 +67,9 @@ func InitializeEchoRouterConfig(ln *core.LightNode) {
 	ConfigureHealthCheckRouter(defaultOpenRoute, ln)
 	ConfigureNodeInfoRouter(defaultOpenRoute, ln)
 
+	debugGroup := e.Group("/debug")
+	ConfigureDebugProfileRouter(debugGroup, ln)
+
 	apiGroup := e.Group("/api/v1")
 	apiGroup.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
 		return func(c echo.Context) error {