org.apache.hudi.aws.utils

Class DynamoTableUtils

java.lang.Object

org.apache.hudi.aws.utils.DynamoTableUtils

public class DynamoTableUtils
extends Object

Reused code from https://github.com/aws/aws-sdk-java-v2/blob/master/services/dynamodb/src/test/java/utils/test/util/TableUtils.java Utility methods for working with DynamoDB tables.

 // ... create DynamoDB table ...
 try {
     waitUntilActive(dynamoDB, myTableName());
 } catch (SdkClientException e) {
     // table didn't become active
 }
 // ... start making calls to table ...
 

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	DynamoTableUtils.TableNeverTransitionedToStateException Thrown by DynamoTableUtils when a table never reaches a desired state

Constructor Summary

Constructors

Constructor and Description
DynamoTableUtils()

Method Summary

Modifier and Type	Method and Description
static boolean	createTableIfNotExists(software.amazon.awssdk.services.dynamodb.DynamoDbClient dynamo, software.amazon.awssdk.services.dynamodb.model.CreateTableRequest createTableRequest) Creates the table and ignores any errors if it already exists.
static boolean	deleteTableIfExists(software.amazon.awssdk.services.dynamodb.DynamoDbClient dynamo, software.amazon.awssdk.services.dynamodb.model.DeleteTableRequest deleteTableRequest) Deletes the table and ignores any errors if it doesn't exist.
static void	waitUntilActive(software.amazon.awssdk.services.dynamodb.DynamoDbClient dynamo, String tableName) Waits up to 10 minutes for a specified DynamoDB table to move into the ACTIVE state.
static void	waitUntilActive(software.amazon.awssdk.services.dynamodb.DynamoDbClient dynamo, String tableName, int timeout, int interval) Waits up to a specified amount of time for a specified DynamoDB table to move into the ACTIVE state.
static void	waitUntilExists(software.amazon.awssdk.services.dynamodb.DynamoDbClient dynamo, String tableName) Waits up to 10 minutes for a specified DynamoDB table to resolve, indicating that it exists.
static void	waitUntilExists(software.amazon.awssdk.services.dynamodb.DynamoDbClient dynamo, String tableName, int timeout, int interval) Waits up to a specified amount of time for a specified DynamoDB table to resolve, indicating that it exists.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

DynamoTableUtils

public DynamoTableUtils()

Method Detail

waitUntilExists

public static void waitUntilExists(software.amazon.awssdk.services.dynamodb.DynamoDbClient dynamo,
                                   String tableName)
                            throws InterruptedException

Waits up to 10 minutes for a specified DynamoDB table to resolve, indicating that it exists. If the table doesn't return a result after this time, a SdkClientException is thrown.

Parameters:

dynamo - The DynamoDB client to use to make requests.

tableName - The name of the table being resolved.

Throws:

software.amazon.awssdk.core.exception.SdkClientException - If the specified table does not resolve before this method times out and stops polling.

InterruptedException - If the thread is interrupted while waiting for the table to resolve.

waitUntilExists

public static void waitUntilExists(software.amazon.awssdk.services.dynamodb.DynamoDbClient dynamo,
                                   String tableName,
                                   int timeout,
                                   int interval)
                            throws InterruptedException

Waits up to a specified amount of time for a specified DynamoDB table to resolve, indicating that it exists. If the table doesn't return a result after this time, a SdkClientException is thrown.

Parameters:

dynamo - The DynamoDB client to use to make requests.

tableName - The name of the table being resolved.

timeout - The maximum number of milliseconds to wait.

interval - The poll interval in milliseconds.

Throws:

software.amazon.awssdk.core.exception.SdkClientException - If the specified table does not resolve before this method times out and stops polling.

InterruptedException - If the thread is interrupted while waiting for the table to resolve.

waitUntilActive

public static void waitUntilActive(software.amazon.awssdk.services.dynamodb.DynamoDbClient dynamo,
                                   String tableName)
                            throws InterruptedException,
                                   DynamoTableUtils.TableNeverTransitionedToStateException

Waits up to 10 minutes for a specified DynamoDB table to move into the ACTIVE state. If the table does not exist or does not transition to the ACTIVE state after this time, then SdkClientException is thrown.

Parameters:

dynamo - The DynamoDB client to use to make requests.

tableName - The name of the table whose status is being checked.

Throws:

DynamoTableUtils.TableNeverTransitionedToStateException - If the specified table does not exist or does not transition into the ACTIVE state before this method times out and stops polling.

InterruptedException - If the thread is interrupted while waiting for the table to transition into the ACTIVE state.

waitUntilActive

public static void waitUntilActive(software.amazon.awssdk.services.dynamodb.DynamoDbClient dynamo,
                                   String tableName,
                                   int timeout,
                                   int interval)
                            throws InterruptedException,
                                   DynamoTableUtils.TableNeverTransitionedToStateException

Waits up to a specified amount of time for a specified DynamoDB table to move into the ACTIVE state. If the table does not exist or does not transition to the ACTIVE state after this time, then a SdkClientException is thrown.

Parameters:

dynamo - The DynamoDB client to use to make requests.

tableName - The name of the table whose status is being checked.

timeout - The maximum number of milliseconds to wait.

interval - The poll interval in milliseconds.

Throws:

DynamoTableUtils.TableNeverTransitionedToStateException - If the specified table does not exist or does not transition into the ACTIVE state before this method times out and stops polling.

InterruptedException - If the thread is interrupted while waiting for the table to transition into the ACTIVE state.

createTableIfNotExists

public static boolean createTableIfNotExists(software.amazon.awssdk.services.dynamodb.DynamoDbClient dynamo,
                                             software.amazon.awssdk.services.dynamodb.model.CreateTableRequest createTableRequest)

Creates the table and ignores any errors if it already exists.

Parameters:

dynamo - The Dynamo client to use.

createTableRequest - The create table request.

Returns:

True if created, false otherwise.

deleteTableIfExists

public static boolean deleteTableIfExists(software.amazon.awssdk.services.dynamodb.DynamoDbClient dynamo,
                                          software.amazon.awssdk.services.dynamodb.model.DeleteTableRequest deleteTableRequest)

Deletes the table and ignores any errors if it doesn't exist.

Parameters:

dynamo - The Dynamo client to use.

deleteTableRequest - The delete table request.

Returns:

True if deleted, false otherwise.