lbqms.git - Gitblit

			@@ -14,11 +14,14 @@
			import org.dromara.common.oss.properties.OssProperties;
			import software.amazon.awssdk.auth.credentials.AwsBasicCredentials;
			import software.amazon.awssdk.auth.credentials.StaticCredentialsProvider;
			import software.amazon.awssdk.core.ResponseInputStream;
			import software.amazon.awssdk.core.async.AsyncRequestBody;
			import software.amazon.awssdk.core.async.AsyncResponseTransformer;
			import software.amazon.awssdk.core.async.BlockingInputStreamAsyncRequestBody;
			import software.amazon.awssdk.regions.Region;
			import software.amazon.awssdk.services.s3.S3AsyncClient;
			import software.amazon.awssdk.services.s3.S3Configuration;
			import software.amazon.awssdk.services.s3.model.GetObjectResponse;
			import software.amazon.awssdk.services.s3.model.NoSuchBucketException;
			import software.amazon.awssdk.services.s3.model.S3Exception;
			import software.amazon.awssdk.services.s3.presigner.S3Presigner;
			@@ -26,10 +29,7 @@
			import software.amazon.awssdk.transfer.s3.model.*;
			import software.amazon.awssdk.transfer.s3.progress.LoggingTransferListener;

			import java.io.ByteArrayInputStream;
			import java.io.File;
			import java.io.IOException;
			import java.io.InputStream;
			import java.io.*;
			import java.net.URI;
			import java.net.URL;
			import java.nio.file.Files;
			@@ -84,7 +84,7 @@
			AwsBasicCredentials.create(properties.getAccessKey(), properties.getSecretKey()));

			//MinIO 使用 HTTPS 限制使用域名访问，站点填域名。需要启用路径样式访问
			boolean pathStyle = !StringUtils.containsAny(properties.getEndpoint(), OssConstant.CLOUD_SERVICE);
			boolean isStyle = !StringUtils.containsAny(properties.getEndpoint(), OssConstant.CLOUD_SERVICE);

			//创建AWS基于 CRT 的 S3 客户端
			this.client = S3AsyncClient.crtBuilder()
			@@ -94,15 +94,15 @@
			.targetThroughputInGbps(20.0)
			.minimumPartSizeInBytes(10 * 1025 * 1024L)
			.checksumValidationEnabled(false)
			.forcePathStyle(pathStyle)
			.forcePathStyle(isStyle)
			.build();

			//AWS基于 CRT 的 S3 AsyncClient 实例用作 S3 传输管理器的底层客户端
			this.transferManager = S3TransferManager.builder().s3Client(this.client).build();

			// 检查是否连接到 MinIO
			// 创建 S3 配置对象
			S3Configuration config = S3Configuration.builder().chunkedEncodingEnabled(false)
			.pathStyleAccessEnabled(pathStyle).build();
			.pathStyleAccessEnabled(isStyle).build();

			// 创建预签名 URL 的生成器实例，用于生成 S3 预签名 URL
			this.presigner = S3Presigner.builder()
			@@ -162,13 +162,14 @@
			/**
			* 上传文件到 Amazon S3，并返回上传结果
			*
			* @param filePath 本地文件路径
			* @param key 在 Amazon S3 中的对象键
			* @param md5Digest 本地文件的 MD5 哈希值（可选）
			* @param filePath 本地文件路径
			* @param key 在 Amazon S3 中的对象键
			* @param md5Digest 本地文件的 MD5 哈希值（可选）
			* @param contentType 文件内容类型
			* @return UploadResult 包含上传后的文件信息
			* @throws OssException 如果上传失败，抛出自定义异常
			*/
			public UploadResult upload(Path filePath, String key, String md5Digest) {
			public UploadResult upload(Path filePath, String key, String md5Digest, String contentType) {
			try {
			// 构建上传请求对象
			FileUpload fileUpload = transferManager.uploadFile(
			@@ -176,6 +177,7 @@
			y -> y.bucket(properties.getBucketName())
			.key(key)
			.contentMD5(StringUtils.isNotEmpty(md5Digest) ? md5Digest : null)
			.contentType(contentType)
			.build())
			.addTransferListener(LoggingTransferListener.create())
			.source(filePath).build());
			@@ -201,10 +203,11 @@
			* @param inputStream 要上传的输入流
			* @param key 在 Amazon S3 中的对象键
			* @param length 输入流的长度
			* @param contentType 文件内容类型
			* @return UploadResult 包含上传后的文件信息
			* @throws OssException 如果上传失败，抛出自定义异常
			*/
			public UploadResult upload(InputStream inputStream, String key, Long length) {
			public UploadResult upload(InputStream inputStream, String key, Long length, String contentType) {
			// 如果输入流不是 ByteArrayInputStream，则将其读取为字节数组再创建 ByteArrayInputStream
			if (!(inputStream instanceof ByteArrayInputStream)) {
			inputStream = new ByteArrayInputStream(IoUtil.readBytes(inputStream));
			@@ -219,6 +222,8 @@
			.putObjectRequest(
			y -> y.bucket(properties.getBucketName())
			.key(key)
			.contentType(contentType)
			.acl(getAccessPolicy().getObjectCannedACL())
			.build())
			.build());

			@@ -258,6 +263,37 @@
			// 等待文件下载操作完成
			downloadFile.completionFuture().join();
			return tempFilePath;
			}

			/**
			* 下载文件从 Amazon S3 到输出流
			*
			* @param key 文件在 Amazon S3 中的对象键
			* @param out 输出流
			* @return 输出流中写入的字节数（长度）
			* @throws OssException 如果下载失败，抛出自定义异常
			*/
			public long download(String key, OutputStream out) {
			try {
			// 构建下载请求
			DownloadRequest<ResponseInputStream<GetObjectResponse>> downloadRequest = DownloadRequest.builder()
			// 文件对象
			.getObjectRequest(y -> y.bucket(properties.getBucketName())
			.key(key)
			.build())
			.addTransferListener(LoggingTransferListener.create())
			// 使用订阅转换器
			.responseTransformer(AsyncResponseTransformer.toBlockingInputStream())
			.build();
			// 使用 S3TransferManager 下载文件
			Download<ResponseInputStream<GetObjectResponse>> responseFuture = transferManager.download(downloadRequest);
			// 输出到流中
			try (ResponseInputStream<GetObjectResponse> responseStream = responseFuture.completionFuture().join().result()) { // auto-closeable stream
			return responseStream.transferTo(out); // 阻塞调用线程 blocks the calling thread
			}
			} catch (Exception e) {
			throw new OssException("文件下载失败，错误信息:[" + e.getMessage() + "]");
			}
			}

			/**
			@@ -304,7 +340,7 @@
			* @throws OssException 如果上传失败，抛出自定义异常
			*/
			public UploadResult uploadSuffix(byte[] data, String suffix) {
			return upload(new ByteArrayInputStream(data), getPath(properties.getPrefix(), suffix), Long.valueOf(data.length));
			return upload(new ByteArrayInputStream(data), getPath(properties.getPrefix(), suffix), Long.valueOf(data.length), FileUtils.getMimeType(suffix));
			}

			/**
			@@ -317,7 +353,7 @@
			* @throws OssException 如果上传失败，抛出自定义异常
			*/
			public UploadResult uploadSuffix(InputStream inputStream, String suffix, Long length) {
			return upload(inputStream, getPath(properties.getPrefix(), suffix), length);
			return upload(inputStream, getPath(properties.getPrefix(), suffix), length, FileUtils.getMimeType(suffix));
			}

			/**
			@@ -329,7 +365,7 @@
			* @throws OssException 如果上传失败，抛出自定义异常
			*/
			public UploadResult uploadSuffix(File file, String suffix) {
			return upload(file.toPath(), getPath(properties.getPrefix(), suffix), null);
			return upload(file.toPath(), getPath(properties.getPrefix(), suffix), null, FileUtils.getMimeType(suffix));
			}

			/**