com.crabshue.commons.aws.s3.AwsS3ClientBuilder Maven / Gradle / Ivy

Go to download
package com.crabshue.commons.aws.s3;

import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.Validate;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.amazonaws.auth.AWSStaticCredentialsProvider;
import com.amazonaws.auth.BasicAWSCredentials;
import com.amazonaws.services.s3.AmazonS3;
import com.amazonaws.services.s3.AmazonS3ClientBuilder;
import com.crabshue.commons.aws.AwsCredentials;

/**
 * {@link AwsS3Client} builder.
 *
 */
public class AwsS3ClientBuilder {


    private static Logger logger = LoggerFactory.getLogger(AwsS3ClientBuilder.class);

    private AwsS3Bucket awsS3Bucket;

    private AwsCredentials awsCredentials;

    /**
     * Prepare an {@link AwsS3Client} with a given {@link AwsS3Bucket}.
     *
     * @param awsS3Bucket the AWS S3 bucket
     * @return new instance.
     */
    public static AwsS3ClientBuilder of(final AwsS3Bucket awsS3Bucket) {
        Validate.notNull(awsS3Bucket);

        final AwsS3ClientBuilder ret = new AwsS3ClientBuilder();
        ret.awsS3Bucket = awsS3Bucket;
        return ret;
    }

    /**
     * Set credentials for {@link AwsS3Client};
     *
     * @param awsCredentials the credentials.
     * @return instance.
     */
    public AwsS3ClientBuilder withCredentials(final AwsCredentials awsCredentials) {
        Validate.notNull(awsCredentials);

        this.awsCredentials = awsCredentials;
        return this;
    }

    /**
     * Build the {@link AwsS3Client}.
     *
     * @return the client.
     */
    public AwsS3Client build() {

        final AmazonS3ClientBuilder amazonS3Builder = AmazonS3ClientBuilder.standard();

        if (this.awsCredentials != null && StringUtils.isNotBlank(awsCredentials.getAccessKeyId()) && StringUtils.isNotBlank(awsCredentials.getSecretKeyId())) {
            final BasicAWSCredentials awsCreds = new BasicAWSCredentials(awsCredentials.getAccessKeyId(), awsCredentials.getSecretKeyId());

            amazonS3Builder.withCredentials(new AWSStaticCredentialsProvider(awsCreds));
        }

        amazonS3Builder.withRegion(awsS3Bucket.getRegion());

        final AmazonS3 amazonS3 = amazonS3Builder.build();

        logger.debug("Created AWS S3 client [{}]", amazonS3);

        return new AwsS3ClientImpl(amazonS3, this.awsS3Bucket);
    }
}