fix: cleanup existed questions in test only when test have questions

zolplay-labs · Oct 21, 2023 · 3fb2685 · 3fb2685
1 parent d1afd0b
commit 3fb2685
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/app/_game/helpers/getQuestions.ts b/app/_game/helpers/getQuestions.ts
@@ -1,4 +1,4 @@
-import { desc, eq, inArray } from 'drizzle-orm'
+import { desc, eq, inArray, sql } from 'drizzle-orm'
 
 import { db } from '~/db'
 import { fetchQuestionCount } from '~/db/queries'
@@ -73,7 +73,15 @@ const generateRandomQuestions = async ({
     .insert(tests)
     .values({ id: testId })
     .onDuplicateKeyUpdate({ set: { id: testId } })
-  await db.delete(questionsToTests).where(eq(questionsToTests.testId, testId))
+  const testQuestionCount = (
+    await db
+      .select({ count: sql<number>`count(*)` })
+      .from(questionsToTests)
+      .where(eq(questionsToTests.testId, testId))
+  )[0]?.count
+  if (testQuestionCount && testQuestionCount > 0) {
+    await db.delete(questionsToTests).where(eq(questionsToTests.testId, testId))
+  }
   await db
     .insert(questionsToTests)
     .values(randomIds.map((questionId) => ({ questionId, testId })))