v1.6.0/cleanup fix

2020-08-08 19:04:31 -04:00
parent ff07646657
commit 17a0355a4f
2 changed files with 47 additions and 33 deletions
@@ -1,32 +1,46 @@
-const admin = require('firebase-admin');
+const admin = require("firebase-admin");

-module.exports = async (roomId) => {
-  admin
+module.exports = async (partyName) => {
+  // DELETE MESSAGES
+  const partyMessagesRef = admin
    .firestore()
-    .collection('rooms')
-    .doc(roomId)
-    .collection('messages')
-    .listDocuments()
-    .then((messages) => messages.forEach((message) => message.delete()))
-    .catch((err) => {
-      console.error(`Failed to run auto clean up (delete room messages) @ ${roomId}`, err);
-    });
-  admin
+    .collection("rooms")
+    .doc(partyName)
+    .collection("messages");
+  const partyMessagesQuery = partyMessagesRef.orderBy("time").limit(200);
+  deleteCollection(partyMessagesQuery);
+  // DELETE USERS
+  const partyUsersRef = admin
    .firestore()
-    .collection('rooms')
-    .doc(roomId)
-    .collection('users')
-    .listDocuments()
-    .then((users) => users.forEach((user) => user.delete()))
-    .catch((err) => {
-      console.error(`Failed to run auto clean up (delete room users) @ ${roomId}`, err);
-    });
-  admin
-    .firestore()
-    .collection('rooms')
-    .doc(roomId)
-    .delete()
-    .catch((err) => {
-      console.error(`Failed to run auto clean up (delete room) @ ${roomId}`, err);
-    });
+    .collection("rooms")
+    .doc(partyName)
+    .collection("messages");
+  const partyUsersQuery = partyUsersRef.orderBy("time").limit(200);
+  deleteCollection(partyUsersQuery);
+  // DELETE PARTY DOC
+  admin.firestore().collection("rooms").doc(partyName).delete();
 };
+
+async function deleteCollection(query) {
+  const db = admin.firestore();
+  return new Promise((resolve, reject) => {
+    deleteQueryBatch(db, query, resolve).catch(reject);
+  });
+}
+
+async function deleteQueryBatch(db, query, resolve) {
+  const snapshot = await query.get();
+  const batchSize = snapshot.size;
+  if (batchSize === 0) {
+    resolve();
+    return;
+  }
+  const batch = admin.firestore().batch();
+  snapshot.docs.forEach((doc) => {
+    batch.delete(doc.ref);
+  });
+  await batch.commit();
+  process.nextTick(() => {
+    deleteQueryBatch(db, query, resolve);
+  });
+}