Java V2 updated DynamoDB to use billingmode

javav2/example_code/dynamodb/src/main/java/com/example/dynamodb/CreateTable.java

@@ -5,17 +5,20 @@
 
 // snippet-start:[dynamodb.java2.create_table.main]
 // snippet-start:[dynamodb.java2.create_table.import]
+
 import software.amazon.awssdk.core.waiters.WaiterResponse;
 import software.amazon.awssdk.regions.Region;
 import software.amazon.awssdk.services.dynamodb.DynamoDbClient;
 import software.amazon.awssdk.services.dynamodb.model.AttributeDefinition;
+import software.amazon.awssdk.services.dynamodb.model.BillingMode;
 import software.amazon.awssdk.services.dynamodb.model.CreateTableRequest;
 import software.amazon.awssdk.services.dynamodb.model.CreateTableResponse;
 import software.amazon.awssdk.services.dynamodb.model.DescribeTableRequest;
 import software.amazon.awssdk.services.dynamodb.model.DescribeTableResponse;
 import software.amazon.awssdk.services.dynamodb.model.DynamoDbException;
 import software.amazon.awssdk.services.dynamodb.model.KeySchemaElement;
 import software.amazon.awssdk.services.dynamodb.model.KeyType;
+import software.amazon.awssdk.services.dynamodb.model.OnDemandThroughput;
 import software.amazon.awssdk.services.dynamodb.model.ProvisionedThroughput;
 import software.amazon.awssdk.services.dynamodb.model.ScalarAttributeType;
 import software.amazon.awssdk.services.dynamodb.waiters.DynamoDbWaiter;
@@ -24,22 +27,22 @@
 /**
  * Before running this Java V2 code example, set up your development
  * environment, including your credentials.
- *
+ * <p>
  * For more information, see the following documentation topic:
- *
+ * <p>
  * https://docs.aws.amazon.com/sdk-for-java/latest/developer-guide/get-started.html
  */
 public class CreateTable {
     public static void main(String[] args) {
         final String usage = """
 
-                Usage:
-                    <tableName> <key>
+            Usage:
+                <tableName> <key>
 
-                Where:
-                    tableName - The Amazon DynamoDB table to create (for example, Music3).
-                    key - The key for the Amazon DynamoDB table (for example, Artist).
-                """;
+            Where:
+                tableName - The Amazon DynamoDB table to create (for example, Music3).
+                key - The key for the Amazon DynamoDB table (for example, Artist).
+            """;
 
         if (args.length != 2) {
             System.out.println(usage);
@@ -51,8 +54,8 @@ public static void main(String[] args) {
         System.out.println("Creating an Amazon DynamoDB table " + tableName + " with a simple primary key: " + key);
         Region region = Region.US_EAST_1;
         DynamoDbClient ddb = DynamoDbClient.builder()
-                .region(region)
-                .build();
+            .region(region)
+            .build();
 
         String result = createTable(ddb, tableName, key);
         System.out.println("New table is " + result);
@@ -62,27 +65,24 @@ public static void main(String[] args) {
     public static String createTable(DynamoDbClient ddb, String tableName, String key) {
         DynamoDbWaiter dbWaiter = ddb.waiter();
         CreateTableRequest request = CreateTableRequest.builder()
-                .attributeDefinitions(AttributeDefinition.builder()
-                        .attributeName(key)
-                        .attributeType(ScalarAttributeType.S)
-                        .build())
-                .keySchema(KeySchemaElement.builder()
-                        .attributeName(key)
-                        .keyType(KeyType.HASH)
-                        .build())
-                .provisionedThroughput(ProvisionedThroughput.builder()
-                        .readCapacityUnits(10L)
-                        .writeCapacityUnits(10L)
-                        .build())
-                .tableName(tableName)
-                .build();
+            .attributeDefinitions(AttributeDefinition.builder()
+                .attributeName(key)
+                .attributeType(ScalarAttributeType.S)
+                .build())
+            .keySchema(KeySchemaElement.builder()
+                .attributeName(key)
+                .keyType(KeyType.HASH)
+                .build())
+            .billingMode(BillingMode.PAY_PER_REQUEST) //  DynamoDB automatically scales based on traffic.
+            .tableName(tableName)
+            .build();
 
         String newTable;
         try {
             CreateTableResponse response = ddb.createTable(request);
             DescribeTableRequest tableRequest = DescribeTableRequest.builder()
-                    .tableName(tableName)
-                    .build();
+                .tableName(tableName)
+                .build();
 
             // Wait until the Amazon DynamoDB table is created.
             WaiterResponse<DescribeTableResponse> waiterResponse = dbWaiter.waitUntilTableExists(tableRequest);

javav2/example_code/dynamodb/src/main/java/com/example/dynamodb/CreateTableCompositeKey.java

@@ -6,6 +6,7 @@
 // snippet-start:[dynamodb.java2.create_table_composite_key.main]
 // snippet-start:[dynamodb.java2.create_table_composite_key.import]
 import software.amazon.awssdk.regions.Region;
+import software.amazon.awssdk.services.dynamodb.model.BillingMode;
 import software.amazon.awssdk.services.dynamodb.model.DynamoDbException;
 import software.amazon.awssdk.services.dynamodb.model.AttributeDefinition;
 import software.amazon.awssdk.services.dynamodb.model.CreateTableRequest;
@@ -74,9 +75,7 @@ public static String createTableComKey(DynamoDbClient ddb, String tableName) {
                                 .attributeName("Greeting")
                                 .keyType(KeyType.RANGE)
                                 .build())
-                .provisionedThroughput(ProvisionedThroughput.builder()
-                        .readCapacityUnits(10L)
-                        .writeCapacityUnits(10L).build())
+            .billingMode(BillingMode.PAY_PER_REQUEST) //  DynamoDB automatically scales based on traffic.
                 .tableName(tableName)
                 .build();
 

javav2/example_code/dynamodb/src/main/java/com/example/dynamodb/enhanced/EnhancedCreateTable.java

@@ -7,6 +7,7 @@
 import software.amazon.awssdk.enhanced.dynamodb.DynamoDbEnhancedClient;
 import software.amazon.awssdk.enhanced.dynamodb.DynamoDbTable;
 import software.amazon.awssdk.enhanced.dynamodb.TableSchema;
+import software.amazon.awssdk.enhanced.dynamodb.model.CreateTableEnhancedRequest;
 import software.amazon.awssdk.services.dynamodb.model.DescribeTableResponse;
 import software.amazon.awssdk.services.dynamodb.waiters.DynamoDbWaiter;
 // snippet-end:[dynamodb.java2.mapping.enhancedcreatetable.import]
@@ -16,14 +17,9 @@ public class EnhancedCreateTable {
     public static void createTable(DynamoDbEnhancedClient enhancedClient) {
         // Create a DynamoDbTable object
         DynamoDbTable<Customer> customerTable = enhancedClient.table("Customer", TableSchema.fromBean(Customer.class));
-        // Create the table
-        customerTable.createTable(builder -> builder
-                .provisionedThroughput(b -> b
-                        .readCapacityUnits(10L)
-                        .writeCapacityUnits(10L)
-                        .build())
-        );
 
+        // Create the table (defaults to PAY_PER_REQUEST if no provisioned throughput is set)
+        customerTable.createTable(CreateTableEnhancedRequest.builder().build());
         System.out.println("Waiting for table creation...");
 
         try (DynamoDbWaiter waiter = DynamoDbWaiter.create()) { // DynamoDbWaiter is Autocloseable