优化多角色数据权限匹配规则

RuoYi

2022-08-21 a36dc6635f92b1a6a5ea43c801f028dea08a7f0b

 src/main/java/com/ruoyi/framework/security/service/PermissionService.java

@@ -1,12 +1,12 @@
package com.ruoyi.framework.security.service;



import java.util.Set;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Service;

import org.springframework.util.CollectionUtils;

import com.ruoyi.common.utils.ServletUtils;

import com.ruoyi.common.utils.SecurityUtils;

import com.ruoyi.common.utils.StringUtils;

import com.ruoyi.framework.security.LoginUser;

import com.ruoyi.framework.security.context.PermissionContextHolder;

import com.ruoyi.project.system.domain.SysRole;



/**

@@ -27,9 +27,6 @@


    private static final String PERMISSION_DELIMETER = ",";



    @Autowired

    private TokenService tokenService;



    /**

     * 验证用户是否具备某权限

     * 

@@ -42,11 +39,12 @@
        {

            return false;

        }

        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());

        LoginUser loginUser = SecurityUtils.getLoginUser();

        if (StringUtils.isNull(loginUser) || CollectionUtils.isEmpty(loginUser.getPermissions()))

        {

            return false;

        }

        PermissionContextHolder.setContext(permission);

        return hasPermissions(loginUser.getPermissions(), permission);

    }



@@ -73,7 +71,7 @@
        {

            return false;

        }

        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());

        LoginUser loginUser = SecurityUtils.getLoginUser();

        if (StringUtils.isNull(loginUser) || CollectionUtils.isEmpty(loginUser.getPermissions()))

        {

            return false;

@@ -101,7 +99,7 @@
        {

            return false;

        }

        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());

        LoginUser loginUser = SecurityUtils.getLoginUser();

        if (StringUtils.isNull(loginUser) || CollectionUtils.isEmpty(loginUser.getUser().getRoles()))

        {

            return false;

@@ -109,7 +107,7 @@
        for (SysRole sysRole : loginUser.getUser().getRoles())

        {

            String roleKey = sysRole.getRoleKey();

            if (SUPER_ADMIN.contains(roleKey) || roleKey.contains(StringUtils.trim(role)))

            if (SUPER_ADMIN.equals(roleKey) || roleKey.equals(StringUtils.trim(role)))

            {

                return true;

            }

@@ -140,7 +138,7 @@
        {

            return false;

        }

        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());

        LoginUser loginUser = SecurityUtils.getLoginUser();

        if (StringUtils.isNull(loginUser) || CollectionUtils.isEmpty(loginUser.getUser().getRoles()))

        {

            return false;