Added wrapper class for CRT retry config

Author: joviegas

services/s3/src/main/java/software/amazon/awssdk/services/s3/S3CrtAsyncClientBuilder.java

@@ -22,6 +22,7 @@
 import software.amazon.awssdk.auth.credentials.AwsCredentialsProvider;
 import software.amazon.awssdk.regions.Region;
 import software.amazon.awssdk.services.s3.crt.S3CrtHttpConfiguration;
+import software.amazon.awssdk.services.s3.crt.S3CrtRetryConfiguration;
 import software.amazon.awssdk.services.s3.model.GetObjectRequest;
 import software.amazon.awssdk.services.s3.model.PutObjectRequest;
 import software.amazon.awssdk.utils.Validate;
@@ -161,6 +162,14 @@ public interface S3CrtAsyncClientBuilder extends SdkBuilder<S3CrtAsyncClientBuil
      */
     S3CrtAsyncClientBuilder httpConfiguration(S3CrtHttpConfiguration configuration);
 
+    /**
+     * Sets the Retry configuration to use for this client.
+     *
+     * @param retryConfiguration The retry configurations to be used.
+     * @return The builder of the method chaining.
+     */
+    S3CrtAsyncClientBuilder retryConfiguration(S3CrtRetryConfiguration retryConfiguration);
+
     /**
      * A convenience method that creates an instance of the {@link S3CrtHttpConfiguration} builder, avoiding the
      * need to create one manually via {@link S3CrtHttpConfiguration#builder()}.
@@ -177,6 +186,25 @@ default S3CrtAsyncClientBuilder httpConfiguration(Consumer<S3CrtHttpConfiguratio
     }
 
 
+    /**
+     * A convenience method that creates an instance of the {@link S3CrtRetryConfiguration} builder, avoiding the
+     * need to create one manually via {@link S3CrtRetryConfiguration#builder()}.
+     *
+     * @param retryConfigurationBuilder The retry config builder to use
+     * @return The builder of the method chaining.
+     * @see #retryConfiguration(S3CrtRetryConfiguration)
+     */
+    default S3CrtAsyncClientBuilder retryConfiguration(Consumer<S3CrtRetryConfiguration.Builder> retryConfigurationBuilder) {
+        Validate.paramNotNull(retryConfigurationBuilder, "configurationBuilder");
+        return retryConfiguration(S3CrtRetryConfiguration.builder()
+                                                         .applyMutation(retryConfigurationBuilder)
+                                                         .build());
+    }
+
+
+
+
+
     @Override
     S3AsyncClient build();
 }

services/s3/src/main/java/software/amazon/awssdk/services/s3/crt/S3CrtRetryConfiguration.java

@@ -0,0 +1,112 @@
+/*
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License").
+ * You may not use this file except in compliance with the License.
+ * A copy of the License is located at
+ *
+ *  http://aws.amazon.com/apache2.0
+ *
+ * or in the "license" file accompanying this file. This file is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+ * express or implied. See the License for the specific language governing
+ * permissions and limitations under the License.
+ */
+
+package software.amazon.awssdk.services.s3.crt;
+
+import java.util.Objects;
+import software.amazon.awssdk.annotations.Immutable;
+import software.amazon.awssdk.annotations.SdkPublicApi;
+import software.amazon.awssdk.annotations.ThreadSafe;
+import software.amazon.awssdk.services.s3.S3CrtAsyncClientBuilder;
+import software.amazon.awssdk.utils.Validate;
+import software.amazon.awssdk.utils.builder.CopyableBuilder;
+import software.amazon.awssdk.utils.builder.ToCopyableBuilder;
+
+/**
+ * Retry option configuration for AWS CRT-based S3 client.
+ *
+ * @see S3CrtAsyncClientBuilder#retryConfiguration
+ */
+@SdkPublicApi
+@Immutable
+@ThreadSafe
+public final class S3CrtRetryConfiguration implements ToCopyableBuilder<S3CrtRetryConfiguration.Builder,
+    S3CrtRetryConfiguration> {
+    private final Integer numRetries;
+
+    private S3CrtRetryConfiguration(DefaultBuilder builder) {
+        Validate.isPositiveOrNull(builder.numRetries, "numRetries");
+        this.numRetries = builder.numRetries;
+    }
+
+    /**
+     * Creates a default builder for {@link S3CrtRetryConfiguration}.
+     */
+    public static Builder builder() {
+        return new S3CrtRetryConfiguration.DefaultBuilder();
+    }
+
+    /**
+     * Return the amount of time to wait when initially establishing a connection before giving up and timing out.
+     */
+    public Integer numRetries() {
+        return numRetries;
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) {
+            return true;
+        }
+        if (o == null || getClass() != o.getClass()) {
+            return false;
+        }
+        S3CrtRetryConfiguration that = (S3CrtRetryConfiguration) o;
+        return Objects.equals(numRetries, that.numRetries);
+    }
+
+    @Override
+    public int hashCode() {
+        return numRetries != null ? numRetries.hashCode() : 0;
+    }
+
+    @Override
+    public Builder toBuilder() {
+        return new S3CrtRetryConfiguration.DefaultBuilder(this);
+    }
+
+    public interface Builder extends CopyableBuilder<S3CrtRetryConfiguration.Builder, S3CrtRetryConfiguration> {
+
+        /**
+         * Configure the maximum number of times that a single request should be retried.
+         * @param numRetries
+         * @return The builder of the method chaining.
+         */
+        Builder numRetries(Integer numRetries);
+
+    }
+
+    private static final class DefaultBuilder implements Builder {
+        private Integer numRetries;
+
+        private DefaultBuilder() {
+        }
+
+        private DefaultBuilder(S3CrtRetryConfiguration crtRetryConfiguration) {
+            this.numRetries = crtRetryConfiguration.numRetries;
+        }
+
+        @Override
+        public Builder numRetries(Integer numRetries) {
+            this.numRetries = numRetries;
+            return this;
+        }
+
+        @Override
+        public S3CrtRetryConfiguration build() {
+            return new S3CrtRetryConfiguration(this);
+        }
+    }
+}

services/s3/src/main/java/software/amazon/awssdk/services/s3/internal/crt/DefaultS3CrtAsyncClient.java

@@ -37,13 +37,16 @@
 import software.amazon.awssdk.core.internal.util.ClassLoaderHelper;
 import software.amazon.awssdk.core.retry.RetryPolicy;
 import software.amazon.awssdk.core.signer.NoOpSigner;
+import software.amazon.awssdk.crt.io.ExponentialBackoffRetryOptions;
+import software.amazon.awssdk.crt.io.StandardRetryOptions;
 import software.amazon.awssdk.http.SdkHttpExecutionAttributes;
 import software.amazon.awssdk.regions.Region;
 import software.amazon.awssdk.services.s3.DelegatingS3AsyncClient;
 import software.amazon.awssdk.services.s3.S3AsyncClient;
 import software.amazon.awssdk.services.s3.S3Configuration;
 import software.amazon.awssdk.services.s3.S3CrtAsyncClientBuilder;
 import software.amazon.awssdk.services.s3.crt.S3CrtHttpConfiguration;
+import software.amazon.awssdk.services.s3.crt.S3CrtRetryConfiguration;
 import software.amazon.awssdk.services.s3.model.CopyObjectRequest;
 import software.amazon.awssdk.services.s3.model.CopyObjectResponse;
 import software.amazon.awssdk.services.s3.model.GetObjectRequest;
@@ -96,21 +99,28 @@ private static S3CrtAsyncHttpClient.Builder initializeS3CrtAsyncHttpClient(Defau
         Validate.isPositiveOrNull(builder.targetThroughputInGbps, "targetThroughputInGbps");
         Validate.isPositiveOrNull(builder.minimalPartSizeInBytes, "minimalPartSizeInBytes");
 
-        S3NativeClientConfiguration s3NativeClientConfiguration =
+        S3NativeClientConfiguration.Builder nativeClientBuilder =
             S3NativeClientConfiguration.builder()
                                        .checksumValidationEnabled(builder.checksumValidationEnabled)
                                        .targetThroughputInGbps(builder.targetThroughputInGbps)
                                        .partSizeInBytes(builder.minimalPartSizeInBytes)
                                        .maxConcurrency(builder.maxConcurrency)
-                                       .signingRegion(builder.region == null ? null : builder.region.id())
+                                       .signingRegion(builder.region == null ? null
+                                                                             :
+                                                      builder.region.id())
                                        .endpointOverride(builder.endpointOverride)
                                        .credentialsProvider(builder.credentialsProvider)
                                        .readBufferSizeInBytes(builder.readBufferSizeInBytes)
-                                       .httpConfiguration(builder.httpConfiguration)
-                                       .build();
+                                       .httpConfiguration(builder.httpConfiguration);
 
+        if (builder.retryConfiguration != null) {
+            nativeClientBuilder.standardRetryOptions(
+                new StandardRetryOptions()
+                    .withBackoffRetryOptions(new ExponentialBackoffRetryOptions()
+                                                 .withMaxRetries(builder.retryConfiguration.numRetries())));
+        }
         return S3CrtAsyncHttpClient.builder()
-                                   .s3ClientConfiguration(s3NativeClientConfiguration);
+                                   .s3ClientConfiguration(nativeClientBuilder.build());
     }
 
     public static final class DefaultS3CrtClientBuilder implements S3CrtAsyncClientBuilder {
@@ -123,6 +133,7 @@ public static final class DefaultS3CrtClientBuilder implements S3CrtAsyncClientB
         private URI endpointOverride;
         private Boolean checksumValidationEnabled;
         private S3CrtHttpConfiguration httpConfiguration;
+        private S3CrtRetryConfiguration retryConfiguration;
 
         public AwsCredentialsProvider credentialsProvider() {
             return credentialsProvider;
@@ -206,6 +217,12 @@ public S3CrtAsyncClientBuilder httpConfiguration(S3CrtHttpConfiguration configur
             return this;
         }
 
+        @Override
+        public S3CrtAsyncClientBuilder retryConfiguration(S3CrtRetryConfiguration retryConfiguration) {
+            this.retryConfiguration = retryConfiguration;
+            return this;
+        }
+
         @Override
         public S3CrtAsyncClient build() {
             return new DefaultS3CrtAsyncClient(this);