From 7fa0a141a3c88ca90c67c966260bf26f03f255da Mon Sep 17 00:00:00 2001
From: xiaoyong931011 <15274802129@163.com>
Date: Tue, 07 Jun 2022 16:00:09 +0800
Subject: [PATCH] 20220606

---
 src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/MenuServiceImpl.java |   63 +++++++++++++++++++++++++++++++
 1 files changed, 62 insertions(+), 1 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 b148be8..044e6da 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
@@ -3,16 +3,22 @@
 import cn.hutool.core.util.StrUtil;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.core.toolkit.StringPool;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.xcong.farmer.cms.common.response.Result;
 import com.xcong.farmer.cms.modules.system.dto.AdminAddMenuDto;
+import com.xcong.farmer.cms.modules.system.dto.AdminDeleteDto;
 import com.xcong.farmer.cms.modules.system.dto.AdminMenuDto;
 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 lombok.extern.slf4j.Slf4j;
@@ -37,11 +43,12 @@
     @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?0L:userlogin.getBelongId());
         IPage<AdminMenuVo> list = this.baseMapper.selectAdminMenuVoInPage(page,menuEntity);
         List<AdminMenuVo> records = list.getRecords();
-        list.getTotal();
         if(CollUtil.isNotEmpty(records)){
             for(AdminMenuVo adminMenuVo : records){
                 Long id = adminMenuVo.getId();
@@ -70,6 +77,8 @@
     @Override
     @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("请输入菜单名称");
@@ -95,6 +104,7 @@
         if(StrUtil.isNotEmpty(routeUrl)){
             menuEntity.setRouteUrl(routeUrl);
         }
+        menuEntity.setBelongId(belongId);
         this.baseMapper.insert(menuEntity);
         return Result.ok("新增成功");
     }
@@ -168,4 +178,55 @@
         this.baseMapper.updateById(menuEntity);
         return Result.ok("更新成功");
     }
+
+    @Override
+    @Transactional
+    public Result delObjs(AdminDeleteDto adminDeleteDto) {
+        String ids = adminDeleteDto.getIds();
+        if(StrUtil.isNotEmpty(ids)){
+            String[] menuIds = ids.split(StringPool.COMMA);
+            for(String menuIdStr : menuIds){
+                Long menuId = Long.valueOf(menuIdStr);
+                QueryWrapper<MenuEntity> objectQueryWrapper = new QueryWrapper<>();
+                objectQueryWrapper.eq("parent_id",menuId);
+                this.baseMapper.delete(objectQueryWrapper);
+
+                this.baseMapper.deleteById(menuId);
+
+                roleMenuMapper.deleteByMenuId(menuId);
+            }
+        }
+        return Result.ok("删除成功");
+    }
+
+    @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);
+        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);
+                List<MenuEntity> menuEntities = this.baseMapper.selectList(objectQueryWrapper);
+                List<AdminMenuVo> adminMenuVoChilds = new ArrayList<>();
+                if(CollUtil.isNotEmpty(menuEntities)){
+                    for(MenuEntity menuEntityChild : menuEntities){
+                        AdminMenuVo child = new AdminMenuVo();
+                        child.setId(menuEntityChild.getId());
+                        child.setMenuName(menuEntityChild.getMenuName());
+                        child.setMenuType(menuEntityChild.getMenuType());
+                        child.setRouteUrl(menuEntityChild.getRouteUrl());
+                        child.setOrderNum(menuEntityChild.getOrderNum());
+                        child.setCreateTime(menuEntityChild.getCreateTime());
+                        adminMenuVoChilds.add(child);
+                    }
+                }
+                adminMenuVo.setChild(adminMenuVoChilds);
+            }
+        }
+        return Result.ok(records);
+    }
 }

--
Gitblit v1.9.1