并行流对于多文件上传到AWS S3无法工作的原因是并行流在处理I/O操作时可能会出现并发冲突。
解决这个问题的一个方法是使用Java的CompletableFuture来处理并行上传。下面是一个使用CompletableFuture的示例代码:
import java.io.File;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutionException;
import java.util.stream.Collectors;
import software.amazon.awssdk.core.sync.RequestBody;
import software.amazon.awssdk.services.s3.S3Client;
import software.amazon.awssdk.services.s3.model.PutObjectRequest;
import software.amazon.awssdk.services.s3.model.PutObjectResponse;
public class S3Uploader {
private static final String BUCKET_NAME = "your-bucket-name";
private static final S3Client S3_CLIENT = S3Client.create();
public static void main(String[] args) throws ExecutionException, InterruptedException {
List filesToUpload = getFilesToUpload();
List> uploadFutures = filesToUpload.stream()
.map(file -> CompletableFuture.supplyAsync(() -> uploadFile(file)))
.collect(Collectors.toList());
CompletableFuture allFutures = CompletableFuture
.allOf(uploadFutures.toArray(new CompletableFuture[uploadFutures.size()]));
CompletableFuture> allResponsesFuture = allFutures.thenApply(v -> uploadFutures.stream()
.map(CompletableFuture::join)
.collect(Collectors.toList()));
List allResponses = allResponsesFuture.get();
// 处理上传响应
S3_CLIENT.close();
}
private static PutObjectResponse uploadFile(File file) {
PutObjectRequest request = PutObjectRequest.builder()
.bucket(BUCKET_NAME)
.key(file.getName())
.build();
RequestBody requestBody = RequestBody.fromFile(file);
return S3_CLIENT.putObject(request, requestBody);
}
private static List getFilesToUpload() {
// 返回需要上传的文件列表
List files = new ArrayList<>();
files.add(new File("path/to/file1.txt"));
files.add(new File("path/to/file2.txt"));
files.add(new File("path/to/file3.txt"));
return files;
}
}
上述代码使用了CompletableFuture来并行上传多个文件到AWS S3。首先,通过getFilesToUpload()
方法获取要上传的文件列表。然后,使用CompletableFuture.supplyAsync()
方法将每个文件的上传操作包装成一个CompletableFuture。接下来,使用CompletableFuture.allOf()
方法将所有的CompletableFuture合并成一个新的CompletableFuture。最后,通过CompletableFuture.thenApply()
方法等待所有上传操作完成,并收集所有的上传响应结果。
请注意,上述代码仅提供了一个示例,并没有处理异常、重试、错误处理等情况,你可以根据实际需求进行相应的修改和扩展。
下一篇:并行流函数需要更长的时间。