software.amazon.nio.spi.s3.S3TransferUtil Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of aws-java-nio-spi-for-s3 Show documentation

A Java NIO.2 service provider for S3, allowing Java NIO operations to be performed on paths using the `s3` scheme. This package implements the service provider interface (SPI) defined for Java NIO.2 in JDK 1.7 providing "plug-in" non-blocking access to S3 objects for Java applications using Java NIO.2 for file access.

The newest version!

/*
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 * SPDX-License-Identifier: Apache-2.0
 */

package software.amazon.nio.spi.s3;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
import software.amazon.awssdk.services.s3.S3AsyncClient;
import software.amazon.awssdk.services.s3.model.GetObjectRequest;
import software.amazon.awssdk.services.s3.model.PutObjectRequest;
import software.amazon.awssdk.transfer.s3.S3TransferManager;
import software.amazon.awssdk.transfer.s3.model.DownloadFileRequest;
import software.amazon.awssdk.transfer.s3.model.UploadFileRequest;

final class S3TransferUtil {
    private final S3AsyncClient client;
    private final Long timeout;
    private final TimeUnit timeUnit;

    S3TransferUtil(S3AsyncClient client, Long timeout, TimeUnit timeUnit) {
        this.client = client;
        this.timeout = timeout;
        this.timeUnit = timeUnit;
    }

    void downloadToLocalFile(S3Path path, Path destination) throws InterruptedException, ExecutionException, TimeoutException {
        try (var s3TransferManager = S3TransferManager.builder().s3Client(client).build()) {
            var downloadCompletableFuture = s3TransferManager.downloadFile(
                DownloadFileRequest.builder()
                    .getObjectRequest(GetObjectRequest.builder()
                        .bucket(path.bucketName())
                        .key(path.getKey())
                        .build())
                    .destination(destination)
                    .build()
            ).completionFuture();

            if (timeout != null && timeUnit != null) {
                downloadCompletableFuture.get(timeout, timeUnit);
            } else {
                downloadCompletableFuture.join();
            }
        }
    }

    void uploadLocalFile(S3Path path, Path localFile) throws IOException {
        try (var s3TransferManager = S3TransferManager.builder().s3Client(client).build()) {
            var uploadCompletableFuture = s3TransferManager.uploadFile(
                UploadFileRequest.builder()
                    .putObjectRequest(PutObjectRequest.builder()
                        .bucket(path.bucketName())
                        .key(path.getKey())
                        .contentType(Files.probeContentType(localFile))
                        .build())
                    .source(localFile)
                    .build()
            ).completionFuture();

            if (timeout != null && timeUnit != null) {
                uploadCompletableFuture.get(timeout, timeUnit);
            } else {
                uploadCompletableFuture.join();
            }
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
            throw new IOException("Could not write to path:" + path, e);
        } catch (TimeoutException | ExecutionException e) {
            throw new IOException("Could not write to path:" + path, e);
        }
    }
}