All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
software.amazon.awssdk.services.neptune.waiters.DefaultNeptuneAsyncWaiter Maven / Gradle / Ivy
/*
* 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.neptune.waiters;
import java.time.Duration;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.function.Consumer;
import software.amazon.awssdk.annotations.Generated;
import software.amazon.awssdk.annotations.SdkInternalApi;
import software.amazon.awssdk.annotations.ThreadSafe;
import software.amazon.awssdk.awscore.AwsRequestOverrideConfiguration;
import software.amazon.awssdk.awscore.exception.AwsServiceException;
import software.amazon.awssdk.core.ApiName;
import software.amazon.awssdk.core.internal.waiters.WaiterAttribute;
import software.amazon.awssdk.core.waiters.AsyncWaiter;
import software.amazon.awssdk.core.waiters.WaiterAcceptor;
import software.amazon.awssdk.core.waiters.WaiterOverrideConfiguration;
import software.amazon.awssdk.core.waiters.WaiterResponse;
import software.amazon.awssdk.retries.api.BackoffStrategy;
import software.amazon.awssdk.services.neptune.NeptuneAsyncClient;
import software.amazon.awssdk.services.neptune.jmespath.internal.JmesPathRuntime;
import software.amazon.awssdk.services.neptune.model.DescribeDbInstancesRequest;
import software.amazon.awssdk.services.neptune.model.DescribeDbInstancesResponse;
import software.amazon.awssdk.services.neptune.model.NeptuneRequest;
import software.amazon.awssdk.services.neptune.waiters.internal.WaitersRuntime;
import software.amazon.awssdk.utils.AttributeMap;
import software.amazon.awssdk.utils.SdkAutoCloseable;
import software.amazon.awssdk.utils.ThreadFactoryBuilder;
@Generated("software.amazon.awssdk:codegen")
@SdkInternalApi
@ThreadSafe
final class DefaultNeptuneAsyncWaiter implements NeptuneAsyncWaiter {
private static final WaiterAttribute CLIENT_ATTRIBUTE = new WaiterAttribute<>(SdkAutoCloseable.class);
private static final WaiterAttribute SCHEDULED_EXECUTOR_SERVICE_ATTRIBUTE = new WaiterAttribute<>(
ScheduledExecutorService.class);
private final NeptuneAsyncClient client;
private final AttributeMap managedResources;
private final AsyncWaiter dBInstanceAvailableWaiter;
private final AsyncWaiter dBInstanceDeletedWaiter;
private final ScheduledExecutorService executorService;
private DefaultNeptuneAsyncWaiter(DefaultBuilder builder) {
AttributeMap.Builder attributeMapBuilder = AttributeMap.builder();
if (builder.client == null) {
this.client = NeptuneAsyncClient.builder().build();
attributeMapBuilder.put(CLIENT_ATTRIBUTE, this.client);
} else {
this.client = builder.client;
}
if (builder.executorService == null) {
this.executorService = Executors.newScheduledThreadPool(1,
new ThreadFactoryBuilder().threadNamePrefix("waiters-ScheduledExecutor").build());
attributeMapBuilder.put(SCHEDULED_EXECUTOR_SERVICE_ATTRIBUTE, this.executorService);
} else {
this.executorService = builder.executorService;
}
managedResources = attributeMapBuilder.build();
this.dBInstanceAvailableWaiter = AsyncWaiter.builder(DescribeDbInstancesResponse.class)
.acceptors(dBInstanceAvailableWaiterAcceptors())
.overrideConfiguration(dBInstanceAvailableWaiterConfig(builder.overrideConfiguration))
.scheduledExecutorService(executorService).build();
this.dBInstanceDeletedWaiter = AsyncWaiter.builder(DescribeDbInstancesResponse.class)
.acceptors(dBInstanceDeletedWaiterAcceptors())
.overrideConfiguration(dBInstanceDeletedWaiterConfig(builder.overrideConfiguration))
.scheduledExecutorService(executorService).build();
}
private static String errorCode(Throwable error) {
if (error instanceof AwsServiceException) {
return ((AwsServiceException) error).awsErrorDetails().errorCode();
}
return null;
}
@Override
public CompletableFuture> waitUntilDBInstanceAvailable(
DescribeDbInstancesRequest describeDbInstancesRequest) {
return dBInstanceAvailableWaiter.runAsync(() -> client
.describeDBInstances(applyWaitersUserAgent(describeDbInstancesRequest)));
}
@Override
public CompletableFuture> waitUntilDBInstanceAvailable(
DescribeDbInstancesRequest describeDbInstancesRequest, WaiterOverrideConfiguration overrideConfig) {
return dBInstanceAvailableWaiter.runAsync(
() -> client.describeDBInstances(applyWaitersUserAgent(describeDbInstancesRequest)),
dBInstanceAvailableWaiterConfig(overrideConfig));
}
@Override
public CompletableFuture> waitUntilDBInstanceDeleted(
DescribeDbInstancesRequest describeDbInstancesRequest) {
return dBInstanceDeletedWaiter.runAsync(() -> client
.describeDBInstances(applyWaitersUserAgent(describeDbInstancesRequest)));
}
@Override
public CompletableFuture> waitUntilDBInstanceDeleted(
DescribeDbInstancesRequest describeDbInstancesRequest, WaiterOverrideConfiguration overrideConfig) {
return dBInstanceDeletedWaiter.runAsync(
() -> client.describeDBInstances(applyWaitersUserAgent(describeDbInstancesRequest)),
dBInstanceDeletedWaiterConfig(overrideConfig));
}
private static List> dBInstanceAvailableWaiterAcceptors() {
List> result = new ArrayList<>();
result.add(WaiterAcceptor.successOnResponseAcceptor(response -> {
JmesPathRuntime.Value input = new JmesPathRuntime.Value(response);
List resultValues = input.field("DBInstances").flatten().field("DBInstanceStatus").values();
return !resultValues.isEmpty() && resultValues.stream().allMatch(v -> Objects.equals(v, "available"));
}));
result.add(WaiterAcceptor.errorOnResponseAcceptor(
response -> {
JmesPathRuntime.Value input = new JmesPathRuntime.Value(response);
List resultValues = input.field("DBInstances").flatten().field("DBInstanceStatus").values();
return !resultValues.isEmpty() && resultValues.stream().anyMatch(v -> Objects.equals(v, "deleted"));
},
"A waiter acceptor with the matcher (pathAny) was matched on parameter (DBInstances[].DBInstanceStatus=deleted) and transitioned the waiter to failure state"));
result.add(WaiterAcceptor.errorOnResponseAcceptor(
response -> {
JmesPathRuntime.Value input = new JmesPathRuntime.Value(response);
List resultValues = input.field("DBInstances").flatten().field("DBInstanceStatus").values();
return !resultValues.isEmpty() && resultValues.stream().anyMatch(v -> Objects.equals(v, "deleting"));
},
"A waiter acceptor with the matcher (pathAny) was matched on parameter (DBInstances[].DBInstanceStatus=deleting) and transitioned the waiter to failure state"));
result.add(WaiterAcceptor.errorOnResponseAcceptor(
response -> {
JmesPathRuntime.Value input = new JmesPathRuntime.Value(response);
List resultValues = input.field("DBInstances").flatten().field("DBInstanceStatus").values();
return !resultValues.isEmpty() && resultValues.stream().anyMatch(v -> Objects.equals(v, "failed"));
},
"A waiter acceptor with the matcher (pathAny) was matched on parameter (DBInstances[].DBInstanceStatus=failed) and transitioned the waiter to failure state"));
result.add(WaiterAcceptor.errorOnResponseAcceptor(
response -> {
JmesPathRuntime.Value input = new JmesPathRuntime.Value(response);
List resultValues = input.field("DBInstances").flatten().field("DBInstanceStatus").values();
return !resultValues.isEmpty()
&& resultValues.stream().anyMatch(v -> Objects.equals(v, "incompatible-restore"));
},
"A waiter acceptor with the matcher (pathAny) was matched on parameter (DBInstances[].DBInstanceStatus=incompatible-restore) and transitioned the waiter to failure state"));
result.add(WaiterAcceptor.errorOnResponseAcceptor(
response -> {
JmesPathRuntime.Value input = new JmesPathRuntime.Value(response);
List resultValues = input.field("DBInstances").flatten().field("DBInstanceStatus").values();
return !resultValues.isEmpty()
&& resultValues.stream().anyMatch(v -> Objects.equals(v, "incompatible-parameters"));
},
"A waiter acceptor with the matcher (pathAny) was matched on parameter (DBInstances[].DBInstanceStatus=incompatible-parameters) and transitioned the waiter to failure state"));
result.addAll(WaitersRuntime.DEFAULT_ACCEPTORS);
return result;
}
private static List> dBInstanceDeletedWaiterAcceptors() {
List> result = new ArrayList<>();
result.add(WaiterAcceptor.successOnResponseAcceptor(response -> {
JmesPathRuntime.Value input = new JmesPathRuntime.Value(response);
List resultValues = input.field("DBInstances").flatten().field("DBInstanceStatus").values();
return !resultValues.isEmpty() && resultValues.stream().allMatch(v -> Objects.equals(v, "deleted"));
}));
result.add(WaiterAcceptor.successOnExceptionAcceptor(error -> Objects.equals(errorCode(error), "DBInstanceNotFound")));
result.add(WaiterAcceptor.errorOnResponseAcceptor(
response -> {
JmesPathRuntime.Value input = new JmesPathRuntime.Value(response);
List resultValues = input.field("DBInstances").flatten().field("DBInstanceStatus").values();
return !resultValues.isEmpty() && resultValues.stream().anyMatch(v -> Objects.equals(v, "creating"));
},
"A waiter acceptor with the matcher (pathAny) was matched on parameter (DBInstances[].DBInstanceStatus=creating) and transitioned the waiter to failure state"));
result.add(WaiterAcceptor.errorOnResponseAcceptor(
response -> {
JmesPathRuntime.Value input = new JmesPathRuntime.Value(response);
List resultValues = input.field("DBInstances").flatten().field("DBInstanceStatus").values();
return !resultValues.isEmpty() && resultValues.stream().anyMatch(v -> Objects.equals(v, "modifying"));
},
"A waiter acceptor with the matcher (pathAny) was matched on parameter (DBInstances[].DBInstanceStatus=modifying) and transitioned the waiter to failure state"));
result.add(WaiterAcceptor.errorOnResponseAcceptor(
response -> {
JmesPathRuntime.Value input = new JmesPathRuntime.Value(response);
List resultValues = input.field("DBInstances").flatten().field("DBInstanceStatus").values();
return !resultValues.isEmpty() && resultValues.stream().anyMatch(v -> Objects.equals(v, "rebooting"));
},
"A waiter acceptor with the matcher (pathAny) was matched on parameter (DBInstances[].DBInstanceStatus=rebooting) and transitioned the waiter to failure state"));
result.add(WaiterAcceptor.errorOnResponseAcceptor(
response -> {
JmesPathRuntime.Value input = new JmesPathRuntime.Value(response);
List resultValues = input.field("DBInstances").flatten().field("DBInstanceStatus").values();
return !resultValues.isEmpty()
&& resultValues.stream().anyMatch(v -> Objects.equals(v, "resetting-master-credentials"));
},
"A waiter acceptor with the matcher (pathAny) was matched on parameter (DBInstances[].DBInstanceStatus=resetting-master-credentials) and transitioned the waiter to failure state"));
result.addAll(WaitersRuntime.DEFAULT_ACCEPTORS);
return result;
}
private static WaiterOverrideConfiguration dBInstanceAvailableWaiterConfig(WaiterOverrideConfiguration overrideConfig) {
Optional optionalOverrideConfig = Optional.ofNullable(overrideConfig);
int maxAttempts = optionalOverrideConfig.flatMap(WaiterOverrideConfiguration::maxAttempts).orElse(60);
BackoffStrategy backoffStrategy = optionalOverrideConfig.flatMap(WaiterOverrideConfiguration::backoffStrategyV2).orElse(
BackoffStrategy.fixedDelayWithoutJitter(Duration.ofSeconds(30)));
Duration waitTimeout = optionalOverrideConfig.flatMap(WaiterOverrideConfiguration::waitTimeout).orElse(null);
return WaiterOverrideConfiguration.builder().maxAttempts(maxAttempts).backoffStrategyV2(backoffStrategy)
.waitTimeout(waitTimeout).build();
}
private static WaiterOverrideConfiguration dBInstanceDeletedWaiterConfig(WaiterOverrideConfiguration overrideConfig) {
Optional optionalOverrideConfig = Optional.ofNullable(overrideConfig);
int maxAttempts = optionalOverrideConfig.flatMap(WaiterOverrideConfiguration::maxAttempts).orElse(60);
BackoffStrategy backoffStrategy = optionalOverrideConfig.flatMap(WaiterOverrideConfiguration::backoffStrategyV2).orElse(
BackoffStrategy.fixedDelayWithoutJitter(Duration.ofSeconds(30)));
Duration waitTimeout = optionalOverrideConfig.flatMap(WaiterOverrideConfiguration::waitTimeout).orElse(null);
return WaiterOverrideConfiguration.builder().maxAttempts(maxAttempts).backoffStrategyV2(backoffStrategy)
.waitTimeout(waitTimeout).build();
}
@Override
public void close() {
managedResources.close();
}
public static NeptuneAsyncWaiter.Builder builder() {
return new DefaultBuilder();
}
private T applyWaitersUserAgent(T request) {
Consumer userAgentApplier = b -> b.addApiName(ApiName.builder()
.version("waiter").name("hll").build());
AwsRequestOverrideConfiguration overrideConfiguration = request.overrideConfiguration()
.map(c -> c.toBuilder().applyMutation(userAgentApplier).build())
.orElse((AwsRequestOverrideConfiguration.builder().applyMutation(userAgentApplier).build()));
return (T) request.toBuilder().overrideConfiguration(overrideConfiguration).build();
}
public static final class DefaultBuilder implements NeptuneAsyncWaiter.Builder {
private NeptuneAsyncClient client;
private WaiterOverrideConfiguration overrideConfiguration;
private ScheduledExecutorService executorService;
private DefaultBuilder() {
}
@Override
public NeptuneAsyncWaiter.Builder scheduledExecutorService(ScheduledExecutorService executorService) {
this.executorService = executorService;
return this;
}
@Override
public NeptuneAsyncWaiter.Builder overrideConfiguration(WaiterOverrideConfiguration overrideConfiguration) {
this.overrideConfiguration = overrideConfiguration;
return this;
}
@Override
public NeptuneAsyncWaiter.Builder client(NeptuneAsyncClient client) {
this.client = client;
return this;
}
public NeptuneAsyncWaiter build() {
return new DefaultNeptuneAsyncWaiter(this);
}
}
}
