1 /*
2  * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License").
5  * You may not use this file except in compliance with the License.
6  * A copy of the License is located at
7  *
8  *  http://aws.amazon.com/apache2.0
9  *
10  * or in the "license" file accompanying this file. This file is distributed
11  * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
12  * express or implied. See the License for the specific language governing
13  * permissions and limitations under the License.
14  */
15 
16 package software.amazon.awssdk.auth.credentials.internal;
17 
18 import java.io.IOException;
19 import software.amazon.awssdk.annotations.SdkInternalApi;
20 import software.amazon.awssdk.http.HttpStatusFamily;
21 import software.amazon.awssdk.regions.util.ResourcesEndpointRetryParameters;
22 import software.amazon.awssdk.regions.util.ResourcesEndpointRetryPolicy;
23 
24 @SdkInternalApi
25 public final class ContainerCredentialsRetryPolicy implements ResourcesEndpointRetryPolicy {
26 
27     /** Max number of times a request is retried before failing. */
28     private static final int MAX_RETRIES = 5;
29 
30     @Override
shouldRetry(int retriesAttempted, ResourcesEndpointRetryParameters retryParams)31     public boolean shouldRetry(int retriesAttempted, ResourcesEndpointRetryParameters retryParams) {
32         if (retriesAttempted >= MAX_RETRIES) {
33             return false;
34         }
35 
36         Integer statusCode = retryParams.getStatusCode();
37         if (statusCode != null && HttpStatusFamily.of(statusCode) == HttpStatusFamily.SERVER_ERROR) {
38             return true;
39         }
40 
41         return retryParams.getException() instanceof IOException;
42     }
43 
44 }
45