Added a batch delete operation to DataSync

digitallyinduced · Feb 4, 2022 · 87dbe2c · 87dbe2c
1 parent 7752e0c
commit 87dbe2c
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 0 deletions.
diff --git a/IHP/DataSync/Controller.hs b/IHP/DataSync/Controller.hs
@@ -247,6 +247,13 @@ instance (
                 sqlExecWithRLS "DELETE FROM ? WHERE id = ?" (PG.Identifier table, id)
 
                 sendJSON DidDeleteRecord { requestId }
+            
+            handleMessage DeleteRecordsMessage { table, ids, requestId } = do
+                ensureRLSEnabled table
+
+                sqlExecWithRLS "DELETE FROM ? WHERE id IN ?" (PG.Identifier table, PG.In ids)
+
+                sendJSON DidDeleteRecords { requestId }
 
 
         forever do

diff --git a/IHP/DataSync/Types.hs b/IHP/DataSync/Types.hs
@@ -16,6 +16,7 @@ data DataSyncMessage
     | UpdateRecordMessage { table :: !Text, id :: !UUID, patch :: !(HashMap Text Value), requestId :: !Int }
     | UpdateRecordsMessage { table :: !Text, ids :: ![UUID], patch :: !(HashMap Text Value), requestId :: !Int }
     | DeleteRecordMessage { table :: !Text, id :: !UUID, requestId :: !Int }
+    | DeleteRecordsMessage { table :: !Text, ids :: ![UUID], requestId :: !Int }
     deriving (Eq, Show)
 
 data DataSyncResponse
@@ -32,6 +33,7 @@ data DataSyncResponse
     | DidUpdateRecord { requestId :: !Int, record :: ![Field] } -- ^ Response to 'UpdateRecordMessage'
     | DidUpdateRecords { requestId :: !Int, records :: ![[Field]] } -- ^ Response to 'UpdateRecordsMessage'
     | DidDeleteRecord { requestId :: !Int }
+    | DidDeleteRecords { requestId :: !Int }
 
 data Subscription = Subscription { id :: !UUID, channelSubscription :: !PGListener.Subscription }
 

diff --git a/lib/IHP/DataSync/ihp-datasync.js b/lib/IHP/DataSync/ihp-datasync.js
@@ -417,6 +417,25 @@ export async function deleteRecord(table, id) {
     }
 }
 
+export async function deleteRecords(table, ids) {
+    if (typeof table !== "string") {
+        throw new Error(`Table name needs to be a string, you passed ${JSON.stringify(table)} in a call to deleteRecords(${JSON.stringify(table)}, ${JSON.stringify(ids)})`);
+    }
+    if (!Array.isArray(ids)) {
+        throw new Error(`IDs needs to be an array, you passed ${JSON.stringify(ids)} in a call to deleteRecords(${JSON.stringify(table)}, ${JSON.stringify(ids)})`);
+    }
+
+    const request = { tag: 'DeleteRecordsMessage', table, ids };
+
+    try {
+        const response = await DataSyncController.getInstance().sendMessage(request);
+
+        return;
+    } catch (e) {
+        throw new Error(e.message);
+    }
+}
+
 export async function createRecords(table, records) {
     if (typeof table !== "string") {
         throw new Error(`Table name needs to be a string, you passed ${JSON.stringify(table)} in a call to createRecords(${JSON.stringify(table)}, ${JSON.stringify(records, null, 4)})`);