From f87640188a5322fcec02415ea25fab8e2450dd0a Mon Sep 17 00:00:00 2001
From: zhanzhiqin <895896009@qq.com>
Date: 星期四, 08 九月 2022 14:13:19 +0800
Subject: [PATCH] 大气fix

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AirEquipmentController.java |   12 ++++++++++++
 dao/src/main/java/com/sandu/ximon/dao/enums/MenuEnum.java                              |    3 +++
 2 files changed, 15 insertions(+), 0 deletions(-)

diff --git a/dao/src/main/java/com/sandu/ximon/dao/enums/MenuEnum.java b/dao/src/main/java/com/sandu/ximon/dao/enums/MenuEnum.java
index 5e2f3f0..f20d9de 100644
--- a/dao/src/main/java/com/sandu/ximon/dao/enums/MenuEnum.java
+++ b/dao/src/main/java/com/sandu/ximon/dao/enums/MenuEnum.java
@@ -69,6 +69,9 @@
     ADMIN_INFO_UPDATE("缂栬緫绠$悊鍛樹俊鎭�"),
     ADMIN_DELETE("鍒犻櫎绠$悊鍛�"),
     AIR_EQUIPMENT_DELETE("鍒犻櫎澶ф皵璁惧"),
+    AIR_HEARTBEATTIME("澶ф皵璁惧蹇冭烦鍖呮椂闂�"),
+    AIR_QUERYVERSION("鏌ヨ杞‖浠剁増鏈�"),
+    AIR_REBOOT("杞噸鍚�"),
     AIR_DATA_UPDATE("鎵嬪姩鏇存柊澶ф皵鏁版嵁"),
     BANNER_ADD("娣诲姞寰俊logo"),
     BANNER_UPDATE("缂栬緫寰俊logo"),
diff --git a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AirEquipmentController.java b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AirEquipmentController.java
index e082e00..44448f1 100644
--- a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AirEquipmentController.java
+++ b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AirEquipmentController.java
@@ -56,6 +56,9 @@
      */
     @GetMapping("/QueryHeartBeatTime")
     public ResponseVO<Object> QueryHeartBeatTime(@RequestParam(value = "mac", required = false) String mac) {
+        if (!permissionConfig.check(MenuEnum.AIR_HEARTBEATTIME.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         return ResponseUtil.success(airEquipmentService.QueryHeartBeatTime(mac));
     }
 
@@ -69,6 +72,9 @@
     @GetMapping("/setHeartBeatTime")
     public ResponseVO<Object> setHeartBeatTime(@RequestParam(value = "mac", required = false) String mac,
                                                @RequestParam(value = "time", required = false) Integer time) {
+        if (!permissionConfig.check(MenuEnum.AIR_HEARTBEATTIME.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         return ResponseUtil.success(airEquipmentService.setHeartBeatTime(mac, time));
     }
 
@@ -91,6 +97,9 @@
      */
     @GetMapping("/QueryVersion")
     public ResponseVO<Object> QueryVersion(@RequestParam(value = "mac", required = false) String mac) {
+        if (!permissionConfig.check(MenuEnum.AIR_QUERYVERSION.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         return ResponseUtil.success(airEquipmentService.QueryVersion(mac));
     }
 
@@ -102,6 +111,9 @@
      */
     @GetMapping("/Reboot")
     public ResponseVO<Object> Reboot(@RequestParam(value = "mac", required = false) String mac) {
+        if (!permissionConfig.check(MenuEnum.AIR_REBOOT.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         return ResponseUtil.success(airEquipmentService.Reboot(mac));
     }
 

--
Gitblit v1.9.3