From 0ecfecd2b85b71884fcfba8fa6849be963e434fd Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Mon, 11 Jul 2022 16:47:53 +0800
Subject: [PATCH] fix

---
 src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/MenuServiceImpl.java |   34 ++++++++++++++++++++++------------
 1 files changed, 22 insertions(+), 12 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 6507b54..fcc7874 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
@@ -13,12 +13,15 @@
 import com.xcong.farmer.cms.modules.system.dto.AdminUpdateMenuDto;
 import com.xcong.farmer.cms.modules.system.entity.MenuEntity;
 import com.xcong.farmer.cms.modules.system.entity.RoleMenuEntity;
+import com.xcong.farmer.cms.modules.system.entity.UserEntity;
 import com.xcong.farmer.cms.modules.system.entity.UserRoleEntity;
 import com.xcong.farmer.cms.modules.system.mapper.MenuMapper;
 import com.xcong.farmer.cms.modules.system.mapper.RoleMenuMapper;
 import com.xcong.farmer.cms.modules.system.service.IMenuService;
+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;
 
@@ -41,8 +44,10 @@
     @Override
     public Result getMenuInPage(AdminMenuDto adminMenuDto) {
         Page<AdminMenuVo> page = new Page<>(adminMenuDto.getPageNum(), adminMenuDto.getPageSize());
+        UserEntity userlogin = LoginUserUtil.getLoginUser();
         MenuEntity menuEntity = new MenuEntity();
         menuEntity.setParentId(MenuEntity.PARENTID_DEFAULT);
+//        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)){
@@ -73,14 +78,9 @@
     @Override
     @Transactional
     public Result addMenu(AdminAddMenuDto adminAddMenuDto) {
+        UserEntity userlogin = LoginUserUtil.getLoginUser();
         String menuName = adminAddMenuDto.getMenuName();
-        if(StrUtil.isEmpty(menuName)){
-            return Result.fail("请输入菜单名称");
-        }
         Integer menuType = adminAddMenuDto.getMenuType();
-        if(ObjectUtil.isEmpty(menuType)){
-            return Result.fail("请输入菜单类型");
-        }
         MenuEntity menuEntity = new MenuEntity();
         menuEntity.setMenuName(menuName);
         menuEntity.setMenuType(menuType);
@@ -145,14 +145,8 @@
             return Result.fail("菜单不存在");
         }
         String menuName = adminUpdateMenuDto.getMenuName();
-        if(StrUtil.isEmpty(menuName)){
-            return Result.fail("请输入菜单名称");
-        }
         menuEntity.setMenuName(menuName);
         Integer menuType = adminUpdateMenuDto.getMenuType();
-        if(ObjectUtil.isEmpty(menuType)){
-            return Result.fail("请输入菜单类型");
-        }
         menuEntity.setMenuType(menuType);
         String routeUrl = adminUpdateMenuDto.getRouteUrl();
         if(StrUtil.isNotEmpty(routeUrl)){
@@ -194,6 +188,7 @@
 
     @Override
     public Result getMenuInList() {
+        UserEntity userlogin = LoginUserUtil.getLoginUser();
         List<AdminMenuVo> records = this.baseMapper.selectAdminMenuVoInListByParentId(MenuEntity.PARENTID_DEFAULT);
         if(CollUtil.isNotEmpty(records)){
             for(AdminMenuVo adminMenuVo : records){
@@ -217,6 +212,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