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

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/controller/RemoteUpdateController.java |   21 ++++++++++++++++++---
 1 files changed, 18 insertions(+), 3 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 9146987..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
@@ -37,6 +37,9 @@
     @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);
@@ -45,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());
     }
 
@@ -62,9 +71,9 @@
     @AnonymousAccess
     @PostMapping("/StartRemoteUpdate")
     public ResponseVO<Object> startRemoteUpdate(@RequestBody @Validated RemotePrarm remotePrarm) {
-//        if (!permissionConfig.check(MenuEnum.MENU_LIST.getCode())) {
-//            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
-//        }
+        if (!permissionConfig.check(MenuEnum.REMOTE_UPDATE_START.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         String orderType;
         boolean UpdateFlag = false;
 
@@ -128,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("缂哄皯鐩稿叧鍙傛暟锛�");
         }
@@ -164,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