From 06000a6dace3863ddd28aba037d416cdba2ebfd6 Mon Sep 17 00:00:00 2001 From: zhangwenjian Date: Mon, 31 May 2021 18:11:19 +0800 Subject: [PATCH] =?UTF-8?q?perf=F0=9F=91=8C=20runtime=E6=8E=A5=E7=AE=A1=20?= =?UTF-8?q?=E4=B8=AD=E9=97=B4=E4=BB=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- common/middleware/init.go | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/common/middleware/init.go b/common/middleware/init.go index 5f66019a..a434bbe4 100644 --- a/common/middleware/init.go +++ b/common/middleware/init.go @@ -2,6 +2,15 @@ package middleware import ( "github.com/gin-gonic/gin" + "github.com/go-admin-team/go-admin-core/sdk" + jwt "github.com/go-admin-team/go-admin-core/sdk/pkg/jwtauth" + "go-admin/common/actions" +) + +const ( + JwtTokenCheck string = "JwtToken" + RoleCheck string = "AuthCheckRole" + PermissionCheck string = "PermissionAction" ) func InitMiddleware(r *gin.Engine) { @@ -19,4 +28,7 @@ func InitMiddleware(r *gin.Engine) { r.Use(Secure) // 链路追踪 //r.Use(middleware.Trace()) -} \ No newline at end of file + sdk.Runtime.SetMiddleware(JwtTokenCheck, (*jwt.GinJWTMiddleware).MiddlewareFunc) + sdk.Runtime.SetMiddleware(RoleCheck, AuthCheckRole()) + sdk.Runtime.SetMiddleware(PermissionCheck, actions.PermissionAction()) +}