代码优化

linwe

2024-07-15 8f5aeec641d916806553ef9772d55e17e93db150

 src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java

@@ -16,9 +16,14 @@
 */
package org.springblade.modules.resource.endpoint;

import com.alibaba.fastjson.JSONObject;
import io.swagger.annotations.Api;
import lombok.AllArgsConstructor;
import lombok.SneakyThrows;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springblade.common.utils.ImageUtils;
import org.springblade.common.utils.WeiXinSecurityUtil;
import org.springblade.core.launch.constant.AppConstant;
import org.springblade.core.oss.model.BladeFile;
import org.springblade.core.oss.model.OssFile;
@@ -30,9 +35,14 @@
import org.springblade.core.tool.utils.Func;
import org.springblade.modules.resource.builder.oss.OssBuilder;
import org.springblade.modules.resource.entity.Attach;
import org.springblade.modules.resource.entity.AttachData;
import org.springblade.modules.resource.service.IAttachDataService;
import org.springblade.modules.resource.service.IAttachService;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.util.Arrays;
import java.util.List;

/**
 * 对象存储端点
@@ -45,6 +55,7 @@
@Api(value = "对象存储端点", tags = "对象存储端点")
@RequestMapping(AppConstant.APPLICATION_RESOURCE_NAME + "/oss/endpoint")
public class OssEndpoint {
   private static final Logger logger = LoggerFactory.getLogger(OssEndpoint.class);

   /**
    * 对象存储构建类
@@ -55,6 +66,11 @@
    * 附件表服务
    */
   private final IAttachService attachService;

   /**
    * 附件数据表服务
    */
   private final IAttachDataService attachDataService;

   /**
    * 创建存储桶
@@ -146,7 +162,46 @@
   @PostMapping("/put-file")
   public R<BladeFile> putFile(@RequestParam MultipartFile file) {
      BladeFile bladeFile = ossBuilder.template().putFile(file.getOriginalFilename(), file.getInputStream());
      return R.data(bladeFile);
      // 修改link
      changeLink(bladeFile);
      // 返回
      return R.data(200, bladeFile, "");
   }

   /**
    * 上传文件
    *
    * @param file 文件
    * @return ObjectStat
    */
   @SneakyThrows
   @PostMapping("/put-pdfFile")
   public R<BladeFile> putPdfFile(@RequestParam MultipartFile file) {
      String originalFilename = file.getOriginalFilename();
      String fileExtension = originalFilename.substring(originalFilename.lastIndexOf(".") + 1);
      if (fileExtension.equalsIgnoreCase("pdf")) {
         BladeFile bladeFile = ossBuilder.template().putFile(file.getOriginalFilename(), file.getInputStream());
         // 修改link
         changeLink(bladeFile);
         // 返回
         return R.data(200, bladeFile, "");
      } else {
         return R.fail("上传的附件必须是pdf格式！");
      }

   }

   /**
    * 修改link
    *
    * @param bladeFile
    */
   private void changeLink(BladeFile bladeFile) {
      if (null != bladeFile) {
         // 替换url link 前缀
         String newLink = ossBuilder.ossProperties().getEndpoint() + bladeFile.getName();
         bladeFile.setLink(newLink);
      }
   }

   /**
@@ -176,7 +231,36 @@
      BladeFile bladeFile = ossBuilder.template().putFile(fileName, file.getInputStream());
      Long attachId = buildAttach(fileName, file.getSize(), bladeFile);
      bladeFile.setAttachId(attachId);
      // 修改link
      changeLink(bladeFile);
      // 返回
      return R.data(bladeFile);
   }

   /**
    * 上传文件指定后缀并保存至附件表
    *
    * @param file 文件
    * @return ObjectStat
    */
   @SneakyThrows
   @PostMapping("/put-file-attach-suffix")
   public R<BladeFile> putFileAttachSuffix(@RequestParam MultipartFile file, @RequestParam String suffix) {
      String fileName = file.getOriginalFilename();
      String fileExtension = fileName.substring(fileName.lastIndexOf(".") + 1);
      String[] split = suffix.split(",");
      List<String> list = Arrays.asList(split);
      if (list.contains(fileExtension)) {
         BladeFile bladeFile = ossBuilder.template().putFile(fileName, file.getInputStream());
         Long attachId = buildAttach(fileName, file.getSize(), bladeFile);
         bladeFile.setAttachId(attachId);
         // 修改link
         changeLink(bladeFile);
         // 返回
         return R.data(bladeFile);
      } else {
         return R.fail("上传的附件必须是pdf,doc,docx格式！");
      }
   }

   /**
@@ -188,11 +272,71 @@
    */
   @SneakyThrows
   @PostMapping("/put-file-attach-by-name")
   public R<BladeFile> putFileAttach(@RequestParam String fileName, @RequestParam MultipartFile file) {
   public R<BladeFile> putFileAttach(@RequestParam(required = false) String fileName, @RequestParam MultipartFile file) {
      BladeFile bladeFile = ossBuilder.template().putFile(fileName, file.getInputStream());
      Long attachId = buildAttach(fileName, file.getSize(), bladeFile);
      bladeFile.setAttachId(attachId);
      return R.data(bladeFile);
   }

   /**
    * 自定义前缀上传文件
    *
    * @param file       文件
    * @param prefixPath 文件
    * @param isCheck    是否校验
    * @return ObjectStat
    */
   @SneakyThrows
   @PostMapping("/put-file-by-prefix-path")
   public R<BladeFile> putFileByPrefixPath(@RequestParam MultipartFile file,
                                 @RequestParam(required = false) String prefixPath,
                                 @RequestParam(required = false) Integer isCheck) {
      // 不传默认不检验
      if (null != isCheck) {
         String checkResult = WeiXinSecurityUtil.checkImg(file);
         JSONObject jsonObject = JSONObject.parseObject(checkResult);
         Integer errorCode = Integer.parseInt(jsonObject.getString("errcode"));
         if (errorCode != 0) {
            return R.data(201, null, "图片存在敏感内容，请更换其他图片!");
         }
      }
      BladeFile bladeFile = ossBuilder.templateByPrefixPath(prefixPath).putFile(file.getOriginalFilename(), file.getInputStream());
      // 修改link
      changeLink(bladeFile);
      return R.data(200, bladeFile, "");
   }

   /**
    * 自定义前缀上传文件并保存至附件表
    *
    * @param file       文件
    * @param prefixPath 文件
    * @param isCheck    是否校验
    * @return ObjectStat
    */
   @SneakyThrows
   @PostMapping("/put-file-attach-by-prefix-path")
   public R<BladeFile> putFileAttachByPrefixPath(@RequestParam MultipartFile file,
                                      @RequestParam(required = false) String prefixPath,
                                      @RequestParam(required = false) Integer isCheck) {
      // 不传默认不检验
      if (null != isCheck) {
         String checkResult = WeiXinSecurityUtil.checkImg(file);
         JSONObject jsonObject = JSONObject.parseObject(checkResult);
         Integer errorCode = Integer.parseInt(jsonObject.getString("errcode"));
         if (errorCode != 0) {
            return R.data(201, null, "图片存在敏感内容，请更换其他图片!");
         }
      }
      String fileName = file.getOriginalFilename();
      BladeFile bladeFile = ossBuilder.templateByPrefixPath(prefixPath).putFile(file.getOriginalFilename(), file.getInputStream());
      Long attachId = buildAttach(fileName, file.getSize(), bladeFile);
      buildAttachData(attachId, fileName, file.getSize(), bladeFile, file);
      bladeFile.setAttachId(attachId);
      // 修改link
      changeLink(bladeFile);
      return R.data(200, bladeFile, "");
   }

   /**
@@ -214,6 +358,33 @@
      attach.setExtension(fileExtension);
      attachService.save(attach);
      return attach.getId();
   }

   /**
    * 构建附件数据表
    *
    * @param attachId  附件id
    * @param fileName  文件名
    * @param fileSize  文件大小
    * @param bladeFile 对象存储文件
    * @return attachId
    */
   private Long buildAttachData(Long attachId, String fileName, Long fileSize, BladeFile bladeFile, MultipartFile file) {
      try {
         String fileExtension = FileUtil.getFileExtension(fileName);
         AttachData attach = new AttachData();
         attach.setAttachId(attachId);
         attach.setName(bladeFile.getName());
         attach.setOriginalName(bladeFile.getOriginalName());
         attach.setSize(fileSize);
         attach.setExtension(fileExtension);
         attach.setData(ImageUtils.mulToBase64(file));
         attachDataService.save(attach);
         return attach.getId();
      } catch (Exception e) {
         logger.error("附件数据表保存失败", e);
      }
      return 0L;
   }

   /**
@@ -244,4 +415,5 @@
      return R.success("操作成功");
   }


}