poll-answers data-plane: add pollAnswerCount, pollAnswers to post_agg…

… table
bluesky-social · Dec 15, 2024 · 55baf04 · 55baf04
1 parent 0db3150
commit 55baf04
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/packages/bsky/src/data-plane/server/db/migrations/20241215T172532040Z-poll-answers.ts b/packages/bsky/src/data-plane/server/db/migrations/20241215T172532040Z-poll-answers.ts
@@ -20,8 +20,29 @@ export async function up(db: Kysely<unknown>): Promise<void> {
     // Aids in index uniqueness plus answer counting
     .addUniqueConstraint('poll_answer_unique_subject_answer', ['subject', 'answer', 'creator'])
     .execute()
+
+  await db.schema
+    .alterTable('post_agg')
+    .addColumn('pollAnswerCount', 'bigint', (col) => col.notNull().defaultTo(0))
+    .execute()
+
+  await db.schema
+    .alterTable('post_agg')
+    .addColumn('pollAnswers', 'json', (col) => col.notNull().defaultTo(sql`${JSON.stringify([])}`))
+    .execute()
+
 }
 
 export async function down(db: Kysely<unknown>): Promise<void> {
   await db.schema.dropTable('poll_answer').execute()
+
+  await db.schema
+    .alterTable('post_agg')
+    .dropColumn('pollAnswerCount')
+    .execute()
+
+  await db.schema
+    .alterTable('post_agg')
+    .dropColumn('pollAnswers')
+    .execute()
 }
diff --git a/packages/bsky/src/data-plane/server/db/tables/post-agg.ts b/packages/bsky/src/data-plane/server/db/tables/post-agg.ts
@@ -8,6 +8,8 @@ export interface PostAgg {
   replyCount: Generated<number>
   repostCount: Generated<number>
   quoteCount: Generated<number>
+  pollAnswerCount: Generated<number>
+  pollAnswers: Generated<number[]>
 }
 
 export type PartialDB = {