From 15fe445c820f28139e4e7e1a4cb286ad6ae37e6e Mon Sep 17 00:00:00 2001
From: liuhaonan <31457034@qq.com>
Date: 星期一, 06 六月 2022 11:05:17 +0800
Subject: [PATCH] bug修复

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/controller/MenuController.java |   25 +++++++++++++++++++++----
 1 files changed, 21 insertions(+), 4 deletions(-)

diff --git a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/MenuController.java b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/MenuController.java
index dc7e9e6..a09bf91 100644
--- a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/MenuController.java
+++ b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/MenuController.java
@@ -1,10 +1,15 @@
 package com.sandu.ximon.admin.controller;
 
 import com.sandu.common.domain.ResponseVO;
+import com.sandu.common.execption.BusinessException;
 import com.sandu.common.util.ResponseUtil;
 import com.sandu.ximon.admin.param.MenuParam;
+import com.sandu.ximon.admin.security.PermissionConfig;
+import com.sandu.ximon.admin.security.SecurityUtils;
 import com.sandu.ximon.admin.service.MenuService;
 import com.sandu.ximon.dao.bo.MenuNode;
+import com.sandu.ximon.dao.enums.AdministratorEnums;
+import com.sandu.ximon.dao.enums.MenuEnum;
 import lombok.AllArgsConstructor;
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
@@ -20,6 +25,8 @@
 @RequestMapping("/admin/menu")
 public class MenuController {
 
+    private PermissionConfig permissionConfig;
+
     private final MenuService menuService;
 
 //    @GetMapping("treeMenuList")
@@ -32,14 +39,20 @@
 
     @GetMapping("/treeList")
     public ResponseVO<Object> treeList() {
+
+        if (!permissionConfig.check(MenuEnum.MENU_LIST.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         List<MenuNode> menuTrees = menuService.treeList();
         return ResponseUtil.success(menuTrees);
     }
 
     @PostMapping("/update/{menuId}")
     public ResponseVO<Object> update(@PathVariable Long menuId, @Validated @RequestBody MenuParam param) {
-
-        boolean result = menuService.updateMenu(menuId,param);
+        if (!SecurityUtils.getUserDetails().getAdministratorType().equals(AdministratorEnums.ADMIN.getCode())) {
+            throw new BusinessException("闈炶秴绾х鐞嗗憳鏃犳硶鎿嶄綔锛�");
+        }
+        boolean result = menuService.updateMenu(menuId, param);
         if (result) {
             return ResponseUtil.success("缂栬緫鎴愬姛");
         } else {
@@ -49,7 +62,9 @@
 
     @PostMapping("/del/{menuId}")
     public ResponseVO<Object> delMenu(@PathVariable Long menuId) {
-
+        if (!SecurityUtils.getUserDetails().getAdministratorType().equals(AdministratorEnums.ADMIN.getCode())) {
+            throw new BusinessException("闈炶秴绾х鐞嗗憳鏃犳硶鎿嶄綔锛�");
+        }
         boolean result = menuService.delMenu(menuId);
         if (result) {
             return ResponseUtil.success("鍒犻櫎鎴愬姛");
@@ -60,7 +75,9 @@
 
     @PostMapping("/add")
     public ResponseVO<Object> addMenu(@Validated @RequestBody MenuParam param) {
-
+        if (!SecurityUtils.getUserDetails().getAdministratorType().equals(AdministratorEnums.ADMIN.getCode())) {
+            throw new BusinessException("闈炶秴绾х鐞嗗憳鏃犳硶鎿嶄綔锛�");
+        }
         boolean result = menuService.addMenu(param);
         if (result) {
             return ResponseUtil.success("娣诲姞鎴愬姛");

--
Gitblit v1.9.3