From 3eda4de769e0852dd051182efe74c2a49865fa84 Mon Sep 17 00:00:00 2001
From: zhanzhiqin <895896009@qq.com>
Date: 星期一, 06 六月 2022 18:46:32 +0800
Subject: [PATCH] 灯杆倾斜处理

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/controller/RemoteUpdateController.java |   28 ++++++++++++++++++++++------
 1 files changed, 22 insertions(+), 6 deletions(-)

diff --git a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/RemoteUpdateController.java b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/RemoteUpdateController.java
index 674d261..b445138 100644
--- a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/RemoteUpdateController.java
+++ b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/RemoteUpdateController.java
@@ -8,9 +8,11 @@
 import com.sandu.ximon.admin.manager.iot.rrpc.enums.A5OrderEnum;
 import com.sandu.ximon.admin.param.RemotePrarm;
 import com.sandu.ximon.admin.param.RemoteUpdateParam;
+import com.sandu.ximon.admin.security.PermissionConfig;
 import com.sandu.ximon.admin.service.RemoteUpdateService;
 import com.sandu.ximon.admin.utils.HexUtils;
 import com.sandu.ximon.admin.utils.StringUtil;
+import com.sandu.ximon.dao.enums.MenuEnum;
 import lombok.AllArgsConstructor;
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
@@ -29,23 +31,32 @@
 @RequestMapping("/v1/remoteupdate")
 public class RemoteUpdateController {
     private RemoteUpdateService remoteUpdateService;
+    private PermissionConfig permissionConfig;
 
-    @AnonymousAccess
+
     @PostMapping("/add")
     public ResponseVO<Object> addRemoteUpdateFile(@RequestBody @Validated RemoteUpdateParam param) {
+        if (!permissionConfig.check(MenuEnum.REMOTE_UPDATE_FILE_ADD.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         boolean flag = remoteUpdateService.addRemoteUpdateFile(param.getFilename(), param.getFileType()
                 , param.getSoftwareVersion(), param.getHardwareVersion(), param.getAliAddress(), param.getFileLength());
         return ResponseUtil.success(flag);
     }
 
-    @AnonymousAccess
     @GetMapping("/getRemoteFileList")
     public ResponseVO<Object> getRemoteFileList() {
+        if (!permissionConfig.check(MenuEnum.REMOTE_UPDATE_FILE_LIST.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         return ResponseUtil.success(remoteUpdateService.getRemoteFileList());
     }
 
     @GetMapping("/getRemoteUpdateTypeList")
     public ResponseVO<Object> getRemoteUpdateTypeList() {
+        if (!permissionConfig.check(MenuEnum.REMOTE_UPDATE_TYPE_LIST.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         return ResponseUtil.success(remoteUpdateService.getRemoteUpdateTypeList());
     }
 
@@ -54,10 +65,11 @@
      *
      * @return
      */
-    //TODO
-    @AnonymousAccess
     @PostMapping("/StartRemoteUpdate")
     public ResponseVO<Object> startRemoteUpdate(@RequestBody @Validated RemotePrarm remotePrarm) {
+        if (!permissionConfig.check(MenuEnum.REMOTE_UPDATE_START.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         String orderType;
         boolean UpdateFlag = false;
 
@@ -118,9 +130,11 @@
      * @param remotePrarm
      * @return
      */
-    @AnonymousAccess
     @PostMapping("/searchUpdateResult")
     public ResponseVO<Object> searchUpdateResult(@RequestBody RemotePrarm remotePrarm) {
+        if (!permissionConfig.check(MenuEnum.REMOTE_UPDATE_VERSION.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         if (remotePrarm.getMac() == null || remotePrarm.getDeviceType() == null) {
             throw new BusinessException("缂哄皯鐩稿叧鍙傛暟锛�");
         }
@@ -154,9 +168,11 @@
      * @param remotePrarm
      * @return
      */
-    @AnonymousAccess
     @PostMapping("/stopUpdate")
     public ResponseVO<Object> stopUpdate(@RequestBody RemotePrarm remotePrarm) {
+        if (!permissionConfig.check(MenuEnum.REMOTE_UPDATE_STOP.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         if (remotePrarm.getMac() == null || remotePrarm.getDeviceType() == null) {
             throw new BusinessException("缂哄皯鐩稿叧鍙傛暟锛�");
         }

--
Gitblit v1.9.3