From f736df1648a4d9bb8620d38cf4421cf36eacdfc0 Mon Sep 17 00:00:00 2001
From: liuhaonan <konodioda2333@vip.qq.com>
Date: 星期五, 04 三月 2022 14:37:39 +0800
Subject: [PATCH] 功能完善

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/service/MenuService.java |   94 ++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 92 insertions(+), 2 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 7ed7d10..4557978 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
@@ -3,15 +3,25 @@
 import cn.hutool.core.collection.CollectionUtil;
 import cn.hutool.core.util.StrUtil;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.sandu.common.enums.RoleLevelStatus;
+import com.sandu.common.execption.BusinessException;
+import com.sandu.common.execption.EntityExistException;
 import com.sandu.common.service.impl.BaseServiceImpl;
+import com.sandu.common.util.SpringContextHolder;
+import com.sandu.ximon.admin.param.MenuParam;
+import com.sandu.ximon.admin.security.SecurityUtils;
+import com.sandu.ximon.dao.bo.MenuNode;
 import com.sandu.ximon.dao.domain.AdminRoleRelation;
 import com.sandu.ximon.dao.domain.Menu;
+import com.sandu.ximon.dao.domain.Role;
 import com.sandu.ximon.dao.domain.RoleMenuRelation;
 import com.sandu.ximon.dao.mapper.AdminRoleRelationMapper;
 import com.sandu.ximon.dao.mapper.MenuMapper;
 import lombok.AllArgsConstructor;
+import org.springframework.beans.BeanUtils;
 import org.springframework.stereotype.Service;
 
+import java.util.Collections;
 import java.util.List;
 import java.util.Set;
 import java.util.stream.Collectors;
@@ -25,6 +35,7 @@
 public class MenuService extends BaseServiceImpl<MenuMapper, Menu> {
 
     private final RoleMenuRelationService roleMenuRelationService;
+
     /**
      * 鑾峰彇瑙掕壊涓嬬殑鑿滃崟
      */
@@ -34,7 +45,86 @@
         if (CollectionUtil.isEmpty(menuIdList)) {
             return null;
         }
-        List<Menu> menus = listByIds(menuIdList);
-        return menus;
+        return listByIds(menuIdList);
+    }
+
+    public boolean addMenu(MenuParam param) {
+        Menu menu = new Menu();
+        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();
+        BeanUtils.copyProperties(menu, node);
+        List<MenuNode> children = menuList.stream()
+                .filter(subMenu -> subMenu.getPid().equals(menu.getId()))
+                .map(subMenu -> covertMenuNode(subMenu, menuList)).collect(Collectors.toList());
+        node.setChildren(children);
+        return node;
+    }
+
+    public boolean updateMenu(Long menuId, MenuParam param) {
+        Menu one = getById(menuId);
+        if (one == null) {
+            throw new BusinessException("鎵句笉鍒拌彍鍗曚俊鎭�");
+        }
+
+        List<Role> roles = SpringContextHolder.getBean(RoleService.class).listByAdminId(SecurityUtils.getUserId());
+        Integer roleLevel = roles.stream().map(Role::getLevel).min(Integer::compareTo).orElse(RoleLevelStatus.COMMON.getCode());
+        if (!RoleLevelStatus.SUPER.getCode().equals(roleLevel)) {
+            throw new BusinessException("瓒呯骇绠$悊鍛樻墠鍙互缂栬緫鑿滃崟");
+        }
+
+        Menu menu = new Menu();
+        BeanUtils.copyProperties(param, menu);
+        menu.setId(menuId);
+        return updateById(menu);
+    }
+
+    public List<MenuNode> treeList() {
+        List<Menu> list = list();
+        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);
+        if (one == null) {
+            throw new BusinessException("鎵句笉鍒拌彍鍗曚俊鎭�");
+        }
+
+        List<Role> roles = SpringContextHolder.getBean(RoleService.class).listByAdminId(SecurityUtils.getUserId());
+        Integer roleLevel = roles.stream().map(Role::getLevel).min(Integer::compareTo).orElse(RoleLevelStatus.COMMON.getCode());
+        if (!RoleLevelStatus.SUPER.getCode().equals(roleLevel)) {
+            throw new BusinessException("瓒呯骇绠$悊鍛樻墠鍙互鍒犻櫎鑿滃崟");
+        }
+        List<Long> list = menuMapper.listMenuIfBinding(menuId);
+        List<Long> list1 = menuMapper.listMenuIfBinding1(menuId);
+
+        if (list.size() == 0 && list1.size() == 0) {
+            return removeById(menuId);
+        } else {
+            throw new BusinessException("鑿滃崟姝e湪浣跨敤锛岃鍕垮垹闄�");
+        }
+
     }
 }

--
Gitblit v1.9.3