From f2a081018c58042c642db2e59203de9ab9412c83 Mon Sep 17 00:00:00 2001
From: lin <sbla5888@163.com>
Date: Sat, 23 Mar 2024 16:29:24 +0800
Subject: [PATCH] 统计查询调整+议题附件

---
 src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java |   58 +++++++++++++++++++++++++++++++++++++++++-----------------
 1 files changed, 41 insertions(+), 17 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 c40fa1e..5aab59f 100644
--- a/src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java
+++ b/src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java
@@ -171,15 +171,39 @@
 		// 修改link
 		changeLink(bladeFile);
 		// 返回
-		return R.data(200,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);
@@ -238,9 +262,9 @@
 	/**
 	 * 自定义前缀上传文件
 	 *
-	 * @param file 文件
+	 * @param file       文件
 	 * @param prefixPath 文件
-	 * @param isCheck 是否校验
+	 * @param isCheck    是否校验
 	 * @return ObjectStat
 	 */
 	@SneakyThrows
@@ -249,26 +273,26 @@
 											@RequestParam(required = false) String prefixPath,
 											@RequestParam(required = false) Integer isCheck) {
 		// 不传默认不检验
-		if(null!=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,"图片存在敏感内容,请更换其他图片!");
+			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,"");
+		return R.data(200, bladeFile, "");
 	}
 
 	/**
 	 * 自定义前缀上传文件并保存至附件表
 	 *
-	 * @param file 文件
+	 * @param file       文件
 	 * @param prefixPath 文件
-	 * @param isCheck 是否校验
+	 * @param isCheck    是否校验
 	 * @return ObjectStat
 	 */
 	@SneakyThrows
@@ -277,22 +301,22 @@
 												  @RequestParam(required = false) String prefixPath,
 												  @RequestParam(required = false) Integer isCheck) {
 		// 不传默认不检验
-		if(null!=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,"图片存在敏感内容,请更换其他图片!");
+			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);
+		buildAttachData(attachId, fileName, file.getSize(), bladeFile, file);
 		bladeFile.setAttachId(attachId);
 		// 修改link
 		changeLink(bladeFile);
-		return R.data(200,bladeFile,"");
+		return R.data(200, bladeFile, "");
 	}
 
 	/**
@@ -319,13 +343,13 @@
 	/**
 	 * 构建附件数据表
 	 *
-	 * @param attachId 附件id
+	 * @param attachId  附件id
 	 * @param fileName  文件名
 	 * @param fileSize  文件大小
 	 * @param bladeFile 对象存储文件
 	 * @return attachId
 	 */
-	private Long buildAttachData(Long attachId,String fileName, Long fileSize, BladeFile bladeFile,MultipartFile file) {
+	private Long buildAttachData(Long attachId, String fileName, Long fileSize, BladeFile bladeFile, MultipartFile file) {
 		String fileExtension = FileUtil.getFileExtension(fileName);
 		AttachData attach = new AttachData();
 		attach.setAttachId(attachId);

--
Gitblit v1.9.3