wuqiping
2021-06-09 f342a0d88b2a7168f794d0cf545d16a3138acb6d
screen-manage/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java
@@ -19,6 +19,7 @@
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.moral.constant.Constants;
import com.moral.constant.ResponseCodeEnum;
import com.sun.corba.se.spi.presentation.rmi.IDLNameTranslator;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
@@ -47,6 +48,7 @@
    GroupService groupService;
    @Autowired
    GroupMenuMapper groupMenuMapper;
    @Override
    public MenuQueryDTO queryAllMenus() {
@@ -82,18 +84,16 @@
            dto.setMsg(ResponseCodeEnum.ORGANIZATION_USER_NOT_EXIST.getMsg());
            return dto;
        }
        //根据角色查询对应的所有菜单
        List<Menu> allMenus = menuMapper.getMenusByGroupId(group.getId());
        //将菜单转换为父子结构
        combinationParentChildrenMenus(allMenus);
        //删除所有菜单只保留最底层菜单
        List<Menu> menus = new ArrayList<>();
        for (Menu menu : allMenus) {
            getNotParentMenu(menu, menus);
        }
        //根据角色查询拥有的所有菜单
        List<Menu> ownMenus = menuMapper.getMenusByGroupId(group.getId());
        /*判断每个菜单是否有子菜单,如果有则不传递
         * 前端无法解决父菜单选中,子菜单没有全部选中回显问题。由后端处理的代码。*/
        removeMenuWithChildren(ownMenus);
        //封装返回结果
        List<MenuDTO> dtos = new ArrayList<>();
        for (Menu menu : menus) {
        for (Menu menu : ownMenus) {
            MenuDTO menuDTO = new MenuDTO();
            menuDTO.setMenu(menu);
            dtos.add(menuDTO);
@@ -162,7 +162,9 @@
        UpdateWrapper deleteGroupMenuWrapper = new UpdateWrapper();
        List<Integer> allMenuIds = new ArrayList<>();
        allMenuIds.add(id);
        if (!ObjectUtils.isEmpty(childrenIds)) {
        allMenuIds.addAll(childrenIds);
        }
        deleteGroupMenuWrapper.in("menu_id",allMenuIds);
        deleteGroupMenuWrapper.set("is_delete",Constants.DELETE);
        groupMenuMapper.update(null,deleteGroupMenuWrapper);
@@ -351,21 +353,35 @@
    }
    /**
     * @Description: 获取menu菜单中的最底层菜单添加到menus中
     * @Param: [menu, menus]
     * @return: void
     * @Description: 移除含有子菜单的菜单
     * @Param: [menus]
     * @return: java.util.List<com.moral.api.entity.Menu>
     * @Author: 陈凯裕
     * @Date: 2021/5/13
     * @Date: 2021/5/28
     */
    private void getNotParentMenu(Menu menu, List<Menu> menus) {
        if (ObjectUtils.isEmpty(menu.getChildren())) {
            menus.add(menu);
        } else {
            List<Menu> children = menu.getChildren();
            for (Menu child : children) {
                getNotParentMenu(child, menus);
    private void removeMenuWithChildren(List<Menu> menus) {
        //查询所有菜单
        QueryWrapper wrapper = new QueryWrapper();
        wrapper.eq("is_delete", Constants.NOT_DELETE);
        List<Menu> allMenus = menuMapper.selectList(wrapper);
        //组合menu父子结构
        Map<Integer, Menu> menuMap = new HashMap<>();
        for (Menu menu : allMenus) {
            menuMap.put(menu.getId(), menu);
            menu.setChildren(new ArrayList<>());//初始化集合
            }
        for (Menu menu : allMenus) {
            putMenuToParentMenu(menuMap, menu);
        }
        //移除含有子菜单的菜单
        menus.removeIf(new Predicate<Menu>() {
            @Override
            public boolean test(Menu menu) {
                if (ObjectUtils.isEmpty(menuMap.get(menu.getId()).getChildren()))
                    return false;
                return true;
            }
        });
    }
}