add generateDeleteTableRequest method on DynamoDBMapper

Author: Paulo Miguel Almeida

aws-java-sdk-dynamodb/src/main/java/com/amazonaws/services/dynamodbv2/datamodeling/AbstractDynamoDBMapper.java

@@ -20,6 +20,7 @@
 import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBMapper.FailedBatch;
 import com.amazonaws.services.dynamodbv2.model.AttributeValue;
 import com.amazonaws.services.dynamodbv2.model.CreateTableRequest;
+import com.amazonaws.services.dynamodbv2.model.DeleteTableRequest;
 import com.amazonaws.services.s3.model.Region;
 
 /**
@@ -300,4 +301,8 @@ public CreateTableRequest generateCreateTableRequest(Class<?> clazz) {
         throw new UnsupportedOperationException();
     }
 
+    @Override
+    public DeleteTableRequest generateDeleteTableRequest(Class<?> clazz) {
+        throw new UnsupportedOperationException();
+    }
 }

aws-java-sdk-dynamodb/src/main/java/com/amazonaws/services/dynamodbv2/datamodeling/DynamoDBMapper.java

@@ -55,6 +55,7 @@
 import com.amazonaws.services.dynamodbv2.model.ConditionalCheckFailedException;
 import com.amazonaws.services.dynamodbv2.model.ConditionalOperator;
 import com.amazonaws.services.dynamodbv2.model.CreateTableRequest;
+import com.amazonaws.services.dynamodbv2.model.DeleteTableRequest;
 import com.amazonaws.services.dynamodbv2.model.DeleteItemRequest;
 import com.amazonaws.services.dynamodbv2.model.DeleteRequest;
 import com.amazonaws.services.dynamodbv2.model.ExpectedAttributeValue;
@@ -2675,4 +2676,10 @@ public CreateTableRequest generateCreateTableRequest(Class<?> clazz) {
         return schemaParser.parseTablePojoToCreateTableRequest(
                 clazz, config, reflector, converter);
     }
+
+    @Override
+    public DeleteTableRequest generateDeleteTableRequest(Class<?> clazz) {
+        return schemaParser.parseTablePojoToDeleteTableRequest(
+                clazz, config);
+    }
 }

aws-java-sdk-dynamodb/src/main/java/com/amazonaws/services/dynamodbv2/datamodeling/DynamoDBTableSchemaParser.java

@@ -26,6 +26,7 @@
 import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBMapperFieldModel.DynamoDBAttributeType;
 import com.amazonaws.services.dynamodbv2.model.AttributeDefinition;
 import com.amazonaws.services.dynamodbv2.model.CreateTableRequest;
+import com.amazonaws.services.dynamodbv2.model.DeleteTableRequest;
 import com.amazonaws.services.dynamodbv2.model.GlobalSecondaryIndex;
 import com.amazonaws.services.dynamodbv2.model.KeySchemaElement;
 import com.amazonaws.services.dynamodbv2.model.KeyType;
@@ -105,6 +106,26 @@ CreateTableRequest parseTablePojoToCreateTableRequest(
         return createTableRequest;
     }
 
+    /**
+     * Parse the given POJO class and return the DeleteTableRequest for the
+     * DynamoDB table it represents.
+     *
+     * @param clazz
+     *            The POJO class.
+     * @param config
+     *            The DynamoDBMapperConfig which contains the TableNameOverrides
+     *            parameter used to determine the table name.
+     */
+    DeleteTableRequest parseTablePojoToDeleteTableRequest(
+            Class<?> clazz,
+            DynamoDBMapperConfig config) {
+
+        DeleteTableRequest deleteTableRequest = new DeleteTableRequest();
+        deleteTableRequest.setTableName(DynamoDBMapper.internalGetTableName(clazz, null, config));
+
+        return deleteTableRequest;
+    }
+
     TableIndexesInfo parseTableIndexes(final Class<?> clazz, final DynamoDBReflector reflector) {
         synchronized(tableIndexesInfoCache) {
             if ( !tableIndexesInfoCache.containsKey(clazz) ) {

aws-java-sdk-dynamodb/src/main/java/com/amazonaws/services/dynamodbv2/datamodeling/IDynamoDBMapper.java

@@ -25,6 +25,7 @@
 import com.amazonaws.services.dynamodbv2.model.BatchGetItemRequest;
 import com.amazonaws.services.dynamodbv2.model.BatchWriteItemRequest;
 import com.amazonaws.services.dynamodbv2.model.CreateTableRequest;
+import com.amazonaws.services.dynamodbv2.model.DeleteTableRequest;
 import com.amazonaws.services.dynamodbv2.model.PutItemRequest;
 import com.amazonaws.services.dynamodbv2.model.UpdateItemRequest;
 import com.amazonaws.services.s3.model.Region;
@@ -742,4 +743,11 @@ <T> QueryResultPage<T> queryPage(Class<T> clazz,
      */
     CreateTableRequest generateCreateTableRequest(Class<?> clazz);
 
+
+    /**
+     * Parse the given POJO class and return the DeleteTableRequest for the DynamoDB table it
+     * represents.
+     */
+    DeleteTableRequest generateDeleteTableRequest(Class<?> clazz);
+
 }

aws-java-sdk-dynamodb/src/test/java/com/amazonaws/services/dynamodbv2/datamodeling/DynamoDBMapperTest.java

@@ -0,0 +1,29 @@
+package com.amazonaws.services.dynamodbv2.datamodeling;
+
+import static org.junit.Assert.assertEquals;
+
+import org.junit.Test;
+import com.amazonaws.services.dynamodbv2.model.DeleteTableRequest;
+
+public class DynamoDBMapperTest {
+
+    @Test
+    public void testGenerateDeleteTableRequest(){
+        String prefix = "DEV-";
+        final String tableName = "OBJECTORMEXAMPLE";
+
+        DynamoDBMapperConfig.TableNameOverride tableNameOverride =
+                DynamoDBMapperConfig.TableNameOverride.withTableNamePrefix(prefix);
+        DynamoDBMapperConfig config = new DynamoDBMapperConfig(tableNameOverride);
+        DynamoDBMapper dynamoDBMapper = new DynamoDBMapper(null, config);
+
+        // Create the simplest table as possible for our test purposes only
+        @DynamoDBTable(tableName = tableName)
+        class ObjectORMExample{}
+
+        // test
+        DeleteTableRequest deleteTableRequest = dynamoDBMapper.generateDeleteTableRequest(ObjectORMExample.class);
+        assertEquals(deleteTableRequest.getTableName(), prefix.concat(tableName));
+    }
+
+}