代码优化

稚屿

2022-02-09 c99eb98001e22f671ce514268231edcd27fc73ef

 ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysRoleServiceImpl.java

@@ -11,7 +11,9 @@
import com.ruoyi.common.annotation.DataScope;

import com.ruoyi.common.constant.UserConstants;

import com.ruoyi.common.core.domain.entity.SysRole;

import com.ruoyi.common.core.domain.entity.SysUser;

import com.ruoyi.common.exception.ServiceException;

import com.ruoyi.common.utils.SecurityUtils;

import com.ruoyi.common.utils.StringUtils;

import com.ruoyi.common.utils.spring.SpringUtils;

import com.ruoyi.system.domain.SysRoleDept;

@@ -120,7 +122,7 @@
     * @return 选中角色ID列表

     */

    @Override

    public List<Integer> selectRoleListByUserId(Long userId)

    public List<Long> selectRoleListByUserId(Long userId)

    {

        return roleMapper.selectRoleListByUserId(userId);

    }

@@ -184,6 +186,26 @@
        if (StringUtils.isNotNull(role.getRoleId()) && role.isAdmin())

        {

            throw new ServiceException("不允许操作超级管理员角色");

        }

    }



    /**

     * 校验角色是否有数据权限

     * 
     * @param roleId 角色id

     */

    @Override

    public void checkRoleDataScope(Long roleId)

    {

        if (!SysUser.isAdmin(SecurityUtils.getUserId()))

        {

            SysRole role = new SysRole();

            role.setRoleId(roleId);

            List<SysRole> roles = SpringUtils.getAopProxy(this).selectRoleList(role);

            if (StringUtils.isEmpty(roles))

            {

                throw new ServiceException("没有权限访问角色数据！");

            }

        }

    }



@@ -339,6 +361,7 @@
        for (Long roleId : roleIds)

        {

            checkRoleAllowed(new SysRole(roleId));

            checkRoleDataScope(roleId);

            SysRole role = selectRoleById(roleId);

            if (countUserRoleByRoleId(roleId) > 0)

            {