feat: add dynamodb bulk delete operations

packages/data-migration/src/DriverTypes/NoSQL.ts

@@ -2,7 +2,9 @@ import Driver from "./Driver";
 import { Observable } from "rxjs";
 
 export default interface NoSqlDriver<P, R> extends Driver<P, R> {
-  getAllRecords: <T>() => Observable<T>;
-  putRecord: <T>(record: T) => Promise<T>;
-  putRecordsBulk: <T>(records: Array<T>) => Promise<T>;
+  getAllRecords<T>(): Observable<T>;
+  putRecord<T>(record: T): Promise<T>;
+  deleteRecord<T>(key: T): Promise<any>;
+  putRecordsBulk<T>(records: Array<T>): Promise<T>;
+  deleteRecordsBulk<T>(keys: Array<T>): Promise<any>;
 }

plugins/drivers/dynamodb/src/index.ts

@@ -107,6 +107,39 @@ const dynamoDbDriver: DriverBuilder<DynamoDbParameters, DynamoDbDriver> = (
 
       return records;
     },
+
+    async deleteRecord<T>(key: T) {
+      const result = await DocumentDb.delete({
+        TableName,
+        Key: key,
+      }).promise();
+
+      return result;
+    },
+
+    async deleteRecordsBulk<T>(keys: Array<T>) {
+      const writeItems = keys.map(
+        (Key) =>
+          ({
+            DeleteRequest: { Key },
+          } as AWS.DynamoDB.DocumentClient.WriteRequest)
+      );
+
+      for (let index = 0; index < writeItems.length; index += 25) {
+        const deleteRequest = {
+          RequestItems: {},
+        } as AWS.DynamoDB.DocumentClient.BatchWriteItemInput;
+
+        deleteRequest.RequestItems[TableName] = writeItems.slice(
+          index,
+          Math.min(index + 25, writeItems.length)
+        );
+
+        logger(`Deleting with request ${JSON.stringify(deleteRequest)}`);
+
+        await DocumentDb.batchWrite(deleteRequest).promise();
+      }
+    },
   } as DynamoDbDriver;
 };