From 6467405b15ec5a7ce03faa83909aec0dfc7672fa Mon Sep 17 00:00:00 2001
From: liuhaonan <31457034@qq.com>
Date: 星期五, 02 九月 2022 16:00:59 +0800
Subject: [PATCH] changes

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/service/MenuService.java |   49 +++++++++++++++++++++++++++----------------------
 1 files changed, 27 insertions(+), 22 deletions(-)

diff --git a/ximon-admin/src/main/java/com/sandu/ximon/admin/service/MenuService.java b/ximon-admin/src/main/java/com/sandu/ximon/admin/service/MenuService.java
index 04b36e6..02994a6 100644
--- a/ximon-admin/src/main/java/com/sandu/ximon/admin/service/MenuService.java
+++ b/ximon-admin/src/main/java/com/sandu/ximon/admin/service/MenuService.java
@@ -2,6 +2,7 @@
 
 import cn.hutool.core.collection.CollectionUtil;
 import cn.hutool.core.util.StrUtil;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.sandu.common.enums.RoleLevelStatus;
 import com.sandu.common.execption.BusinessException;
@@ -35,17 +36,20 @@
 public class MenuService extends BaseServiceImpl<MenuMapper, Menu> {
 
     private final RoleMenuRelationService roleMenuRelationService;
+    private MenuMapper menuMapper;
 
     /**
      * 鑾峰彇瑙掕壊涓嬬殑鑿滃崟
      */
     public List<Menu> listByRoleIds(List<Long> roleIdList) {
-        List<Long> menuIdList = roleMenuRelationService.list(Wrappers.lambdaQuery(RoleMenuRelation.class).in(RoleMenuRelation::getRoleId, roleIdList).select(RoleMenuRelation::getMenuId))
+        List<Long> menuIdList = roleMenuRelationService.list(Wrappers.lambdaQuery(RoleMenuRelation.class)
+                        .in(RoleMenuRelation::getRoleId, roleIdList).select(RoleMenuRelation::getMenuId))
                 .stream().map(RoleMenuRelation::getMenuId).distinct().collect(Collectors.toList());
         if (CollectionUtil.isEmpty(menuIdList)) {
             return null;
         }
-        return listByIds(menuIdList);
+
+        return menuMapper.listMenuById(menuIdList);
     }
 
     public boolean addMenu(MenuParam param) {
@@ -53,21 +57,6 @@
         BeanUtils.copyProperties(param, menu);
         return save(menu);
     }
-
-//    public Set<Menu> listMenuByRoles(Set<Role> roles) {
-//        List<Long> roleIds = roles.stream().map(Role::getId).collect(Collectors.toList());
-//        Set<Menu> menus = menuDao.listMenuByRoleIds(roleIds);
-//        return menus;
-//    }
-//
-//    /**
-//     * 鎶婅彍鍗曡浆鎴愭爲褰㈢粨鏋�
-//     */
-//    public List<MenuNode> buildTree(List<Menu> menuList) {
-//        return menuList.stream()
-//                .filter(menu -> menu.getPid().equals(0L))
-//                .map(menu -> covertMenuNode(menu, menuList)).collect(Collectors.toList());
-//    }
 
     private MenuNode covertMenuNode(Menu menu, List<Menu> menuList) {
         MenuNode node = new MenuNode();
@@ -93,18 +82,18 @@
 
         Menu menu = new Menu();
         BeanUtils.copyProperties(param, menu);
+        menu.setType(one.getType());
         menu.setId(menuId);
         return updateById(menu);
     }
 
     public List<MenuNode> treeList() {
-        List<Menu> list = list();
+        LambdaQueryWrapper<Menu> menuLambdaQueryWrapper = Wrappers.lambdaQuery(Menu.class).orderByAsc(Menu::getSeq);
+        List<Menu> list = list(menuLambdaQueryWrapper);
         return list.stream()
                 .filter(menu -> menu.getPid().equals(0L))
                 .map(menu -> covertMenuNode(menu, list)).collect(Collectors.toList());
     }
-
-    private MenuMapper menuMapper;
 
     public boolean delMenu(Long menuId) {
         Menu one = getById(menuId);
@@ -118,11 +107,27 @@
             throw new BusinessException("瓒呯骇绠$悊鍛樻墠鍙互鍒犻櫎鑿滃崟");
         }
         List<Long> list = menuMapper.listMenuIfBinding(menuId);
-        if (list.size() == 0) {
+        List<Long> list1 = menuMapper.listMenuIfBinding1(menuId);
+
+        if (list.size() == 0 && list1.size() == 0) {
             return removeById(menuId);
-        }else{
+        } else {
             throw new BusinessException("鑿滃崟姝e湪浣跨敤锛岃鍕垮垹闄�");
         }
 
     }
+
+    /**
+     * 閫氳繃鑿滃崟ID闆嗗悎鑾峰彇鑿滃崟瀹炰綋鍒楄〃
+     *
+     * @param menuIds 鑿滃崟ID闆嗗悎
+     */
+    public List<MenuNode> getUserPermissionListById(List<Long> menuIds) {
+        List<Menu> menuList = menuMapper.getUserPermissionListById(menuIds);
+
+        List<MenuNode> resultList = menuList.stream()
+                .filter(menu -> menu.getPid().equals(0L))
+                .map(menu -> covertMenuNode(menu, menuList)).collect(Collectors.toList());
+        return resultList;
+    }
 }

--
Gitblit v1.9.3