From 93e281b3cc60c8d24a0ed5547d3d399c03f9bab6 Mon Sep 17 00:00:00 2001
From: xiaoyong931011 <15274802129@163.com>
Date: Fri, 10 Jun 2022 18:03:52 +0800
Subject: [PATCH] 20220606

---
 src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/MenuServiceImpl.java |   25 +++++++++++++++++++------
 1 files changed, 19 insertions(+), 6 deletions(-)

diff --git a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/MenuServiceImpl.java b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/MenuServiceImpl.java
index 044e6da..179b0e4 100644
--- a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/MenuServiceImpl.java
+++ b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/MenuServiceImpl.java
@@ -21,6 +21,7 @@
 import com.xcong.farmer.cms.modules.system.util.LoginUserUtil;
 import com.xcong.farmer.cms.modules.system.vo.AdminMenuVo;
 import com.xcong.farmer.cms.modules.system.vo.AdminSeeMenuInfoVo;
+import com.xcong.farmer.cms.modules.system.vo.AdminUserMenuVo;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
 
@@ -46,7 +47,7 @@
         UserEntity userlogin = LoginUserUtil.getLoginUser();
         MenuEntity menuEntity = new MenuEntity();
         menuEntity.setParentId(MenuEntity.PARENTID_DEFAULT);
-        menuEntity.setBelongId(userlogin.getBelongId()==null?0L:userlogin.getBelongId());
+//        menuEntity.setBelongId(userlogin.getBelongId()==null?UserEntity.USER_BELONG_TOP:userlogin.getBelongId());
         IPage<AdminMenuVo> list = this.baseMapper.selectAdminMenuVoInPage(page,menuEntity);
         List<AdminMenuVo> records = list.getRecords();
         if(CollUtil.isNotEmpty(records)){
@@ -78,7 +79,6 @@
     @Transactional
     public Result addMenu(AdminAddMenuDto adminAddMenuDto) {
         UserEntity userlogin = LoginUserUtil.getLoginUser();
-        long belongId = userlogin.getBelongId() == null ? 0L : userlogin.getBelongId();
         String menuName = adminAddMenuDto.getMenuName();
         if(StrUtil.isEmpty(menuName)){
             return Result.fail("请输入菜单名称");
@@ -104,7 +104,6 @@
         if(StrUtil.isNotEmpty(routeUrl)){
             menuEntity.setRouteUrl(routeUrl);
         }
-        menuEntity.setBelongId(belongId);
         this.baseMapper.insert(menuEntity);
         return Result.ok("新增成功");
     }
@@ -202,14 +201,13 @@
     @Override
     public Result getMenuInList() {
         UserEntity userlogin = LoginUserUtil.getLoginUser();
-        long belongId = userlogin.getBelongId() == null ? 0L : userlogin.getBelongId();
-        List<AdminMenuVo> records = this.baseMapper.selectAdminMenuVoInListByParentId(MenuEntity.PARENTID_DEFAULT,belongId);
+        List<AdminMenuVo> records = this.baseMapper.selectAdminMenuVoInListByParentId(MenuEntity.PARENTID_DEFAULT);
         if(CollUtil.isNotEmpty(records)){
             for(AdminMenuVo adminMenuVo : records){
                 Long id = adminMenuVo.getId();
                 QueryWrapper<MenuEntity> objectQueryWrapper = new QueryWrapper<>();
                 objectQueryWrapper.eq("parent_id",id);
-                objectQueryWrapper.eq("belong_id",belongId);
+//                objectQueryWrapper.eq("belong_id",belongId);
                 List<MenuEntity> menuEntities = this.baseMapper.selectList(objectQueryWrapper);
                 List<AdminMenuVo> adminMenuVoChilds = new ArrayList<>();
                 if(CollUtil.isNotEmpty(menuEntities)){
@@ -227,6 +225,21 @@
                 adminMenuVo.setChild(adminMenuVoChilds);
             }
         }
+
+        if(CollUtil.isNotEmpty(records)){
+            for(AdminMenuVo adminMenuVo : records){
+                List<AdminMenuVo> child = adminMenuVo.getChild();
+                if(CollUtil.isNotEmpty(child)){
+                    List<Long> menuIds = new ArrayList<>();
+                    for(AdminMenuVo adminMenuVoChild : child){
+                        Long id = adminMenuVoChild.getId();
+                        menuIds.add(id);
+                    }
+                    List<AdminMenuVo> adminMenuVoByOrderNum = this.baseMapper.selectAdminMenuVoOrderByOrderNum(menuIds);
+                    adminMenuVo.setChild(adminMenuVoByOrderNum);
+                }
+            }
+        }
         return Result.ok(records);
     }
 }

--
Gitblit v1.9.1