feat(aws-dynamodb): Support DynamoDB PITR (#701)

This patch supports DynamoDB PITR.
aws · Sep 13, 2018 · 7a4d7b7 · 7a4d7b7
1 parent 57b92be
commit 7a4d7b7
Show file tree

Hide file tree

Showing 4 changed files with 41 additions and 0 deletions.
diff --git a/packages/@aws-cdk/aws-dynamodb/lib/table.ts b/packages/@aws-cdk/aws-dynamodb/lib/table.ts
@@ -26,6 +26,12 @@ export interface TableProps {
      */
     tableName?: string;
 
+    /**
+     * Whether point-in-time recovery is enabled.
+     * @default undefined, point-in-time recovery is disabled
+     */
+    pitrEnabled?: boolean;
+
     /**
      * Whether server-side encryption is enabled.
      * @default undefined, server-side encryption is disabled
@@ -122,6 +128,7 @@ export class Table extends Construct {
             tableName: props.tableName,
             keySchema: this.keySchema,
             attributeDefinitions: this.attributeDefinitions,
+            pointInTimeRecoverySpecification: props.pitrEnabled ? { pointInTimeRecoveryEnabled: props.pitrEnabled } : undefined,
             provisionedThroughput: { readCapacityUnits, writeCapacityUnits },
             sseSpecification: props.sseEnabled ? { sseEnabled: props.sseEnabled } : undefined,
             streamSpecification: props.streamSpecification ? { streamViewType: props.streamSpecification } : undefined,

diff --git a/packages/@aws-cdk/aws-dynamodb/test/integ.dynamodb.expected.json b/packages/@aws-cdk/aws-dynamodb/test/integ.dynamodb.expected.json
@@ -27,6 +27,9 @@
             "AttributeType": "N"
           }
         ],
+        "PointInTimeRecoverySpecification": {
+          "PointInTimeRecoveryEnabled": true
+        },
         "SSESpecification": {
           "SSEEnabled": true
         },

diff --git a/packages/@aws-cdk/aws-dynamodb/test/integ.dynamodb.ts b/packages/@aws-cdk/aws-dynamodb/test/integ.dynamodb.ts
@@ -6,6 +6,7 @@ const app = new App(process.argv);
 const stack = new Stack(app, 'aws-cdk-dynamodb');
 
 const table = new Table(stack, 'Table', {
+    pitrEnabled: true,
     sseEnabled: true,
     streamSpecification: StreamViewType.KeysOnly,
     ttlAttributeName: 'timeToLive'

diff --git a/packages/@aws-cdk/aws-dynamodb/test/test.dynamodb.ts b/packages/@aws-cdk/aws-dynamodb/test/test.dynamodb.ts
@@ -60,6 +60,34 @@ export = {
 
             test.done();
         },
+        'point-in-time recovery is not enabled'(test: Test) {
+            const app = new TestApp();
+            new Table(app.stack, 'MyTable')
+                .addPartitionKey('partitionKey', KeyAttributeType.Binary)
+                .addSortKey('sortKey', KeyAttributeType.Number);
+            const template = app.synthesizeTemplate();
+
+            test.deepEqual(template, {
+                Resources: {
+                    MyTable794EDED1: {
+                        Type: 'AWS::DynamoDB::Table',
+                        Properties: {
+                            AttributeDefinitions: [
+                                { AttributeName: 'partitionKey', AttributeType: 'B' },
+                                { AttributeName: 'sortKey', AttributeType: 'N' }
+                            ],
+                            KeySchema: [
+                                { AttributeName: 'partitionKey', KeyType: 'HASH' },
+                                { AttributeName: 'sortKey', KeyType: 'RANGE' }
+                            ],
+                            ProvisionedThroughput: { ReadCapacityUnits: 5, WriteCapacityUnits: 5 },
+                        }
+                    }
+                }
+            });
+
+            test.done();
+        },
         'server-side encryption is not enabled'(test: Test) {
             const app = new TestApp();
             new Table(app.stack, 'MyTable')
@@ -257,6 +285,7 @@ export = {
             tableName: 'MyTable',
             readCapacity: 42,
             writeCapacity: 1337,
+            pitrEnabled: true,
             sseEnabled: true,
             streamSpecification: StreamViewType.KeysOnly,
             ttlAttributeName: 'timeToLive'
@@ -282,6 +311,7 @@ export = {
                             ReadCapacityUnits: 42,
                             WriteCapacityUnits: 1337
                         },
+                        PointInTimeRecoverySpecification: { PointInTimeRecoveryEnabled: true },
                         SSESpecification: { SSEEnabled: true },
                         StreamSpecification: { StreamViewType: 'KEYS_ONLY' },
                         TableName: 'MyTable',