From 8b7258c9427882bb1798f1502eaa35184c6e374e Mon Sep 17 00:00:00 2001
From: linwe <872216996@qq.com>
Date: Fri, 09 Aug 2024 14:29:18 +0800
Subject: [PATCH] 短信指定楼栋发送
---
src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java | 160 ++++++++++++++++++++++++++++++++++++++++++++++++++++-
1 files changed, 157 insertions(+), 3 deletions(-)
diff --git a/src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java b/src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java
index 473107a..a16038c 100644
--- a/src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java
+++ b/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;
/**
* 创建存储桶
@@ -149,15 +165,39 @@
// 修改link
changeLink(bladeFile);
// 返回
- return R.data(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){
+ if (null != bladeFile) {
// 替换url link 前缀
String newLink = ossBuilder.ossProperties().getEndpoint() + bladeFile.getName();
bladeFile.setLink(newLink);
@@ -198,6 +238,32 @@
}
/**
+ * 上传文件指定后缀并保存至附件表
+ *
+ * @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格式！");
+ }
+ }
+
+ /**
* 上传文件并保存至附件表
*
* @param fileName 存储桶对象名称
@@ -206,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, "");
}
/**
@@ -232,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;
}
/**
@@ -262,4 +415,5 @@
return R.success("操作成功");
}
+
}
--
Gitblit v1.9.3