From ac79f760b21529694f4257d508c989beb7a31194 Mon Sep 17 00:00:00 2001
From: mitao <2763622819@qq.com>
Date: 星期四, 03 四月 2025 21:36:19 +0800
Subject: [PATCH] bug修改
---
ruoyi-admin/src/main/java/com/ruoyi/web/controller/tool/TencentCosUtil.java | 122 +++++++++++++++++++++++++++-------------
1 files changed, 82 insertions(+), 40 deletions(-)
diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/tool/TencentCosUtil.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/tool/TencentCosUtil.java
index d74cf21..9a9ac21 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/tool/TencentCosUtil.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/tool/TencentCosUtil.java
@@ -27,6 +27,11 @@
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
+import java.net.URL;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.nio.file.StandardCopyOption;
import java.util.Base64;
import java.util.Date;
import java.util.UUID;
@@ -60,7 +65,7 @@
/**
* 上传文件，并存入sys_file,返回文件的主键ID
* @param file
- * @param folder
+ * @param folder 格式：/xxxxx,/xxxx/xxxx ，最后不用加斜杠
* @return
*/
public TFile upload(MultipartFile file, String folder){
@@ -75,12 +80,11 @@
String fileType = originalFilename.substring(originalFilename.lastIndexOf("."));
//使用UUID工具 创建唯一名称，放置文件重名被覆盖，在拼接上上命令获取的文件类型
String fileName = UUID.randomUUID() + fileType;
-
- String name = (StringUtils.isNotEmpty(folder)?
+ String filePath = (StringUtils.isNotEmpty(folder)?
folder+"/"
:"/default/") + DateUtil.format(new Date(),NORM_DATE_FORMAT)+"/"+fileName;
// 指定文件上传到 COS 上的路径，即对象键。最终文件会传到存储桶名字中的images文件夹下的fileName名字
- String key = cosConfig.getLocation()+"/" + fileName;
+ filePath = cosConfig.getLocation() + filePath;
// 创建上传Object的Metadata
ObjectMetadata objectMetadata = new ObjectMetadata();
// - 使用输入流存储，需要设置请求长度
@@ -90,12 +94,12 @@
// - 设置Content-Type
objectMetadata.setContentType(fileType);
//上传文件
- cosClient.putObject(cosConfig.getBucketName(), key, inputStream, objectMetadata);
+ cosClient.putObject(cosConfig.getBucketName(), filePath, inputStream, objectMetadata);
TFile tFile = new TFile();
- tFile.setFileName(name);
+ tFile.setFileName(filePath);
tFile.setRealName(originalFilename);
tFile.setFileType(fileType);
- tFile.setUrl(cosConfig.getRootSrc()+key);
+ tFile.setUrl(cosConfig.getRootSrc()+filePath);
tFile.setContentType(file.getContentType());
tFile.setFileSize(file.getSize());
sysFileService.save(tFile);
@@ -113,7 +117,7 @@
* @param file
* @return 返回文件的浏览全路径
*/
- public String upLoadFile(MultipartFile file) {
+ public String upLoadFile(MultipartFile file,String folder) {
try {
// 获取上传的文件的输入流
InputStream inputStream = file.getInputStream();
@@ -124,7 +128,11 @@
//使用UUID工具 创建唯一名称，放置文件重名被覆盖，在拼接上上命令获取的文件类型
String fileName = UUID.randomUUID() + fileType;
// 指定文件上传到 COS 上的路径，即对象键。最终文件会传到存储桶名字中的images文件夹下的fileName名字
- String key = cosConfig.getLocation()+"/" + fileName;
+ String filePath = (StringUtils.isNotEmpty(folder)?
+ folder+"/"
+ :"/default/") + DateUtil.format(new Date(),NORM_DATE_FORMAT)+"/"+fileName;
+
+ filePath = cosConfig.getLocation()+"/" + filePath;
// 创建上传Object的Metadata
ObjectMetadata objectMetadata = new ObjectMetadata();
// - 使用输入流存储，需要设置请求长度
@@ -134,9 +142,17 @@
// - 设置Content-Type
objectMetadata.setContentType(fileType);
//上传文件
- PutObjectResult putResult = cosClient.putObject(cosConfig.getBucketName(), key, inputStream, objectMetadata);
+ PutObjectResult putResult = cosClient.putObject(cosConfig.getBucketName(), filePath, inputStream, objectMetadata);
+ TFile tFile = new TFile();
+ tFile.setFileName(filePath);
+ tFile.setRealName(originalFilename);
+ tFile.setFileType(fileType);
+ tFile.setUrl(cosConfig.getRootSrc()+filePath);
+ tFile.setContentType(file.getContentType());
+ tFile.setFileSize(file.getSize());
+ sysFileService.save(tFile);
// 创建文件的网络访问路径
- String url = cosConfig.getRootSrc() + key;
+ String url = cosConfig.getRootSrc() + filePath;
return url;
} catch (Exception e) {
e.printStackTrace();
@@ -155,15 +171,6 @@
HttpServletResponse response = WebUtils.response();
String replace = file.replace(cosConfig.getRootSrc(), "");
response.setHeader("Access-Control-Expose-Headers","File-Type");
- COSCredentials cred = new BasicCOSCredentials(
- cosConfig.getSecretId(),
- cosConfig.getSecretKey());
- // 2.1 设置存储桶的地域（上文获得）
- Region region = new Region(cosConfig.getBucketAddr());
- ClientConfig clientConfig = new ClientConfig(region);
- // 2.2 使用https协议传输
- clientConfig.setHttpProtocol(HttpProtocol.https);
- COSClient cosClient = new COSClient(cred, clientConfig);
try {
// 5. 下载文件并获取输入流
InputStream inputStream = cosClient.getObject(cosConfig.getBucketName(), replace).getObjectContent();
@@ -184,15 +191,6 @@
public String downLoadFileImg(String file) {
byte[] data = null;
String replace = file.replace(cosConfig.getRootSrc(), "");
- COSCredentials cred = new BasicCOSCredentials(
- cosConfig.getSecretId(),
- cosConfig.getSecretKey());
- // 2.1 设置存储桶的地域（上文获得）
- Region region = new Region(cosConfig.getBucketAddr());
- ClientConfig clientConfig = new ClientConfig(region);
- // 2.2 使用https协议传输
- clientConfig.setHttpProtocol(HttpProtocol.https);
- COSClient cosClient = new COSClient(cred, clientConfig);
try {
// 5. 下载文件并获取输入流
InputStream inputStream = cosClient.getObject(cosConfig.getBucketName(), replace).getObjectContent();
@@ -213,22 +211,39 @@
return Base64.getEncoder().encodeToString(data);
}
- public void download(TFile file) {
+ public void download(String fileUrl) {
HttpServletResponse response = WebUtils.response();
- COSCredentials cred = new BasicCOSCredentials(
- cosConfig.getSecretId(),
- cosConfig.getSecretKey());
- // 2.1 设置存储桶的地域（上文获得）
- Region region = new Region(cosConfig.getBucketAddr());
- ClientConfig clientConfig = new ClientConfig(region);
- // 2.2 使用https协议传输
- clientConfig.setHttpProtocol(HttpProtocol.https);
- COSClient cosClient = new COSClient(cred, clientConfig);
+ fileUrl = fileUrl.replace(cosConfig.getRootSrc(), "");
+
// 5. 下载文件并获取输入流
- COSObject object = cosClient.getObject(cosConfig.getBucketName(), file.getUrl());
+ COSObject object = cosClient.getObject(cosConfig.getBucketName(),fileUrl);
try (
InputStream is = object.getObjectContent();
OutputStream os = response.getOutputStream()
+ ) {
+ String fileName = fileUrl.substring(fileUrl.lastIndexOf("/"));
+ response.setContentType(object.getObjectMetadata().getContentType() + ";charset=utf-8");
+ String filename = new String(fileName.getBytes("UTF-8"), "ISO-8859-1");
+ response.addHeader("Content-Disposition", "attachment;filename=" + filename);
+ response.addHeader("Content-Length", "" + object.getObjectMetadata().getContentLength());
+ int len = 0;
+ byte[] buffer = new byte[2048];
+ while ((len = is.read(buffer)) > 0) {
+ os.write(buffer, 0, len);
+ }
+ os.flush();
+ } catch (IOException e) {
+ log.error("读取cos图片发生异常", e);
+ }
+ }
+
+ public void download(TFile file) {
+ HttpServletResponse response = WebUtils.response();
+ // 5. 下载文件并获取输入流
+ COSObject object = cosClient.getObject(cosConfig.getBucketName(), file.getFileName());
+ try (
+ InputStream is = object.getObjectContent();
+ OutputStream os = response.getOutputStream()
) {
response.setContentType(file.getContentType() + ";charset=utf-8");
String filename = new String(file.getRealName().getBytes("UTF-8"), "ISO-8859-1");
@@ -244,4 +259,31 @@
log.error("读取cos图片发生异常", e);
}
}
+
+
+ /**
+ * 将文件下载到指定目录
+ * @param fileUrl
+ * @param saveDir
+ * @param fileName
+ * @throws IOException
+ */
+ public void download(String fileUrl, String saveDir, String fileName){
+ fileUrl = fileUrl.replace(cosConfig.getRootSrc(), "");
+ // 下载文件并获取输入流
+ COSObject object = cosClient.getObject(cosConfig.getBucketName(),fileUrl);
+ try (
+ InputStream in = object.getObjectContent();
+ ){
+ Path targetPath = Paths.get(saveDir, fileName);
+ // 确保目录存在
+ Files.createDirectories(targetPath.getParent());
+ // 将文件保存到目标路径
+ Files.copy(in, targetPath, StandardCopyOption.REPLACE_EXISTING);
+ }catch (IOException e){
+ log.error("读取cos图片发生异常", e);
+ }
+ }
+
+
}
\ No newline at end of file
--
Gitblit v1.7.1