From ae935b68e75e7c90e27aecedefa971106927c38d Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Mon, 08 Jan 2024 20:49:46 +0800
Subject: [PATCH] 上传接口link地址修改

---
 src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java |   18 ++++++++++++++++++
 1 files changed, 18 insertions(+), 0 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 f1b57b4..473107a 100644
--- a/src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java
+++ b/src/main/java/org/springblade/modules/resource/endpoint/OssEndpoint.java
@@ -146,7 +146,22 @@
 	@PostMapping("/put-file")
 	public R<BladeFile> putFile(@RequestParam MultipartFile file) {
 		BladeFile bladeFile = ossBuilder.template().putFile(file.getOriginalFilename(), file.getInputStream());
+		// 修改link
+		changeLink(bladeFile);
+		// 返回
 		return R.data(bladeFile);
+	}
+
+	/**
+	 * 修改link
+	 * @param bladeFile
+	 */
+	private void changeLink(BladeFile bladeFile) {
+		if (null!=bladeFile){
+			// 替换url link 前缀
+			String newLink = ossBuilder.ossProperties().getEndpoint() + bladeFile.getName();
+			bladeFile.setLink(newLink);
+		}
 	}
 
 	/**
@@ -176,6 +191,9 @@
 		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);
 	}
 

--
Gitblit v1.9.3