From 8fb0744ae10e7a1855f0474caf605ad0989c5a34 Mon Sep 17 00:00:00 2001
From: jinpengyong <jpy123456>
Date: Thu, 03 Jun 2021 13:31:45 +0800
Subject: [PATCH] 工艺扩展

---
 screen-manage/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java |   75 +++++++++++++++++++++++++------------
 1 files changed, 50 insertions(+), 25 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 76c4316..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
@@ -5,6 +5,7 @@
 import com.moral.api.config.mybatis.wrapper.NullFilterWrapper;
 import com.moral.api.entity.Group;
 import com.moral.api.entity.Menu;
+import com.moral.api.mapper.GroupMenuMapper;
 import com.moral.api.mapper.MenuMapper;
 import com.moral.api.pojo.dto.menu.MenuDTO;
 import com.moral.api.pojo.dto.menu.MenuQueryDTO;
@@ -18,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;
@@ -44,6 +46,9 @@
     MenuMapper menuMapper;
     @Autowired
     GroupService groupService;
+    @Autowired
+    GroupMenuMapper groupMenuMapper;
+
 
     @Override
     public MenuQueryDTO queryAllMenus() {
@@ -79,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);
@@ -131,6 +134,7 @@
     }
 
     @Override
+    @Transactional
     public MenuDTO deleteMenu(MenuDeleteForm form) {
         //������������������
         MenuDTO dto = new MenuDTO();
@@ -154,7 +158,14 @@
         //������������
         existMenu.setIsDelete(Constants.DELETE);
         menuMapper.updateById(existMenu);
-
+        //���������������������������������������������
+        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);
         //������������������
         dto.setCode(ResponseCodeEnum.SUCCESS.getCode());
         dto.setMsg(ResponseCodeEnum.SUCCESS.getMsg());
@@ -340,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