Source/UBCS/ubcs-ops/ubcs-resource/src/main/java/com/vci/ubcs/resource/controller/FileController.java

@@ -1,8 +1,8 @@
package com.vci.ubcs.resource.controller;

import com.alibaba.fastjson.JSON;
import com.alibaba.nacos.common.utils.StringUtils;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.vci.ubcs.common.validator.ComprehensiveFileValidator;
import com.vci.ubcs.resource.dto.FileObjectDTO;
import com.vci.ubcs.resource.dto.FileReleaseDTO;
import com.vci.ubcs.resource.dto.FileShareDTO;
@@ -10,14 +10,11 @@
import com.vci.ubcs.resource.service.IFileService;
import com.vci.ubcs.resource.utils.FileDownloadUtil;
import com.vci.ubcs.resource.vo.FileObjectVO;
import com.vci.ubcs.starter.exception.VciBaseException;
import com.vci.ubcs.starter.web.util.ControllerUtil;
import com.vci.ubcs.starter.web.util.LangBaseUtil;
import com.vci.ubcs.starter.web.util.VciBaseUtil;
import lombok.extern.java.Log;
import lombok.extern.slf4j.Slf4j;
import org.springblade.core.mp.support.Query;
import org.springblade.core.oss.MinioTemplate;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.utils.StringUtil;
import org.springframework.beans.factory.annotation.Autowired;
@@ -45,6 +42,12 @@
    */
   @Autowired
   private IFileService fileService;

   /**
    * 文件安全检查
    */
   @Autowired
   private ComprehensiveFileValidator fileValidator;

   /**
    * 根据文件主键下载文件
@@ -82,7 +85,7 @@
         if(StringUtil.isBlank(msg)){
            msg = "未知错误";
         }
         log.debug(msg);
         log.error(msg);
         return R.fail(msg);
      }
      return R.success("删除成功");
@@ -97,6 +100,12 @@
   @PostMapping("/uploadFile")
   public R<FileObjectVO> uploadFile(MultipartFile file, FileObjectDTO fileObjectDTO){
      if (file != null ) {
         // 使用文件安全验证器
         ComprehensiveFileValidator.UploadValidationResult result = fileValidator.validateFile(file);
         if (!result.isValid()) {
            return R.fail(result.getMessage());
         }
         //上传文件
         return fileService.uploadFile(file, fileObjectDTO);
      } else {
         return R.fail("无上传的文件");