diff --git a/internal/apiserver/service/router.go b/internal/apiserver/service/router.go index 263c9392..508abf77 100644 --- a/internal/apiserver/service/router.go +++ b/internal/apiserver/service/router.go @@ -329,6 +329,10 @@ func (s *APIServer) initRouter() error { SqlWorkbenchService: s.SqlWorkbenchController.SqlWorkbenchService, })) + sqlWorkbenchV1.Use(sqlWorkbenchService.GetPrivilegeDeniedMiddleware(sqlWorkbenchService.PrivilegeDeniedMiddlewareConfig{ + SqlWorkbenchService: s.SqlWorkbenchController.SqlWorkbenchService, + })) + sqlWorkbenchV1.Use(s.SqlWorkbenchController.SqlWorkbenchService.AuditMiddleware()) sqlWorkbenchV1.Use(middleware.ProxyWithConfig(middleware.ProxyConfig{ Skipper: middleware.DefaultSkipper, diff --git a/internal/sql_workbench/service/privilege_denied_middleware_ce.go b/internal/sql_workbench/service/privilege_denied_middleware_ce.go new file mode 100644 index 00000000..4fc4d03e --- /dev/null +++ b/internal/sql_workbench/service/privilege_denied_middleware_ce.go @@ -0,0 +1,18 @@ +//go:build !enterprise +// +build !enterprise + +package sql_workbench + +import "github.com/labstack/echo/v4" + +// PrivilegeDeniedMiddlewareConfig 非 enterprise 构建占位。 +type PrivilegeDeniedMiddlewareConfig struct { + SqlWorkbenchService *SqlWorkbenchService +} + +// GetPrivilegeDeniedMiddleware 非 enterprise 构建下为空操作。 +func GetPrivilegeDeniedMiddleware(_ PrivilegeDeniedMiddlewareConfig) echo.MiddlewareFunc { + return func(next echo.HandlerFunc) echo.HandlerFunc { + return next + } +}