From d6a0a1bb4c7b27a6521c969046cb969b168fa93a Mon Sep 17 00:00:00 2001 From: kaiyu <404897439@qq.com> Date: Mon, 07 Jun 2021 10:11:08 +0800 Subject: [PATCH] screen-manage 更新组织删除逻辑,存在站点或者子组织存在站点则无法删除。 --- screen-manage/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java | 62 +++++++++++++++++++------------ 1 files changed, 38 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..b220c29 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 @@ -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); @@ -351,21 +351,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