Adding skip if source collection is empty

simonedeponti · Nov 9, 2023 · 6a2fa3e · 6a2fa3e
1 parent 3ee69b2
commit 6a2fa3e
Showing 1 changed file with 13 additions and 8 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -65,19 +65,24 @@ class MongodbAnonymizer extends Command {
       }
 
       this.log("Anonymizing collection: " + collectionName);
-      this.log("Cleaning up target collection: " + collectionName);
-      await targetDb.collection(collectionName).deleteMany({});
       const sourceCollection = await db
         .collection(collectionName);
       const targetCollection = await targetDb
         .collection(collectionName);
       const list = flags.list.split(",");
-      await this.anonymizeCollection(
-        sourceCollection,
-        targetCollection,
-        collectionName,
-        list
-      );
+      if((await sourceCollection.countDocuments()) > 0) {
+        this.log("Cleaning up target collection: " + collectionName);
+        await targetCollection.deleteMany({});
+        await this.anonymizeCollection(
+          sourceCollection,
+          targetCollection,
+          collectionName,
+          list
+        );
+      }
+      else {
+        this.log(`Skipping collection ${collectionName} as it's empty`);
+      }
     }
     this.log("Done!");