From 326334dbf5de8eeb09cdf8dd7a4511ab821e12ff Mon Sep 17 00:00:00 2001
From: liuhaonan <31457034@qq.com>
Date: 星期二, 10 五月 2022 18:23:41 +0800
Subject: [PATCH] Changes

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/controller/IpVolumeFileController.java |   21 +++++++++++++++++----
 1 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/IpVolumeFileController.java b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/IpVolumeFileController.java
index ce53f53..1df4c32 100644
--- a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/IpVolumeFileController.java
+++ b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/IpVolumeFileController.java
@@ -5,15 +5,15 @@
 import com.github.pagehelper.PageHelper;
 import com.sandu.common.domain.ResponseVO;
 import com.sandu.common.execption.BusinessException;
-import com.sandu.common.file.impl.LocalFileUploadImpl;
 import com.sandu.common.object.BaseConditionVO;
 import com.sandu.common.util.ResponseUtil;
 import com.sandu.ximon.admin.param.IpVolumeFileParam;
+import com.sandu.ximon.admin.security.PermissionConfig;
 import com.sandu.ximon.admin.service.BroadcastV2TaskFileService;
-import com.sandu.ximon.admin.service.FileOperationRecordService;
 import com.sandu.ximon.admin.service.IpVolumeFileService;
 import com.sandu.ximon.dao.domain.BroadcastV2TaskFile;
 import com.sandu.ximon.dao.domain.IpVolumeFile;
+import com.sandu.ximon.dao.enums.MenuEnum;
 import lombok.AllArgsConstructor;
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
@@ -33,16 +33,21 @@
 
     private final IpVolumeFileService ipVolumeFileService;
     private final BroadcastV2TaskFileService v2TaskFileService;
-    private final FileOperationRecordService fileOperationRecordService;
-    private final LocalFileUploadImpl fileUpload;
+    private PermissionConfig permissionConfig;
 
     @PostMapping("/add")
     public ResponseVO<Object> addFile(@RequestBody @Validated IpVolumeFileParam fileParam) {
+        if (!permissionConfig.check(MenuEnum.IP_BROADCAST_FILE_ADD.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         return ResponseUtil.success(ipVolumeFileService.addFile(fileParam));
     }
 
     @PostMapping("/delete/{fileId}")
     public ResponseVO<Object> deleteFile(@PathVariable Long fileId) {
+        if (!permissionConfig.check(MenuEnum.IP_BROADCAST_FILE_DELETE.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         List<BroadcastV2TaskFile> list = v2TaskFileService.list(Wrappers.lambdaQuery(BroadcastV2TaskFile.class).eq(BroadcastV2TaskFile::getFileId, fileId));
         if (list.size() != 0) {
             throw new BusinessException("姝ゆ枃浠舵鍦ㄨ鍏朵粬浠诲姟鍗犵敤");
@@ -52,16 +57,24 @@
 
     @GetMapping("/list")
     public ResponseVO<Object> list(BaseConditionVO baseConditionVO, @RequestParam(value = "keyword", required = false) String keyword) {
+        if (!permissionConfig.check(MenuEnum.IP_BROADCAST_FILE_LIST.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         LambdaQueryWrapper<IpVolumeFile> wrapper = ipVolumeFileService.listFile(baseConditionVO, keyword);
+
         PageHelper.startPage(baseConditionVO.getPageNo(), baseConditionVO.getPageSize());
       /*  if (null != keyword) {
             wrapper.like(IpVolumeFile::getFileName, keyword);
         }*/
+
         return ResponseUtil.success(ipVolumeFileService.list(wrapper));
     }
 
     @PostMapping("/IpVolumeFileUploading")
     public ResponseVO<Object> uploadingFile(@RequestParam("file") MultipartFile file) {
+        if (!permissionConfig.check(MenuEnum.IP_BROADCAST_FILE_UPLOADING.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         return ResponseUtil.success(ipVolumeFileService.upload(file));
     }
 }

--
Gitblit v1.9.3