chore: implements use and uses

server.go

@@ -11,7 +11,6 @@ import (
 	"time"
 
 	"github.com/labstack/echo/v4"
-	"github.com/labstack/echo/v4/middleware"
 )
 
 // Kind represents the type of router group
@@ -134,9 +133,10 @@ func NewServer(opts ...Options) (*Server, error) {
 
 	e := echo.New()
 
-	e.Use(middleware.Logger())
-	e.Use(middleware.Recover())
-	e.Use(middleware.CORS())
+	// e.Use(middleware.Logger())
+	// e.Use(middleware.Recover())
+	// e.Use(middleware.CORS())
+
 	e.HideBanner = true
 
 	return &Server{
@@ -147,6 +147,14 @@ func NewServer(opts ...Options) (*Server, error) {
 	}, nil
 }
 
+func (s *Server) Use(middleware MiddlewareFunc) {
+	s.echo.Use(middleware)
+}
+
+func (s *Server) Uses(middlewares ...MiddlewareFunc) {
+	s.echo.Use(middlewares...)
+}
+
 // NewContext creates a new Echo context
 func (s *Server) NewContext(req *http.Request, w http.ResponseWriter) Context {
 	return s.echo.NewContext(req, w)

server_test.go

@@ -1131,3 +1131,65 @@ func TestRoutesWithMultiGroupWithBasicAuth(t *testing.T) {
 		})
 	}
 }
+
+func TestWithUse(t *testing.T) {
+	server, _ := NewServer()
+	rr := NewRouters()
+	rr.AddRouter("/test", map[string]HandlerFunc{
+		http.MethodGet: func(c Context) error {
+			return c.String(http.StatusOK, "test passed")
+		},
+	})
+
+	middlewareA := func(next echo.HandlerFunc) echo.HandlerFunc {
+		return func(c echo.Context) error {
+			return next(c)
+		}
+	}
+
+	_ = server.RegisterRouters(ROOT, rr)
+	server.Use(middlewareA)
+
+	e := server.GetEcho()
+
+	req := httptest.NewRequest(http.MethodGet, "/test", nil)
+	rec := httptest.NewRecorder()
+	e.ServeHTTP(rec, req)
+
+	assert.Equal(t, http.StatusOK, rec.Code)
+	assert.Equal(t, "test passed", rec.Body.String())
+}
+
+func TestWithUses(t *testing.T) {
+	server, _ := NewServer()
+	rr := NewRouters()
+	rr.AddRouter("/test", map[string]HandlerFunc{
+		http.MethodGet: func(c Context) error {
+			return c.String(http.StatusOK, "test passed")
+		},
+	})
+
+	middlewareA := func(next echo.HandlerFunc) echo.HandlerFunc {
+		return func(c echo.Context) error {
+			return next(c)
+		}
+	}
+
+	middlewareB := func(next echo.HandlerFunc) echo.HandlerFunc {
+		return func(c echo.Context) error {
+			return next(c)
+		}
+	}
+
+	_ = server.RegisterRouters(ROOT, rr)
+	server.Uses(middlewareA, middlewareB)
+
+	e := server.GetEcho()
+
+	req := httptest.NewRequest(http.MethodGet, "/test", nil)
+	rec := httptest.NewRecorder()
+	e.ServeHTTP(rec, req)
+
+	assert.Equal(t, http.StatusOK, rec.Code)
+	assert.Equal(t, "test passed", rec.Body.String())
+}