From b9276c27abdac33f7e3fefccb4600a98db465c6e Mon Sep 17 00:00:00 2001
From: zhanzhiqin <895896009@qq.com>
Date: 星期三, 11 五月 2022 16:49:10 +0800
Subject: [PATCH] fix

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/controller/IpVolumeFileController.java |   42 +++++++++++++++++++++++++++++++++++++-----
 1 files changed, 37 insertions(+), 5 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 aaeb3d5..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
@@ -1,18 +1,25 @@
 package com.sandu.ximon.admin.controller;
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.github.pagehelper.PageHelper;
 import com.sandu.common.domain.ResponseVO;
+import com.sandu.common.execption.BusinessException;
 import com.sandu.common.object.BaseConditionVO;
 import com.sandu.common.util.ResponseUtil;
 import com.sandu.ximon.admin.param.IpVolumeFileParam;
-import com.sandu.ximon.admin.param.ReceiveParam;
+import com.sandu.ximon.admin.security.PermissionConfig;
+import com.sandu.ximon.admin.service.BroadcastV2TaskFileService;
 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.domain.LedSFile;
+import com.sandu.ximon.dao.enums.MenuEnum;
 import lombok.AllArgsConstructor;
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.util.List;
 
 /**
  * @Author liuhaonan
@@ -25,24 +32,49 @@
 public class IpVolumeFileController {
 
     private final IpVolumeFileService ipVolumeFileService;
+    private final BroadcastV2TaskFileService v2TaskFileService;
+    private PermissionConfig permissionConfig;
 
     @PostMapping("/add")
-    public ResponseVO<Object> addFile(@RequestBody @Validated IpVolumeFileParam fileParam){
+    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){
+    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("姝ゆ枃浠舵鍦ㄨ鍏朵粬浠诲姟鍗犵敤");
+        }
         return ResponseUtil.success(ipVolumeFileService.deleteFile(fileId));
     }
 
     @GetMapping("/list")
-    public ResponseVO<Object> list(BaseConditionVO baseConditionVO, @RequestParam(value = "keyword", required = false) String keyword){
+    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