#feat .

yanghui

2022-11-28 42521c9b51c1bd9a40036f18fd610cd46601222d

 springcloud_k8s_panzhihuazhihuishequ/service_user/src/main/java/com/panzhihua/service_user/service/impl/RoleServiceImpl.java

@@ -1,11 +1,15 @@
package com.panzhihua.service_user.service.impl;

import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;

import javax.annotation.Resource;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import org.springframework.beans.BeanUtils;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Service;
@@ -30,6 +34,7 @@
import com.panzhihua.service_user.service.RoleService;

import lombok.extern.slf4j.Slf4j;
import org.springframework.util.StringUtils;

/**
 * @program: springcloud_k8s_panzhihuazhihuishequ
@@ -105,19 +110,14 @@
     * @return 角色集合
     */
    @Override
    public R listRoleBackstage(Long communityId) {
        List<SysRoleDO> sysRoleDOS =
            roleDAO.selectList(new QueryWrapper<SysRoleDO>().lambda().eq(SysRoleDO::getCommunityId, communityId));
        if (ObjectUtils.isEmpty(sysRoleDOS)) {
            return R.fail("角色列表为空");
    public R listRoleBackstage(Long communityId,Integer pageNum,Integer pageSize,String areaCode) {
        LambdaQueryWrapper<SysRoleDO> queryWrapper=new QueryWrapper<SysRoleDO>().lambda().eq(SysRoleDO::getCommunityId, communityId);
        if(com.panzhihua.common.utlis.StringUtils.isNotEmpty(areaCode)){
            queryWrapper.eq(SysRoleDO::getAreaCode,areaCode);
        }
        List<RoleVO> roleVOS = new ArrayList<>();
        sysRoleDOS.forEach(sysRoleDO -> {
            RoleVO roleVO = new RoleVO();
            BeanUtils.copyProperties(sysRoleDO, roleVO);
            roleVOS.add(roleVO);
        });
        return R.ok(roleVOS);
        queryWrapper.orderByDesc(SysRoleDO::getCreateAt);
        IPage<SysRoleDO> sysRoleDOIPage=this.roleDAO.selectPage(new Page<>(pageNum,pageSize),queryWrapper);
        return R.ok(sysRoleDOIPage);
    }

    /**
@@ -137,9 +137,7 @@
        } catch (Exception e) {
            e.printStackTrace();
            log.error("新增角色失败，失败原因【{}】", e.getMessage());
            if (e.getMessage().contains("unique_role_name_community_id")) {
                return R.fail("角色已经存在");
            }
        }
        if (insert > 0) {
            SysRoleDO roleDO = roleDAO.selectOne(new QueryWrapper<SysRoleDO>().lambda()
@@ -178,7 +176,13 @@
     * @return 菜单集合
     */
    @Override
    public R listMenuBackstage(Long communityId) {
    public R listMenuBackstage(Long communityId,String appId) {
        if("wx118de8a734d269f0".equals(appId)){
            communityId=2L;
        }
        else {
            communityId=1552661941170868228L;
        }
        List<SysMenuVO> sysMenuVOS = new ArrayList<>();
        List<SysMenuDO> sysMenuDOS =
            sysMenuDAO.selectList(new QueryWrapper<SysMenuDO>().lambda().eq(SysMenuDO::getParentId, 0)
@@ -195,6 +199,15 @@
                    sysMenuDOS1.forEach(sysMenuDO1 -> {
                        SysMenuVO sysMenuVO1 = new SysMenuVO();
                        BeanUtils.copyProperties(sysMenuDO1, sysMenuVO1);
                        List<SysMenuDO> sysMenuDOList = sysMenuDAO.selectList(new QueryWrapper<SysMenuDO>().lambda()
                                .eq(SysMenuDO::getParentId, sysMenuDO1.getMenuId()).orderByAsc(SysMenuDO::getOrderNum));
                        List<SysMenuVO> sysMenuVOList=new ArrayList<>();
                        sysMenuDOList.forEach(sysMenuDO2 -> {
                            SysMenuVO sysMenuVO2=new SysMenuVO();
                            BeanUtils.copyProperties(sysMenuDO2,sysMenuVO2);
                            sysMenuVOList.add(sysMenuVO2);
                        });
                        sysMenuVO1.setSysMenuVOList(sysMenuVOList);
                        sysMenuVOS1.add(sysMenuVO1);
                    });
                    sysMenuVO.setSysMenuVOList(sysMenuVOS1);
@@ -217,6 +230,12 @@
    public R putMenuRole(MenuRoleVO menuRoleVO) {
        List<Long> menuIds = new ArrayList<>();
        Long roleId = menuRoleVO.getRoleId();
        if(!StringUtils.isEmpty(menuRoleVO.getRoleName())){
            SysRoleDO sysRoleDO=new SysRoleDO();
            sysRoleDO.setRoleId(roleId);
            sysRoleDO.setRoleName(menuRoleVO.getRoleName());
            this.roleDAO.updateById(sysRoleDO);
        }
        // 删除旧的
        sysRoleMenuDAO.delete(new QueryWrapper<SysRoleMenuDO>().lambda().eq(SysRoleMenuDO::getRoleId, roleId));
        // 创建新的
@@ -224,7 +243,7 @@
        if (isAll.intValue() == 1) {
            menuIds = sysMenuDAO
                .selectList(
                    new QueryWrapper<SysMenuDO>().lambda().eq(SysMenuDO::getCommunityId, menuRoleVO.getCommunityId()))
                    new QueryWrapper<SysMenuDO>().lambda().eq(SysMenuDO::getCommunityId, 2))
                .stream().map(sysMenuDO -> sysMenuDO.getMenuId()).collect(Collectors.toList());
        } else {
            menuIds = menuRoleVO.getMenuIds();
@@ -263,26 +282,30 @@
        List<Long> collect =
            sysRoleMenuDOS.stream().map(sysRoleMenuDO -> sysRoleMenuDO.getMenuId()).collect(Collectors.toList());
        List<SysMenuDO> sysMenuDOS =
            sysMenuDAO.selectList(new QueryWrapper<SysMenuDO>().lambda().in(SysMenuDO::getMenuId, collect));
            sysMenuDAO.selectList(new QueryWrapper<SysMenuDO>().lambda().in(SysMenuDO::getMenuId, collect).eq(SysMenuDO::getParentId,0));
        List<SysMenuVO> sysMenuVOS = new ArrayList<>();
        sysMenuDOS.forEach(sysMenuDO -> {
            SysMenuVO sysMenuVO = new SysMenuVO();
            List<SysMenuVO> menuVOS=new ArrayList<>();
            BeanUtils.copyProperties(sysMenuDO, sysMenuVO);
            List<SysMenuDO> sysMenuDOList=sysMenuDAO.selectList(new QueryWrapper<SysMenuDO>().lambda().eq(SysMenuDO::getParentId,sysMenuDO.getMenuId()).in(SysMenuDO::getMenuId, collect));
            sysMenuDOList.forEach(sysMenuDO1->{
                SysMenuVO sysMenuVO1 = new SysMenuVO();
                List<SysMenuVO> sysMenuVOList=new ArrayList<>();
                BeanUtils.copyProperties(sysMenuDO1, sysMenuVO1);
                List<SysMenuDO> sysMenuDOList1=sysMenuDAO.selectList(new QueryWrapper<SysMenuDO>().lambda().eq(SysMenuDO::getParentId,sysMenuDO1.getMenuId()).in(SysMenuDO::getMenuId, collect));
                sysMenuDOList1.forEach(sysMenuDO2 -> {
                    SysMenuVO sysMenuVO2=new SysMenuVO();
                    BeanUtils.copyProperties(sysMenuDO2,sysMenuVO2);
                    sysMenuVOList.add(sysMenuVO2);
                });
                sysMenuVO1.setSysMenuVOList(sysMenuVOList.stream().sorted(Comparator.comparing(SysMenuVO::getOrderNum)).collect(Collectors.toList()));
                menuVOS.add(sysMenuVO1);
            });
            sysMenuVO.setSysMenuVOList(menuVOS.stream().sorted(Comparator.comparing(SysMenuVO::getOrderNum)).collect(Collectors.toList()));
            sysMenuVOS.add(sysMenuVO);
        });
        List<SysMenuVO> collect1 =
            sysMenuVOS.stream().filter(sysMenuVO -> sysMenuVO.getParentId() == 0).collect(Collectors.toList());
        if (!ObjectUtils.isEmpty(collect1)) {
            collect1.forEach(sysMenuVO -> {
                Long menuId = sysMenuVO.getMenuId();
                List<SysMenuVO> sysMenuVOS1 =
                    sysMenuVOS.stream().filter(sysMenuVO1 -> sysMenuVO1.getParentId().intValue() == menuId.intValue())
                        .collect(Collectors.toList());
                sysMenuVO.setSysMenuVOList(sysMenuVOS1);
            });
        }

        return R.ok(collect1);
        return R.ok(sysMenuVOS.stream().sorted(Comparator.comparing(SysMenuVO::getOrderNum)).collect(Collectors.toList()));
    }

    /**
@@ -294,22 +317,26 @@
     */
    @Override
    public R listMenuBackstageByRole(Long roleId) {
        List<SysMenuVO> sysMenuVOS = new ArrayList<>();
        List<SysRoleMenuDO> sysRoleMenuDOS =
            sysRoleMenuDAO.selectList(new QueryWrapper<SysRoleMenuDO>().lambda().eq(SysRoleMenuDO::getRoleId, roleId));
        if (!ObjectUtils.isEmpty(sysRoleMenuDOS)) {
            List<Long> longs =
                sysRoleMenuDOS.stream().map(sysRoleMenuDO -> sysRoleMenuDO.getMenuId()).collect(Collectors.toList());
            List<SysMenuDO> sysMenuDOS =
                sysMenuDAO.selectList(new QueryWrapper<SysMenuDO>().lambda().in(SysMenuDO::getMenuId, longs));
            if (!ObjectUtils.isEmpty(sysMenuDOS)) {
                sysMenuDOS.forEach(sysMenuDO -> {
                    SysMenuVO sysMenuVO1 = new SysMenuVO();
                    BeanUtils.copyProperties(sysMenuDO, sysMenuVO1);
                    sysMenuVOS.add(sysMenuVO1);
            List<SysMenuDO> sysMenuDOList=this.sysMenuDAO.selectList(new QueryWrapper<SysMenuDO>().lambda().in(SysMenuDO::getMenuId,longs).eq(SysMenuDO::getParentId,0));
            sysMenuDOList.forEach(sysMenuDO -> {
                List<SysMenuDO> sysMenuDOS=this.sysMenuDAO.selectList(new QueryWrapper<SysMenuDO>().lambda().in(SysMenuDO::getMenuId,longs).eq(SysMenuDO::getParentId,sysMenuDO.getMenuId()));
                sysMenuDOS.forEach(sysMenuDO1 -> {
                    List<SysMenuDO> sysMenuDOS1=this.sysMenuDAO.selectList(new QueryWrapper<SysMenuDO>().lambda().in(SysMenuDO::getMenuId,longs).eq(SysMenuDO::getParentId,sysMenuDO1.getMenuId()));
                    if(!sysMenuDOS1.isEmpty()){
                        longs.remove(sysMenuDO1.getMenuId());
                    }
                });
            }
                if(!sysMenuDOS.isEmpty()){
                    longs.remove(sysMenuDO.getMenuId());
                }
            });
            return R.ok(longs);
        }
        return R.ok(sysMenuVOS);
        return R.fail();
    }
}