|  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; | 
|---|
|  |  |  | import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper; | 
|---|
|  |  |  | import com.moral.api.entity.ManageMenu; | 
|---|
|  |  |  | import com.moral.api.config.mybatis.wrapper.NullFilterWrapper; | 
|---|
|  |  |  | import com.moral.api.entity.Group; | 
|---|
|  |  |  | import com.moral.api.entity.Menu; | 
|---|
|  |  |  | import com.moral.api.mapper.GroupMenuMapper; | 
|---|
|  |  |  | import com.moral.api.mapper.MenuMapper; | 
|---|
|  |  |  | import com.moral.api.pojo.dto.menu.MenuDTO; | 
|---|
|  |  |  | import com.moral.api.pojo.dto.menu.MenuQueryDTO; | 
|---|
|  |  |  | import com.moral.api.pojo.dto.menu.MenuQueryNamesDTO; | 
|---|
|  |  |  | import com.moral.api.pojo.form.menu.MenuDeleteForm; | 
|---|
|  |  |  | import com.moral.api.pojo.form.menu.MenuInsertForm; | 
|---|
|  |  |  | import com.moral.api.pojo.form.menu.MenuQueryNamesForm; | 
|---|
|  |  |  | import com.moral.api.pojo.form.menu.MenuUpdateForm; | 
|---|
|  |  |  | import com.moral.api.service.GroupService; | 
|---|
|  |  |  | import com.moral.api.service.MenuService; | 
|---|
|  |  |  | import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; | 
|---|
|  |  |  | import com.moral.constant.Constants; | 
|---|
|  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | MenuMapper menuMapper; | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | GroupService groupService; | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | GroupMenuMapper groupMenuMapper; | 
|---|
|  |  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | public MenuQueryDTO queryAllMenu() { | 
|---|
|  |  |  | public MenuQueryDTO queryAllMenus() { | 
|---|
|  |  |  | //创建返回对象 | 
|---|
|  |  |  | MenuQueryDTO dto = new MenuQueryDTO(); | 
|---|
|  |  |  | //获取所有菜单 | 
|---|
|  |  |  | List<Menu> menus = menuMapper.selectList(null); | 
|---|
|  |  |  | //组合menu父子结构 | 
|---|
|  |  |  | Map<Integer, Menu> menuMap = new HashMap<>(); | 
|---|
|  |  |  | for (Menu menu : menus) { | 
|---|
|  |  |  | menuMap.put(menu.getId(), menu); | 
|---|
|  |  |  | menu.setChildren(new ArrayList<>());//初始化集合 | 
|---|
|  |  |  | } | 
|---|
|  |  |  | for (Menu menu : menus) { | 
|---|
|  |  |  | combinationParentChildrenMenus(menuMap, menu); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | //删除非跟菜单 | 
|---|
|  |  |  | menus.removeIf(new Predicate<Menu>() { | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | public boolean test(Menu menu) { | 
|---|
|  |  |  | if (menu.getParentId().equals(0)) | 
|---|
|  |  |  | return false; | 
|---|
|  |  |  | return true; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | QueryWrapper wrapper = new QueryWrapper(); | 
|---|
|  |  |  | wrapper.eq("is_delete", Constants.NOT_DELETE); | 
|---|
|  |  |  | List<Menu> menus = menuMapper.selectList(wrapper); | 
|---|
|  |  |  | //组装成父子结构 | 
|---|
|  |  |  | combinationParentChildrenMenus(menus); | 
|---|
|  |  |  | //转换菜单为DTO | 
|---|
|  |  |  | List<MenuDTO> dtos = new ArrayList<>(); | 
|---|
|  |  |  | for (Menu menu : menus) { | 
|---|
|  |  |  | MenuDTO menuDTO = new MenuDTO(); | 
|---|
|  |  |  | menuDTO.setMenu(menu); | 
|---|
|  |  |  | dtos.add(menuDTO); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | dto.setDtos(dtos); | 
|---|
|  |  |  | dto.setCode(ResponseCodeEnum.SUCCESS.getCode()); | 
|---|
|  |  |  | dto.setMsg(ResponseCodeEnum.SUCCESS.getMsg()); | 
|---|
|  |  |  | return dto; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | public MenuQueryDTO queryAdminGroupMenusByOrganizationId(Integer organizationId) { | 
|---|
|  |  |  | //创建返回对象 | 
|---|
|  |  |  | MenuQueryDTO dto = new MenuQueryDTO(); | 
|---|
|  |  |  | //查询组织admin角色 | 
|---|
|  |  |  | Group group = groupService.queryAdminGroupByOrganizationId(organizationId); | 
|---|
|  |  |  | if (ObjectUtils.isEmpty(group)) {//如果没有角色则证明组织还没有账号 | 
|---|
|  |  |  | dto.setCode(ResponseCodeEnum.ORGANIZATION_USER_NOT_EXIST.getCode()); | 
|---|
|  |  |  | dto.setMsg(ResponseCodeEnum.ORGANIZATION_USER_NOT_EXIST.getMsg()); | 
|---|
|  |  |  | return dto; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | //根据角色查询拥有的所有菜单 | 
|---|
|  |  |  | List<Menu> ownMenus = menuMapper.getMenusByGroupId(group.getId()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /*判断每个菜单是否有子菜单,如果有则不传递 | 
|---|
|  |  |  | * 前端无法解决父菜单选中,子菜单没有全部选中回显问题。由后端处理的代码。*/ | 
|---|
|  |  |  | removeMenuWithChildren(ownMenus); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | //封装返回结果 | 
|---|
|  |  |  | List<MenuDTO> dtos = new ArrayList<>(); | 
|---|
|  |  |  | for (Menu menu : ownMenus) { | 
|---|
|  |  |  | MenuDTO menuDTO = new MenuDTO(); | 
|---|
|  |  |  | menuDTO.setMenu(menu); | 
|---|
|  |  |  | dtos.add(menuDTO); | 
|---|
|  |  |  | 
|---|
|  |  |  | dto.setMsg(ResponseCodeEnum.MENU_IS_EXIST.getMsg()); | 
|---|
|  |  |  | return dto; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | //执行插入逻辑 | 
|---|
|  |  |  | menuMapper.insert(menu); | 
|---|
|  |  |  | //封装返回对象 | 
|---|
|  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | @Transactional | 
|---|
|  |  |  | public MenuDTO deleteMenu(MenuDeleteForm form) { | 
|---|
|  |  |  | //创建返回对象 | 
|---|
|  |  |  | MenuDTO dto = new MenuDTO(); | 
|---|
|  |  |  | 
|---|
|  |  |  | dto.setMsg(ResponseCodeEnum.MENU_IS_NULL.getMsg()); | 
|---|
|  |  |  | return dto; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | //执行逻辑删除 | 
|---|
|  |  |  | existMenu.setIsDelete(Constants.DELETE); | 
|---|
|  |  |  | menuMapper.updateById(existMenu); | 
|---|
|  |  |  | //删除所有子菜单 | 
|---|
|  |  |  | List<Integer> childrenIds = getChildrenIdsByParentId(id); | 
|---|
|  |  |  | UpdateWrapper wrapper = new UpdateWrapper(); | 
|---|
|  |  |  | wrapper.in("id",childrenIds); | 
|---|
|  |  |  | wrapper.set("is_delete",Constants.DELETE); | 
|---|
|  |  |  | menuMapper.update(null,wrapper); | 
|---|
|  |  |  | if (!ObjectUtils.isEmpty(childrenIds)) { | 
|---|
|  |  |  | UpdateWrapper wrapper = new UpdateWrapper(); | 
|---|
|  |  |  | wrapper.in("id", childrenIds); | 
|---|
|  |  |  | wrapper.set("is_delete", Constants.DELETE); | 
|---|
|  |  |  | menuMapper.update(null, wrapper); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | //删除菜单 | 
|---|
|  |  |  | existMenu.setIsDelete(Constants.DELETE); | 
|---|
|  |  |  | menuMapper.updateById(existMenu); | 
|---|
|  |  |  | //删除角色菜单关系表中对应的菜单 | 
|---|
|  |  |  | 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); | 
|---|
|  |  |  | //封装返回对象 | 
|---|
|  |  |  | dto.setCode(ResponseCodeEnum.SUCCESS.getCode()); | 
|---|
|  |  |  | dto.setMsg(ResponseCodeEnum.SUCCESS.getMsg()); | 
|---|
|  |  |  | 
|---|
|  |  |  | return dto; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | private void combinationParentChildrenMenus(Map<Integer, Menu> menuMap, Menu menu) { | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | public MenuDTO updateMenu(MenuUpdateForm form) { | 
|---|
|  |  |  | //创建返回对象 | 
|---|
|  |  |  | MenuDTO dto = new MenuDTO(); | 
|---|
|  |  |  | //取参 | 
|---|
|  |  |  | Menu menu = form.formConvertEntity(); | 
|---|
|  |  |  | //查找更新前的菜单用于插入日志 | 
|---|
|  |  |  | QueryWrapper<Menu> oldMenuWrapper = new QueryWrapper<>(); | 
|---|
|  |  |  | Menu oldMenu = new Menu(); | 
|---|
|  |  |  | oldMenu.setId(menu.getId()); | 
|---|
|  |  |  | oldMenu.setIsDelete(Constants.NOT_DELETE); | 
|---|
|  |  |  | oldMenuWrapper.setEntity(oldMenu); | 
|---|
|  |  |  | oldMenu = menuMapper.selectOne(oldMenuWrapper); | 
|---|
|  |  |  | if (ObjectUtils.isEmpty(oldMenu)) { | 
|---|
|  |  |  | dto.setCode(ResponseCodeEnum.MENU_IS_NULL.getCode()); | 
|---|
|  |  |  | dto.setMsg(ResponseCodeEnum.MENU_IS_NULL.getMsg()); | 
|---|
|  |  |  | return dto; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | //更新 | 
|---|
|  |  |  | menuMapper.updateById(menu); | 
|---|
|  |  |  | //获取更新后的对象 | 
|---|
|  |  |  | menu = menuMapper.selectById(menu.getId()); | 
|---|
|  |  |  | //封装返回对象 | 
|---|
|  |  |  | dto.setMenu(menu); | 
|---|
|  |  |  | dto.setCode(ResponseCodeEnum.SUCCESS.getCode()); | 
|---|
|  |  |  | dto.setMsg(ResponseCodeEnum.SUCCESS.getMsg()); | 
|---|
|  |  |  | return dto; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | public MenuQueryNamesDTO queryNames(MenuQueryNamesForm form) { | 
|---|
|  |  |  | //创建返回对象 | 
|---|
|  |  |  | MenuQueryNamesDTO dto = new MenuQueryNamesDTO(); | 
|---|
|  |  |  | //取参 | 
|---|
|  |  |  | Integer id = form.getId(); | 
|---|
|  |  |  | String name = form.getName(); | 
|---|
|  |  |  | //查询结果 | 
|---|
|  |  |  | NullFilterWrapper<Menu> wrapper = new NullFilterWrapper<>(); | 
|---|
|  |  |  | wrapper.eq("is_delete", Constants.NOT_DELETE); | 
|---|
|  |  |  | wrapper.like("name", name); | 
|---|
|  |  |  | List<Menu> menus = menuMapper.selectList(wrapper); | 
|---|
|  |  |  | //过滤该id对应的所有子菜单 | 
|---|
|  |  |  | List<Integer> childrenIds = getChildrenIdsByParentId(id);//获取所有子菜单id结合 | 
|---|
|  |  |  | if (ObjectUtils.isEmpty(childrenIds)) { | 
|---|
|  |  |  | menus.removeIf(new Predicate<Menu>() { | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | public boolean test(Menu m) { | 
|---|
|  |  |  | if (m.getId().equals(id)) | 
|---|
|  |  |  | return true; | 
|---|
|  |  |  | return false; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | } else { | 
|---|
|  |  |  | menus.removeIf(new Predicate<Menu>() { | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | public boolean test(Menu m) { | 
|---|
|  |  |  | if (childrenIds.contains(m.getId()) || m.getId().equals(id)) | 
|---|
|  |  |  | return true; | 
|---|
|  |  |  | return false; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | //封装返回对象 | 
|---|
|  |  |  | dto.setMenus(menus); | 
|---|
|  |  |  | dto.setCode(ResponseCodeEnum.SUCCESS.getCode()); | 
|---|
|  |  |  | dto.setMsg(ResponseCodeEnum.SUCCESS.getMsg()); | 
|---|
|  |  |  | return dto; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * @Description: 将传来的菜单集合封装成父子结构的集合 | 
|---|
|  |  |  | * @Param: [menus] | 
|---|
|  |  |  | * @return: java.util.List<com.moral.api.entity.Menu> | 
|---|
|  |  |  | * @Author: 陈凯裕 | 
|---|
|  |  |  | * @Date: 2021/5/13 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | public void combinationParentChildrenMenus(List<Menu> menus) { | 
|---|
|  |  |  | //组合menu父子结构 | 
|---|
|  |  |  | Map<Integer, Menu> menuMap = new HashMap<>(); | 
|---|
|  |  |  | for (Menu menu : menus) { | 
|---|
|  |  |  | menuMap.put(menu.getId(), menu); | 
|---|
|  |  |  | menu.setChildren(new ArrayList<>());//初始化集合 | 
|---|
|  |  |  | } | 
|---|
|  |  |  | for (Menu menu : menus) { | 
|---|
|  |  |  | putMenuToParentMenu(menuMap, menu); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | //删除非根菜单 | 
|---|
|  |  |  | menus.removeIf(new Predicate<Menu>() { | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | public boolean test(Menu menu) { | 
|---|
|  |  |  | if (menu.getParentId().equals(0)) | 
|---|
|  |  |  | return false; | 
|---|
|  |  |  | return true; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * @Description: 把传来的menu放到menuMap对应的父菜单中 | 
|---|
|  |  |  | * @Param: [menuMap, menu] | 
|---|
|  |  |  | * @return: void | 
|---|
|  |  |  | * @Author: 陈凯裕 | 
|---|
|  |  |  | * @Date: 2021/5/6 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | private void putMenuToParentMenu(Map<Integer, Menu> menuMap, Menu menu) { | 
|---|
|  |  |  | Integer parentId = menu.getParentId(); | 
|---|
|  |  |  | Menu parentMenu = menuMap.get(parentId); | 
|---|
|  |  |  | if (!ObjectUtils.isEmpty(parentMenu)) { | 
|---|
|  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * @Description: 根据菜单的id获取封装好children的menu对象 | 
|---|
|  |  |  | * @Param: [id] | 
|---|
|  |  |  | * @return: com.moral.api.entity.Menu | 
|---|
|  |  |  | * @Author: 陈凯裕 | 
|---|
|  |  |  | * @Date: 2021/4/26 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | private Menu getMenuAndChildrenById(Integer id){ | 
|---|
|  |  |  | * @Description: 根据菜单的id获取封装好children的menu对象 | 
|---|
|  |  |  | * @Param: [id] | 
|---|
|  |  |  | * @return: com.moral.api.entity.Menu | 
|---|
|  |  |  | * @Author: 陈凯裕 | 
|---|
|  |  |  | * @Date: 2021/4/26 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | private Menu getMenuAndChildrenById(Integer id) { | 
|---|
|  |  |  | //获取所有菜单 | 
|---|
|  |  |  | List<Menu> menus = menuMapper.selectList(null); | 
|---|
|  |  |  | QueryWrapper wrapper = new QueryWrapper(); | 
|---|
|  |  |  | wrapper.eq("is_delete", Constants.NOT_DELETE); | 
|---|
|  |  |  | List<Menu> menus = menuMapper.selectList(wrapper); | 
|---|
|  |  |  | //组合menu父子结构 | 
|---|
|  |  |  | Map<Integer, Menu> menuMap = new HashMap<>(); | 
|---|
|  |  |  | for (Menu menu : menus) { | 
|---|
|  |  |  | 
|---|
|  |  |  | menu.setChildren(new ArrayList<>());//初始化集合 | 
|---|
|  |  |  | } | 
|---|
|  |  |  | for (Menu menu : menus) { | 
|---|
|  |  |  | combinationParentChildrenMenus(menuMap, menu); | 
|---|
|  |  |  | putMenuToParentMenu(menuMap, menu); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | Menu menu = menuMap.get(id); | 
|---|
|  |  |  | return menu; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * @Description: 获取该菜单下的所有子菜单id集合 | 
|---|
|  |  |  | * @Param: [id] | 
|---|
|  |  |  | * @return: java.util.List<java.lang.Integer> | 
|---|
|  |  |  | * @Author: 陈凯裕 | 
|---|
|  |  |  | * @Date: 2021/4/26 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | private List<Integer> getChildrenIdsByParentId(Integer id){ | 
|---|
|  |  |  | * @Description: 获取该菜单下的所有子菜单id集合 | 
|---|
|  |  |  | * @Param: [id] | 
|---|
|  |  |  | * @return: java.util.List<java.lang.Integer> | 
|---|
|  |  |  | * @Author: 陈凯裕 | 
|---|
|  |  |  | * @Date: 2021/4/26 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | private List<Integer> getChildrenIdsByParentId(Integer id) { | 
|---|
|  |  |  | Menu parentMenu = getMenuAndChildrenById(id); | 
|---|
|  |  |  | List<Menu> children = parentMenu.getChildren(); | 
|---|
|  |  |  | List<Integer> childrenIds = new ArrayList<>(); | 
|---|
|  |  |  | for (Menu child : children) { | 
|---|
|  |  |  | recursiveAccess(child,childrenIds); | 
|---|
|  |  |  | List<Integer> childrenIds = null; | 
|---|
|  |  |  | if (!ObjectUtils.isEmpty(parentMenu.getChildren())) { | 
|---|
|  |  |  | List<Menu> children = parentMenu.getChildren(); | 
|---|
|  |  |  | childrenIds = new ArrayList<>(); | 
|---|
|  |  |  | for (Menu child : children) { | 
|---|
|  |  |  | recursiveAccess(child, childrenIds); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | return childrenIds; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * @Description: 递归获取菜单的id放入集合中 | 
|---|
|  |  |  | * @Param: [menu, ids] | 
|---|
|  |  |  | * @return: void | 
|---|
|  |  |  | * @Author: 陈凯裕 | 
|---|
|  |  |  | * @Date: 2021/4/26 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | private void recursiveAccess(Menu menu,List<Integer> ids){ | 
|---|
|  |  |  | * @Description: 递归获取菜单以及children的id放入集合中 | 
|---|
|  |  |  | * @Param: [menu, ids] | 
|---|
|  |  |  | * @return: void | 
|---|
|  |  |  | * @Author: 陈凯裕 | 
|---|
|  |  |  | * @Date: 2021/4/26 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | private void recursiveAccess(Menu menu, List<Integer> ids) { | 
|---|
|  |  |  | ids.add(menu.getId()); | 
|---|
|  |  |  | List<Menu> children = menu.getChildren(); | 
|---|
|  |  |  | if(!ObjectUtils.isEmpty(children)){ | 
|---|
|  |  |  | if (!ObjectUtils.isEmpty(children)) { | 
|---|
|  |  |  | for (Menu child : children) { | 
|---|
|  |  |  | recursiveAccess(child,ids); | 
|---|
|  |  |  | recursiveAccess(child, ids); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * @Description: 移除含有子菜单的菜单 | 
|---|
|  |  |  | * @Param: [menus] | 
|---|
|  |  |  | * @return: java.util.List<com.moral.api.entity.Menu> | 
|---|
|  |  |  | * @Author: 陈凯裕 | 
|---|
|  |  |  | * @Date: 2021/5/28 | 
|---|
|  |  |  | */ | 
|---|
|  |  |  | 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; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | } | 
|---|