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