Fix db query to restore previous doc version (vercel#498)

kobe-helicopter-laoda · Nov 6, 2024 · c5f6ac8 · c5f6ac8
1 parent d948b99
commit c5f6ac8
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git a/db/queries.ts b/db/queries.ts
@@ -11,6 +11,7 @@ import {
   User,
   document,
   Suggestion,
+  suggestion,
   Message,
   message,
   vote,
@@ -227,6 +228,15 @@ export async function deleteDocumentsByIdAfterTimestamp({
   timestamp: Date;
 }) {
   try {
+    await db
+      .delete(suggestion)
+      .where(
+        and(
+          eq(suggestion.documentId, id),
+          gt(suggestion.documentCreatedAt, timestamp)
+        )
+      );
+
     return await db
       .delete(document)
       .where(and(eq(document.id, id), gt(document.createdAt, timestamp)));
@@ -244,7 +254,7 @@ export async function saveSuggestions({
   suggestions: Array<Suggestion>;
 }) {
   try {
-    return await db.insert(Suggestion).values(suggestions);
+    return await db.insert(suggestion).values(suggestions);
   } catch (error) {
     console.error('Failed to save suggestions in database');
     throw error;
@@ -259,8 +269,8 @@ export async function getSuggestionsByDocumentId({
   try {
     return await db
       .select()
-      .from(Suggestion)
-      .where(and(eq(Suggestion.documentId, documentId)));
+      .from(suggestion)
+      .where(and(eq(suggestion.documentId, documentId)));
   } catch (error) {
     console.error(
       'Failed to get suggestions by document version from database'

diff --git a/db/schema.ts b/db/schema.ts
@@ -82,7 +82,7 @@ export const document = pgTable(
 
 export type Document = InferSelectModel<typeof document>;
 
-export const Suggestion = pgTable(
+export const suggestion = pgTable(
   'Suggestion',
   {
     id: uuid('id').notNull().defaultRandom(),
@@ -106,4 +106,4 @@ export const Suggestion = pgTable(
   })
 );
 
-export type Suggestion = InferSelectModel<typeof Suggestion>;
+export type Suggestion = InferSelectModel<typeof suggestion>;