4dkankan-center-manage/src/main/java/com/fdkankan/manage/service/impl/SysMenuServiceImpl.java

package com.fdkankan.manage.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.fdkankan.common.request.RequestBase;
import com.fdkankan.common.response.PageInfo;
import com.fdkankan.manage.entity.Meta;
import com.fdkankan.manage.entity.SysMenu;
import com.fdkankan.manage.entity.SysRoleMenu;
import com.fdkankan.manage.entity.SysUser;
import com.fdkankan.manage.mapper.ISysMenuMapper;
import com.fdkankan.manage.service.ISysMenuService;
import com.fdkankan.manage.service.ISysRoleMenuService;
import com.fdkankan.manage.service.ISysUserService;
import com.fdkankan.manage.vo.response.SysMenuVo;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

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

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author 
 * @since 2022-06-09
 */
@Service
public class SysMenuServiceImpl extends ServiceImpl<ISysMenuMapper, SysMenu> implements ISysMenuService {

    @Autowired
    ISysRoleMenuService roleMenuService;
    @Autowired
    ISysUserService userService;


    @Override
    public List<SysMenu> getListByUserId(Object loginId) {
        return getBaseMapper().getListByUserId(loginId);
    }

    @Override
    public PageInfo pageList(RequestBase param) {
        LambdaQueryWrapper<SysMenu> wrapper = new LambdaQueryWrapper<>();
        wrapper.isNull(SysMenu::getParentId);
        wrapper.orderByAsc(SysMenu::getSort);
        Page<SysMenu> page = this.page(new Page<>(param.getPageNum(),param.getPageSize()), wrapper);
       // List<SysMenuVo> voList = getByParentId(null,page.getRecords(),false,null);

        List<SysMenuVo> allVos = setTreeMenuVo(false,null);
        List<Long> ids  = page.getRecords().parallelStream().map(SysMenu::getId).collect(Collectors.toList());
        List<SysMenuVo> voList = allVos.parallelStream().filter(vo -> ids.contains(vo.getId())).collect(Collectors.toList());

        Page<SysMenuVo> resultPage = new Page<>(page.getCurrent(),page.getSize());
        resultPage.setRecords(voList);
        resultPage.setTotal(page.getTotal());
        return PageInfo.PageInfo(resultPage);
    }

    @Override
    public List<SysMenuVo> allShowList() {
        return setTreeMenuVo(true,null);
    }

    @Override
    public List<SysMenuVo> getByUserId(Long userId) {
        SysUser user = userService.getById(userId);
        List<SysMenu> menuList  = this.getListByUserId(userId);
        List<Long> menuIds = menuList.parallelStream().map(SysMenu::getId).collect(Collectors.toList());
        menuList.forEach(sysMenu -> setMenuIds(menuIds,sysMenu));
        return getByRoleId(menuIds,user.getRoleId());
    }


    private void setMenuIds( List<Long> menuIds ,SysMenu sysMenu){
        if(sysMenu.getParentId()!= null && !menuIds.contains(sysMenu.getParentId())) {
            menuIds.add(sysMenu.getParentId());
            SysMenu menu = this.getById(sysMenu.getParentId());
            this.setMenuIds(menuIds,menu);
        }
    }

    @Override
    public List<SysMenuVo> getByRoleId(List<Long> menuIds ,Long roleId) {
        if(menuIds == null){
            List<SysRoleMenu> roleMenus  = roleMenuService.getByRoleId(roleId);
            if(roleMenus == null || roleMenus.size() <=0){
                return new ArrayList<>();
            }
            menuIds = roleMenus.parallelStream().map(SysRoleMenu::getMenuId).collect(Collectors.toList());
        }
        return setTreeMenuVo(true,menuIds);
    }

    //递归设置menuTree
    private List<SysMenuVo> getByParentId(Long parentId, List<SysMenu> list, boolean show ,List<Long> menuIds){
        if(list == null){
            LambdaQueryWrapper<SysMenu> wrapper = new LambdaQueryWrapper<>();
            if(show){
                wrapper.eq(SysMenu::getIsShow,1);
            }
            if(parentId == null){
                wrapper.isNull(SysMenu::getParentId);
            }else {
                wrapper.eq(SysMenu::getParentId,parentId);
            }
            if(menuIds != null){
                wrapper.in(SysMenu::getId,menuIds);
            }
            list = this.list(wrapper);
        }
        List<SysMenuVo> sysMenuVos = new ArrayList<>();
        if(list.size() <=0){
            return sysMenuVos;
        }
        for (SysMenu record : list) {
            SysMenuVo vo = new SysMenuVo();
            BeanUtils.copyProperties(record,vo);
            sysMenuVos.add(vo);
            List<SysMenuVo> children = getByParentId(record.getId(),null,show,menuIds);
            vo.setChildren(children);
        }
        return sysMenuVos;
    }

    //查询到全部list ,根据parentId组装树形结构
    private List<SysMenuVo> setTreeMenuVo( boolean show ,List<Long> menuIds){
        LambdaQueryWrapper<SysMenu> wrapper = new LambdaQueryWrapper<>();
        if(show){
            wrapper.eq(SysMenu::getIsShow,1);
        }
        if(menuIds != null){
            wrapper.in(SysMenu::getId,menuIds);
        }
        List<SysMenu> list = this.list(wrapper);

        List<SysMenuVo> voList = new ArrayList<>();

        HashMap<Long,SysMenuVo> voMap = new HashMap<>();
        for (SysMenu sysMenu : list) {
            SysMenuVo menuVo = new SysMenuVo();
            BeanUtils.copyProperties(sysMenu,menuVo);
            Meta meta = new Meta();
            meta.setIcon(sysMenu.getIcon());
            meta.setTitle(sysMenu.getName());
            meta.setHideMenu(sysMenu.getHideMenu() == 1);
            meta.setFrameSrc(null);
            menuVo.setMeta(meta);
            voMap.put(sysMenu.getId(),menuVo);
        }
        for (SysMenu sysMenu : list) {
            if(sysMenu.getParentId() !=null){
                SysMenuVo menuVo = voMap.get(sysMenu.getId());
                List<SysMenuVo> children = voMap.get(sysMenu.getParentId()).getChildren();
                children.add(menuVo);
            }else {
                voList.add(voMap.get(sysMenu.getId()));
            }
        }
        setChildrenNull(voList);
        return voList;
    }

    private void setChildrenNull(List<SysMenuVo> voList){
        if(voList == null || voList.size() ==0){
            return ;
        }
        for (SysMenuVo menuVo : voList) {
            if(menuVo.getChildren()!=null && menuVo.getChildren().size() == 0 ) {
                menuVo.setChildren(null);
                continue ;
            }
             setChildrenNull(menuVo.getChildren());
        }
    }


}