From 43e5052b91f0568ccbcf3fd6df70cf8209db184a Mon Sep 17 00:00:00 2001 From: lizijie <lzjiiie@163.com> Date: Fri, 28 May 2021 17:36:11 +0800 Subject: [PATCH] Merge remote-tracking branch 'origin/dev' into dev --- screen-manage/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java | 60 ++++++++++++++++++++++++++++++++++++------------------------ 1 files changed, 36 insertions(+), 24 deletions(-) diff --git a/screen-manage/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java b/screen-manage/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java index 8155f11..137a78a 100644 --- a/screen-manage/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java +++ b/screen-manage/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java @@ -82,18 +82,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); @@ -351,21 +349,35 @@ } /** - * @Description: ������menu������������������������������������menus��� - * @Param: [menu, menus] - * @return: void - * @Author: ��������� - * @Date: 2021/5/13 - */ - 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); - } + * @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; + } + }); } } -- Gitblit v1.8.0