add

wangpengfei

2023-07-14 5f00d720ea52ababeb10954da4d33ab6c46e93b7

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
package middleware
 
import (
    "aps_crm/pkg/contextx"
    "aps_crm/pkg/ecode"
    "aps_crm/service"
    "aps_crm/utils"
    "strconv"
    "strings"
 
    "github.com/gin-gonic/gin"
)
 
var casbinService = service.ServiceGroup.CasbinService
 
// CasbinHandler 拦截器
func CasbinHandler() gin.HandlerFunc {
    return func(c *gin.Context) {
        waitUse, _ := utils.GetClaims(c)
        ctx := new(contextx.Context).SetCtx(c)
        //获取请求的PATH
        path := c.Request.URL.Path
        //obj := strings.TrimPrefix(path, global.GVA_CONFIG.System.RouterPrefix)
        obj := strings.TrimPrefix(path, "")
        // 获取请求方法
        act := c.Request.Method
        // 获取用户的角色
        sub := strconv.Itoa(int(waitUse.AuthorityId))
        e := casbinService.Casbin() // 判断策略中是否存在
        success, _ := e.Enforce(sub, obj, act)
        if !success {
            // response.FailWithDetailed(gin.H{}, "权限不足", c)
            c.Abort()
            ctx.Fail(ecode.NoPowerErr)
            return
        }
        c.Next()
    }
}