improvement

enjin · Apr 24, 2023 · 3f6f068 · 3f6f068
1 parent 16186b7
commit 3f6f068
Showing 1 changed file with 23 additions and 17 deletions.
diff --git a/src/handleJobs.ts b/src/handleJobs.ts
@@ -155,23 +155,29 @@ traitsQueue.process(async (job, done) => {
     console.log(
         `Saving TraitToken ${traitTokensToSave.length} and deleting ${traitTokensToDelete.length} in collection ${collectionId}`
     )
-    await em
-        .createQueryBuilder()
-        .insert()
-        .into(TraitToken)
-        .values(traitTokensToSave as any)
-        .orIgnore()
-        .execute()
-    await em.remove(traitTokensToDelete)
-
-    await em
-        .createQueryBuilder()
-        .delete()
-        .from(TraitToken)
-        .where('traitToken.trait IN (:...traitsToDelete)', { traitsToDelete: traitsToDelete.map((t) => t.id) })
-        .execute()
-
-    await em.remove(traitsToDelete)
+    if (traitTokensToSave.length) {
+        await em
+            .createQueryBuilder()
+            .insert()
+            .into(TraitToken)
+            .values(traitTokensToSave as any)
+            .orIgnore()
+            .execute()
+    }
+    if (traitTokensToDelete.length) {
+        await em.remove(traitTokensToDelete)
+    }
+
+    if (traitsToDelete.length) {
+        await em
+            .createQueryBuilder()
+            .delete()
+            .from(TraitToken)
+            .where('trait_id IN (:...traitsToDelete)', { traitsToDelete: traitsToDelete.map((t) => t.id) })
+            .execute()
+
+        await em.remove(traitsToDelete)
+    }
 
     done(null, { timeElapsed: new Date().getTime() - start.getTime(), collectionId })
 })