opensource/RuoYi-Vue-Plus
2
0
Fork 1
mirror of https://gitee.com/dromara/RuoYi-Vue-Plus.git synced 2026-09-20 18:15:56 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add oss上传file后还需处理不自动删除

Browse Source
This commit is contained in:
stone 2024-02-22 21:56:01 +08:00
parent 3b932cfa7b
commit 33d2404248
3 changed files with 30 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
ruoyi-common/ruoyi-common-oss/src/main/java/org/dromara/common/oss/core/OssClient.java
View File

@ -162,10 +162,11 @@ public class OssClient {
* @param filePath 本地文件路径
* @param key 在 Amazon S3 中的对象键
* @param md5Digest 本地文件的 MD5 哈希值（可选）
* @param deleteAfterUpload 是否自动删除临时文件
* @return UploadResult 包含上传后的文件信息
* @throws OssException 如果上传失败，抛出自定义异常
*/
public UploadResult upload(Path filePath, String key, String md5Digest) {
public UploadResult upload(Path filePath, String key, String md5Digest, Boolean deleteAfterUpload) {
try {
// 构建上传请求对象
FileUpload fileUpload = transferManager.uploadFile(
@ -187,8 +188,9 @@ public class OssClient {
// 捕获异常并抛出自定义异常
throw new OssException("上传文件失败，请检查配置信息:[" + e.getMessage() + "]");
} finally {
// 无论上传是否成功，最终都会删除临时文件
FileUtils.del(filePath);
if (deleteAfterUpload){
FileUtils.del(filePath);
}
}
}
@ -322,11 +324,12 @@ public class OssClient {
*
* @param file 要上传的文件
* @param suffix 对象键的后缀
* @param deleteAfterUpload 是否自动删除临时文件
* @return UploadResult 包含上传后的文件信息
* @throws OssException 如果上传失败，抛出自定义异常
*/
public UploadResult uploadSuffix(File file, String suffix) {
return upload(file.toPath(), getPath(properties.getPrefix(), suffix), null);
public UploadResult uploadSuffix(File file, String suffix, Boolean deleteAfterUpload) {
return upload(file.toPath(), getPath(properties.getPrefix(), suffix), null, deleteAfterUpload);
}
/**

9
ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/service/ISysOssService.java
View File

@ -60,6 +60,15 @@ public interface ISysOssService {
*/
SysOssVo upload(File file);
/**
* 上传文件到对象存储服务，并保存文件信息到数据库
*
* @param file 要上传的文件对象
* @param deleteAfterUpload 是否自动删除临时文件
* @return 上传成功后的 SysOssVo 对象，包含文件信息
*/
SysOssVo upload(File file, Boolean deleteAfterUpload);
/**
* 文件下载方法，支持一次性下载完整文件
*

14
ruoyi-modules/ruoyi-system/src/main/java/org/dromara/system/service/impl/SysOssServiceImpl.java
View File

@ -196,10 +196,22 @@ public class SysOssServiceImpl implements ISysOssService, OssService {
*/
@Override
public SysOssVo upload(File file) {
return this.upload(file, true);
}
/**
* 上传文件到对象存储服务，并保存文件信息到数据库
*
* @param file 要上传的文件对象
* @param deleteAfterUpload 是否自动删除临时文件
* @return 上传成功后的 SysOssVo 对象，包含文件信息
*/
@Override
public SysOssVo upload(File file, Boolean deleteAfterUpload) {
String originalfileName = file.getName();
String suffix = StringUtils.substring(originalfileName, originalfileName.lastIndexOf("."), originalfileName.length());
OssClient storage = OssFactory.instance();
UploadResult uploadResult = storage.uploadSuffix(file, suffix);
UploadResult uploadResult = storage.uploadSuffix(file, suffix, deleteAfterUpload);
// 保存文件信息
return buildResultEntity(originalfileName, suffix, storage.getConfigKey(), uploadResult);
}