用户管理新增分配角色功能

RuoYi

2021-07-01 ef5197f03455a0ee823dbadb685bf979c5fa3443

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

@@ -56,6 +56,31 @@
    }



    /**

     * 根据用户ID查询角色

     * 
     * @param userId 用户ID

     * @return 角色列表

     */

    @Override

    public List<SysRole> selectRolesByUserId(Long userId)

    {

        List<SysRole> userRoles = roleMapper.selectRolePermissionByUserId(userId);

        List<SysRole> roles = selectRoleAll();

        for (SysRole role : roles)

        {

            for (SysRole userRole : userRoles)

            {

                if (role.getRoleId().longValue() == userRole.getRoleId().longValue())

                {

                    role.setFlag(true);

                    break;

                }

            }

        }

        return roles;

    }



    /**

     * 根据用户ID查询权限

     * 

     * @param userId 用户ID

@@ -290,8 +315,13 @@
     * @return 结果

     */

    @Override

    @Transactional

    public int deleteRoleById(Long roleId)

    {

        // 删除角色与菜单关联

        roleMenuMapper.deleteRoleMenuByRoleId(roleId);

        // 删除角色与部门关联

        roleDeptMapper.deleteRoleDeptByRoleId(roleId);

        return roleMapper.deleteRoleById(roleId);

    }



@@ -302,6 +332,7 @@
     * @return 结果

     */

    @Override

    @Transactional

    public int deleteRoleByIds(Long[] roleIds)

    {

        for (Long roleId : roleIds)

@@ -313,6 +344,10 @@
                throw new CustomException(String.format("%1$s已分配,不能删除", role.getRoleName()));

            }

        }

        // 删除角色与菜单关联

        roleMenuMapper.deleteRoleMenu(roleIds);

        // 删除角色与部门关联

        roleDeptMapper.deleteRoleDept(roleIds);

        return roleMapper.deleteRoleByIds(roleIds);

    }

}