From 21debf9e346f544e38e5e237fcd40a130fb3f260 Mon Sep 17 00:00:00 2001
From: lizijie <lzjiiie@163.com>
Date: Wed, 25 Aug 2021 08:51:04 +0800
Subject: [PATCH] 删除冗余代码
---
screen-manage/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java | 81 ++++++++++++++++++++++++----------------
1 files changed, 48 insertions(+), 33 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..204f902 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
@@ -48,6 +48,7 @@
@Autowired
GroupMenuMapper groupMenuMapper;
+
@Override
public MenuQueryDTO queryAllMenus() {
//������������������
@@ -82,18 +83,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,10 +161,12 @@
UpdateWrapper deleteGroupMenuWrapper = new UpdateWrapper();
List<Integer> allMenuIds = new ArrayList<>();
allMenuIds.add(id);
- allMenuIds.addAll(childrenIds);
- deleteGroupMenuWrapper.in("menu_id",allMenuIds);
- deleteGroupMenuWrapper.set("is_delete",Constants.DELETE);
- groupMenuMapper.update(null,deleteGroupMenuWrapper);
+ 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());
@@ -245,13 +246,13 @@
/**
- * @Description: ������������������������������������������������������
- * @Param: [menus]
- * @return: java.util.List<com.moral.api.entity.Menu>
- * @Author: ���������
- * @Date: 2021/5/13
- */
- public void combinationParentChildrenMenus(List<Menu> menus){
+ * @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) {
@@ -351,21 +352,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;
+ }
+ });
}
}
--
Gitblit v1.8.0