changes

liuhaonan

2022-10-21 f214709cf1d896e7c9743f206b7a708f020e5322

 ximon-admin/src/main/java/com/sandu/ximon/admin/service/RoleService.java

@@ -2,6 +2,7 @@

import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.system.UserInfo;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.github.pagehelper.PageHelper;
import com.sandu.common.enums.RoleLevelStatus;
@@ -12,6 +13,7 @@
import com.sandu.ximon.admin.param.RoleParam;
import com.sandu.ximon.admin.security.SecurityUtils;
import com.sandu.ximon.dao.domain.*;
import com.sandu.ximon.dao.enums.AdministratorEnums;
import com.sandu.ximon.dao.mapper.MenuMapper;
import com.sandu.ximon.dao.mapper.RoleMapper;
import lombok.AllArgsConstructor;
@@ -32,6 +34,7 @@
public class RoleService extends BaseServiceImpl<RoleMapper, Role> {

    private final AdminRoleRelationService adminRoleRelationService;
    private final ClientRoleRelationService clientRoleRelationService;
    private final RoleMenuRelationService roleMenuRelationService;
    private final MenuService menuService;
    private final MenuMapper menuMapper;
@@ -40,17 +43,21 @@
    /**
     * 获取指定管理员的spring security权限认证列表
     */
    public Collection<GrantedAuthority> mapToGrantedAuthorities(Long adminId) {

        // 获取管理员的角色id
        List<Role> roles = listByAdminId(adminId);
    public Collection<GrantedAuthority> mapToGrantedAuthorities(Long adminId, boolean flag) {
        List<Role> roles;
        if (flag) {
            // 获取管理员的角色id
            roles = listByAdminId(adminId);
        } else {
            roles = listByClientId(adminId);
        }

        if (CollectionUtil.isEmpty(roles)) {
            throw new BusinessException("当前用户没有角色");
        }
        List<Long> roleIdList = roles.stream().map(Role::getId).collect(Collectors.toList());
        List<Long> menuIdList = roleMenuRelationService.list(Wrappers.lambdaQuery(RoleMenuRelation.class).in(RoleMenuRelation::getRoleId, roleIdList).select(RoleMenuRelation::getMenuId))
                .stream().map(RoleMenuRelation::getMenuId).distinct().collect(Collectors.toList());
        List<Long> menuIdList = roleMenuRelationService.list(Wrappers.lambdaQuery(RoleMenuRelation.class).in(RoleMenuRelation::getRoleId, roleIdList)
                        .select(RoleMenuRelation::getMenuId)).stream().map(RoleMenuRelation::getMenuId).distinct().collect(Collectors.toList());

        Set<String> menuPermissions = new HashSet<>();
        // 菜单权限
@@ -74,8 +81,22 @@
     */
    public List<Role> listByAdminId(Long adminId) {
        // 获取管理员的角色id
        List<AdminRoleRelation> list = adminRoleRelationService.list(Wrappers.lambdaQuery(AdminRoleRelation.class).eq(AdminRoleRelation::getAdminId, adminId).select(AdminRoleRelation::getRoleId));
        List<AdminRoleRelation> list = adminRoleRelationService.list(Wrappers.lambdaQuery(AdminRoleRelation.class)
                .eq(AdminRoleRelation::getAdminId, adminId).select(AdminRoleRelation::getRoleId));
        List<Long> roleIdList = list.stream().map(AdminRoleRelation::getRoleId).collect(Collectors.toList());
        if (CollectionUtil.isEmpty(roleIdList)) {
            return null;
        }
        return listByIds(roleIdList);
    }

    /**
     * 获取某个普通用户的角色列表
     */
    public List<Role> listByClientId(Long clientId) {
        // 获取普通用户的角色id
        List<ClientRoleRelation> list = clientRoleRelationService.list(Wrappers.lambdaQuery(ClientRoleRelation.class).eq(ClientRoleRelation::getClientId, clientId).select(ClientRoleRelation::getRoleId));
        List<Long> roleIdList = list.stream().map(ClientRoleRelation::getRoleId).collect(Collectors.toList());
        if (CollectionUtil.isEmpty(roleIdList)) {
            return null;
        }
@@ -90,7 +111,23 @@
        Role role = new Role();
        role.setName(param.getName());
        role.setRemark(param.getRemark());
        role.setLevel(RoleLevelStatus.COMMON.getCode());
        //设置角色权限等级
        if (param.getLevel() != null) {
            if (param.getLevel().equals(RoleLevelStatus.SUPER.getCode())) {
                throw new BusinessException("无法添加超级管理员角色");
            } else if (param.getLevel().equals(RoleLevelStatus.COMMON.getCode())) {
                role.setLevel(RoleLevelStatus.COMMON.getCode());
            } else if (param.getLevel().equals(RoleLevelStatus.NORMAL.getCode())) {
                role.setLevel(RoleLevelStatus.NORMAL.getCode());
            } else if (param.getLevel().equals(RoleLevelStatus.NORMAL2.getCode())) {
                role.setLevel(RoleLevelStatus.NORMAL2.getCode());
            } else {
                role.setLevel(RoleLevelStatus.NORMAL.getCode());
            }
        } else {
            role.setLevel(RoleLevelStatus.NORMAL.getCode());
        }

        if (!save(role)) {
            throw new BusinessException("添加角色失败");
        }
@@ -106,12 +143,35 @@
        if (role == null) {
            throw new BusinessException("找不到角色信息");
        }
        //判断是有权限修改
        assertLevels(role.getLevel());
        Role update = new Role();
        update.setId(roleId);
        update.setName(param.getName());
        update.setRemark(param.getRemark());
        if (!updateById(update)) {

        role.setName(param.getName());
        role.setRemark(param.getRemark());

        if (!role.getLevel().equals(RoleLevelStatus.SUPER.getCode())) {
            //设置角色权限等级
            if (param.getLevel() != null) {
                if (param.getLevel().equals(RoleLevelStatus.SUPER.getCode())) {
                    throw new BusinessException("无法修改成超级管理员角色");
                } else if (param.getLevel().equals(RoleLevelStatus.COMMON.getCode())) {
                    role.setLevel(RoleLevelStatus.COMMON.getCode());
                } else if (param.getLevel().equals(RoleLevelStatus.NORMAL.getCode())) {
                    role.setLevel(RoleLevelStatus.NORMAL.getCode());
                } else if (param.getLevel().equals(RoleLevelStatus.NORMAL2.getCode())) {
                    role.setLevel(RoleLevelStatus.NORMAL2.getCode());
                } else {
                    role.setLevel(RoleLevelStatus.NORMAL.getCode());
                }
            } else {
                role.setLevel(RoleLevelStatus.NORMAL.getCode());
            }
        } else if (!param.getLevel().equals(RoleLevelStatus.SUPER.getCode())) {
            throw new BusinessException("超级管理员角色等级无法修改！");
        }


        if (!updateById(role)) {
            throw new BusinessException("编辑角色失败");
        }
        roleMenuRelationService.remove(Wrappers.lambdaQuery(RoleMenuRelation.class).eq(RoleMenuRelation::getRoleId, roleId));
@@ -149,7 +209,7 @@
            throw new BusinessException("找不到角色");
        }
        //超级管理员不能删除
        if(RoleLevelStatus.SUPER.getCode().equals(role.getLevel())){
        if (RoleLevelStatus.SUPER.getCode().equals(role.getLevel())) {
            throw new BusinessException("当前角色为超级管理员，无法删除");
        }

@@ -157,6 +217,10 @@
        List<AdminRoleRelation> list = adminRoleRelationService.list(Wrappers.lambdaQuery(AdminRoleRelation.class).eq(AdminRoleRelation::getRoleId, role));
        if (CollectionUtil.isNotEmpty(list)) {
            throw new BusinessException("当前角色有管理员使用，无法删除");
        }
        List<ClientRoleRelation> listClient = clientRoleRelationService.list(Wrappers.lambdaQuery(ClientRoleRelation.class).eq(ClientRoleRelation::getRoleId, role));
        if (CollectionUtil.isNotEmpty(listClient)) {
            throw new BusinessException("当前角色有用户使用，无法删除");
        }
        roleMenuRelationService.remove(Wrappers.lambdaQuery(RoleMenuRelation.class).eq(RoleMenuRelation::getRoleId, roleId));
        return removeById(roleId);
@@ -170,13 +234,6 @@
        }

        List<RoleDetail> list = baseMapper.listRole(roleIdList);
        for (RoleDetail roleDetail : list) {
            if (roleDetail.getId() == 1) {
                roleDetail.setMenuIdList(menuMapper.listMenu());

                break;
            }
        }

        return list;
    }