【新增】OSS文件管理工具

This commit is contained in:
2025-11-18 18:18:06 +08:00
parent 9dc9bff8b1
commit 5acb3a34e0

View File

@@ -2,21 +2,31 @@ package com.corewing.app.util;
import com.aliyun.oss.OSS; import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder; import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.model.ObjectMetadata; import com.aliyun.oss.OSSException;
import com.aliyun.oss.model.PutObjectRequest; import com.aliyun.oss.model.*;
import com.aliyun.oss.model.PutObjectResult;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import javax.annotation.PostConstruct; import javax.annotation.PostConstruct;
import java.io.File; import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream; import java.io.InputStream;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
@Component @Component
public class OSSUploadUtil { public class OSSUploadUtil {
/**
* 用户登录名称 oss@1563058769875871.onaliyun.com
* AccessKey ID LTAI5tKUhXrGxZ5Gj3exWpkG
* AccessKey Secret PaBTMp3BhcOUgLQhJWmOkhfJhTlzhV
*/
private static final String ENDPOINT = "oss-cn-shenzhen.aliyuncs.com"; // 地域节点 private static final String ENDPOINT = "oss-cn-shenzhen.aliyuncs.com"; // 地域节点
private static final String ACCESS_KEY_ID = "your-access-key-id"; // 替换为你的 AccessKey ID private static final String ACCESS_KEY_ID = "LTAI5tKUhXrGxZ5Gj3exWpkG"; // 替换为你的 AccessKey ID
private static final String ACCESS_KEY_SECRET = "your-access-key-secret"; // 替换为你的 AccessKey Secret private static final String ACCESS_KEY_SECRET = "PaBTMp3BhcOUgLQhJWmOkhfJhTlzhV"; // 替换为你的 AccessKey Secret
private static final String BUCKET_NAME = "corewing-app"; // 替换为你的 Bucket 名称 private static final String BUCKET_NAME = "corewing-app"; // 替换为你的 Bucket 名称
/** /**
@@ -67,6 +77,137 @@ public class OSSUploadUtil {
} }
} }
/**
* 从 OSS 下载文件到本地路径
* @param objectName OSS 中存储的文件路径(如 "firmware/20231104/update.bin"
* @param localFilePath 本地保存路径(如 "D:/downloads/update.bin"
* @return 下载是否成功
*/
public static boolean downloadFileToLocal(String objectName, String localFilePath) {
OSS ossClient = new OSSClientBuilder().build(ENDPOINT, ACCESS_KEY_ID, ACCESS_KEY_SECRET);
try {
// 检查文件是否存在
if (!ossClient.doesObjectExist(BUCKET_NAME, objectName)) {
throw new FileNotFoundException("OSS 文件不存在:" + objectName);
}
// 下载文件到本地
ossClient.getObject(new GetObjectRequest(BUCKET_NAME, objectName), new File(localFilePath));
return true;
} catch (OSSException | FileNotFoundException e) {
return false;
} finally {
if (ossClient != null) {
ossClient.shutdown();
}
}
}
/**
* 从 OSS 获取文件输入流(需手动关闭流)
* @param objectName OSS 中存储的文件路径
* @return 文件输入流(使用后需调用 close() 关闭)
* @throws IOException 当文件不存在或下载失败时抛出
*/
public static InputStream downloadFileToStream(String objectName) throws IOException {
OSS ossClient = new OSSClientBuilder().build(ENDPOINT, ACCESS_KEY_ID, ACCESS_KEY_SECRET);
try {
// 检查文件是否存在
if (!ossClient.doesObjectExist(BUCKET_NAME, objectName)) {
throw new FileNotFoundException("OSS 文件不存在:" + objectName);
}
// 获取 OSS 文件对象并返回输入流
OSSObject ossObject = ossClient.getObject(BUCKET_NAME, objectName);
return ossObject.getObjectContent();
} catch (OSSException e) {
throw new IOException("OSS 服务异常:" + e.getMessage(), e);
} finally {
// 注意:此处不能关闭 ossClient否则输入流会被提前关闭
// 需在调用方使用完输入流后,手动关闭 ossClient 和输入流
// (或通过 try-with-resources 自动关闭)
}
}
/**
* 从 OSS 删除单个文件
* @param objectName OSS 中存储的文件路径(如 "firmware/20231104/update.bin"
* @return 删除是否成功(文件不存在时返回 false
*/
public static boolean deleteFile(String objectName) {
OSS ossClient = new OSSClientBuilder().build(ENDPOINT, ACCESS_KEY_ID, ACCESS_KEY_SECRET);
try {
// 先检查文件是否存在
if (!ossClient.doesObjectExist(BUCKET_NAME, objectName)) {
System.out.println("OSS 文件不存在,删除失败:" + objectName);
return false;
}
// 执行删除操作
ossClient.deleteObject(BUCKET_NAME, objectName);
System.out.println("OSS 文件删除成功:" + objectName);
return true;
} catch (OSSException e) {
System.err.println("OSS 服务异常,删除失败:" + e.getMessage());
return false;
} finally {
if (ossClient != null) {
ossClient.shutdown();
}
}
}
/**
* 批量删除 OSS 文件3.17.4 版本终极兼容方案)
* 逻辑:先执行批量删除,再逐个校验文件是否存在,确认删除结果
* @param objectNames OSS 文件路径列表(如 ["firmware/update1.bin", "firmware/update2.bin"]
* @return 批量删除结果key文件路径value是否删除成功
*/
public static Map<String, Boolean> batchDeleteFiles(List<String> objectNames) {
Map<String, Boolean> deleteResult = new HashMap<>();
if (objectNames == null || objectNames.isEmpty()) {
System.out.println("批量删除失败:文件列表为空");
return deleteResult;
}
OSS ossClient = new OSSClientBuilder().build(ENDPOINT, ACCESS_KEY_ID, ACCESS_KEY_SECRET);
try {
// 第一步:执行批量删除(忽略 SDK 返回结果,只保证执行删除操作)
DeleteObjectsRequest request = new DeleteObjectsRequest(BUCKET_NAME);
request.setKeys(objectNames); // 直接传入文件路径列表
ossClient.deleteObjects(request); // 执行删除,不依赖返回结果
// 第二步:逐个校验文件是否存在,确认删除结果
for (String objectName : objectNames) {
boolean isDeleted = !ossClient.doesObjectExist(BUCKET_NAME, objectName);
deleteResult.put(objectName, isDeleted);
if (isDeleted) {
System.out.println("删除成功:" + objectName);
} else {
System.err.println("删除失败:文件仍存在 -> " + objectName);
}
}
} catch (OSSException e) {
System.err.printf("OSS 服务异常:错误码=%s原因=%s%n",
e.getErrorCode(), e.getErrorMessage());
// 服务异常时,所有文件标记为删除失败
objectNames.forEach(name -> deleteResult.put(name, false));
} catch (Exception e) {
System.err.println("批量删除未知异常:" + e.getMessage());
objectNames.forEach(name -> deleteResult.put(name, false));
} finally {
if (ossClient != null) {
ossClient.shutdown();
}
}
return deleteResult;
}
/** /**
* 根据文件名获取 Content-Type * 根据文件名获取 Content-Type
* @param fileName 文件名(如 "update.bin" * @param fileName 文件名(如 "update.bin"