diff --git a/internal/api/server.go b/internal/api/server.go index 8e22742bb..5f98485b9 100644 --- a/internal/api/server.go +++ b/internal/api/server.go @@ -32,7 +32,7 @@ import ( type Server interface { AttachHandler(method string, path string, handler gin.HandlerFunc) - // AttachMiddleware(handler gin.HandlerFunc) + AttachMiddleware(handler gin.HandlerFunc) GetAPIGroup() *gin.RouterGroup Start() Stop() @@ -71,6 +71,10 @@ func (s *server) AttachHandler(method string, path string, handler gin.HandlerFu } } +func (s *server) AttachMiddleware(middleware gin.HandlerFunc) { + s.engine.Use(middleware) +} + func New(config *config.Config, logger *logrus.Logger) Server { engine := gin.New() store := memstore.NewStore([]byte("authentication-key"), []byte("encryption-keyencryption-key----"))