From 6f4e852b84c577454a4876f83c7085bd360fe4fb Mon Sep 17 00:00:00 2001
From: jinpengyong <jpy123456>
Date: Tue, 17 Aug 2021 17:07:02 +0800
Subject: [PATCH] 特殊设备数据insert

---
 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