From a8a707e602d644f3207b077d55bfad1a7c3642b7 Mon Sep 17 00:00:00 2001
From: LHN <31457034@qq.com>
Date: 星期三, 19 十月 2022 17:47:48 +0800
Subject: [PATCH] changes

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/controller/LightPoleHeelingController.java |   53 +++++++++++++++++++++++++++++++++++++++--------------
 1 files changed, 39 insertions(+), 14 deletions(-)

diff --git a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/LightPoleHeelingController.java b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/LightPoleHeelingController.java
index 07bf093..e1266b7 100644
--- a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/LightPoleHeelingController.java
+++ b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/LightPoleHeelingController.java
@@ -6,8 +6,10 @@
 import com.sandu.ximon.admin.param.C3ChargingParam;
 import com.sandu.ximon.admin.param.LightPoleHeelingListParam;
 import com.sandu.ximon.admin.param.LightPoleHeelingParam;
+import com.sandu.ximon.admin.security.PermissionConfig;
 import com.sandu.ximon.admin.service.LightPoleHeelingService;
 import com.sandu.ximon.dao.domain.LightPoleHeeling;
+import com.sandu.ximon.dao.enums.MenuEnum;
 import lombok.AllArgsConstructor;
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
@@ -15,13 +17,14 @@
 import java.util.List;
 
 /**
- * 鐏潌鍊炬枩
+ * 鐏潌鍊炬枩璀﹀憡鍒楄〃
  */
 @RestController
 @AllArgsConstructor
 @RequestMapping("/v1/admin/LightPoleHeeling")
 public class LightPoleHeelingController {
     private final LightPoleHeelingService lightPoleHeelingService;
+    private PermissionConfig permissionConfig;
 
     /**
      * 妯$硦鏌ヨ
@@ -31,6 +34,9 @@
     @PostMapping("/listLightPoleHeelingByKeyword")
     public ResponseVO<Object> listLightPoleHeelingByKeyword(BaseConditionVO baseConditionVO
             , @RequestBody(required = false) LightPoleHeelingListParam lightPoleHeelingListParam) {
+        if (!permissionConfig.check(MenuEnum.LIGHT_POLE_HEELING_LIST.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         List<LightPoleHeeling> list = lightPoleHeelingService.listLightPoleHeelingByKeyword(baseConditionVO, lightPoleHeelingListParam);
         return ResponseUtil.successPage(list);
     }
@@ -42,6 +48,9 @@
      */
     @PostMapping("/LightPoleHeelingHeartbeatPackage")
     public ResponseVO<Object> LightPoleHeelingHeartbeatPackage(@RequestBody @Validated LightPoleHeelingParam lightPoleHeelingParam) {
+        if (!permissionConfig.check(MenuEnum.LIGHT_POLE_HEELING_HEARTBEAT_PACKAGE.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         return ResponseUtil.success(lightPoleHeelingService.LightPoleHeelingHeartbeatPackage(lightPoleHeelingParam.getDeviceName()));
 
     }
@@ -53,6 +62,9 @@
      */
     @PostMapping("/QueryIntervalTime")
     public ResponseVO<Object> QueryIntervalTime(@RequestBody @Validated LightPoleHeelingParam lightPoleHeelingParam) {
+        if (!permissionConfig.check(MenuEnum.LIGHT_POLE_HEELING_HEARTBEAT_PACKAGE_QUERY_TIME.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         return ResponseUtil.success(lightPoleHeelingService.QueryIntervalTime(lightPoleHeelingParam.getDeviceName()));
 
     }
@@ -64,11 +76,16 @@
      */
     @PostMapping("/RestartLightPoleHeeling")
     public ResponseVO<Object> RestartLightPoleHeeling(@RequestBody @Validated LightPoleHeelingParam lightPoleHeelingParam) {
-        if ("鎿嶄綔鎴愬姛".equals(lightPoleHeelingService.RestartLightPoleHeeling(lightPoleHeelingParam.getDeviceName()))) {
-            return ResponseUtil.success(lightPoleHeelingService.RestartLightPoleHeeling(lightPoleHeelingParam.getDeviceName()));
-        } else {
-            return ResponseUtil.fail(lightPoleHeelingService.RestartLightPoleHeeling(lightPoleHeelingParam.getDeviceName()));
+        if (!permissionConfig.check(MenuEnum.LIGHT_POLE_HEELING_RESTART.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
         }
+        String result = lightPoleHeelingService.RestartLightPoleHeeling(lightPoleHeelingParam.getDeviceName());
+        if ("鎿嶄綔鎴愬姛".equals(result)) {
+            return ResponseUtil.success(result);
+        } else {
+            return ResponseUtil.fail(result);
+        }
+
     }
 
     /**
@@ -76,11 +93,16 @@
      */
     @PostMapping("/SetHeartbeatPacketTime")
     public ResponseVO<Object> SetHeartbeatPacketTime(@RequestBody @Validated LightPoleHeelingParam lightPoleHeelingParam) {
-        if ("鎿嶄綔鎴愬姛".equals(lightPoleHeelingService.RestartLightPoleHeeling(lightPoleHeelingParam.getDeviceName()))) {
-            return ResponseUtil.success(lightPoleHeelingService.SetHeartbeatPacketTime(lightPoleHeelingParam.getDeviceName(), lightPoleHeelingParam.getTime()));
-        } else {
-            return ResponseUtil.fail(lightPoleHeelingService.SetHeartbeatPacketTime(lightPoleHeelingParam.getDeviceName(), lightPoleHeelingParam.getTime()));
+        if (!permissionConfig.check(MenuEnum.LIGHT_POLE_HEELING_HEARTBEAT_PACKAGE_SET_TIME.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
         }
+        String result = lightPoleHeelingService.SetHeartbeatPacketTime(lightPoleHeelingParam.getDeviceName(), lightPoleHeelingParam.getTime());
+        if ("鎿嶄綔鎴愬姛".equals(result)) {
+            return ResponseUtil.success(result);
+        } else {
+            return ResponseUtil.fail(result);
+        }
+
     }
 
     /**
@@ -88,11 +110,14 @@
      */
     @PostMapping("/SetSensorState")
     public ResponseVO<Object> SetSensorState(@RequestBody @Validated LightPoleHeelingParam lightPoleHeelingParam) {
-        if ("鎿嶄綔鎴愬姛".equals(lightPoleHeelingService.RestartLightPoleHeeling(lightPoleHeelingParam.getDeviceName()))) {
-            return ResponseUtil.success(lightPoleHeelingService.SetSensorState(lightPoleHeelingParam.getDeviceName()));
-        } else {
-            return ResponseUtil.fail(lightPoleHeelingService.SetSensorState(lightPoleHeelingParam.getDeviceName()));
+        if (!permissionConfig.check(MenuEnum.LIGHT_POLE_HEELING_SET_SENSORSTATE.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
         }
-
+        String result = lightPoleHeelingService.SetSensorState(lightPoleHeelingParam.getDeviceName());
+        if ("鎿嶄綔鎴愬姛".equals(result)) {
+            return ResponseUtil.success(result);
+        } else {
+            return ResponseUtil.fail(result);
+        }
     }
 }

--
Gitblit v1.9.3