From 56d8ffd22d0fd65c6a582de2e57286f9f37f4883 Mon Sep 17 00:00:00 2001
From: zhanzhiqin <895896009@qq.com>
Date: 星期二, 10 五月 2022 17:35:41 +0800
Subject: [PATCH] 一键求助

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/controller/RemoteUpdateController.java |   22 ++++++++++++++++++++++
 1 files changed, 22 insertions(+), 0 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..4fb1511 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,10 +31,15 @@
 @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);
@@ -41,11 +48,17 @@
     @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());
     }
 
@@ -58,6 +71,9 @@
     @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;
 
@@ -121,6 +137,9 @@
     @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("缂哄皯鐩稿叧鍙傛暟锛�");
         }
@@ -157,6 +176,9 @@
     @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