go-zero 集成 Casbin 权限认证中间件实现

type AuthorityMiddleware struct {
    CasB *casbin.SyncedCachedEnforcer  // Casbin 同步缓存执行器
    Rds  *redis.Redis                   // Redis 客户端
}

func NewAuthorityMiddleware(CasB *casbin.SyncedCachedEnforcer, rds *redis.Redis) *AuthorityMiddleware {
    return &AuthorityMiddleware{
        CasB: CasB,
        Rds:  rds,
    }
}

func (m *AuthorityMiddleware) Handle(next http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        fmt.Println("========== AuthorityMiddleware:Handle ===========")
        path := r.URL.Path      // 获取请求路径
        method := r.Method      // 获取 HTTP 方法

        // 从 JWT 上下文中获取用户权限 ID
        authorityId := ctxJwt.GetJwtDataAuthorityId(r.Context())
        fmt.Println("authorityId", authorityId)

        // 执行批量权限检查
        result := batchCheck(m.CasB, strconv.FormatInt(authorityId, 10), path, method)
        if !result {
            logx.Errorf("---------- batchCheck: 不通过 ------------")
            return  // 权限不足，直接返回
        }
        next(w, r)  // 权限通过，继续执行后续处理
    }
}

func batchCheck(CasB *casbin.SyncedCachedEnforcer, authorityId, path, method string) bool {
    // 使用 Casbin 执行权限检查
    ok, _ := CasB.Enforce(authorityId, path, method)
    if !ok {
        logx.Errorf("---------- 权限不足 ------------")
        return false
    }
    return true
}