chore(rethinkdb): TimelineEvent: Phase 3 (#9876)

Signed-off-by: Matt Krick <[email protected]>

packages/server/database/rethinkDriver.ts

@@ -36,7 +36,6 @@ import SuggestedActionTryTheDemo from './types/SuggestedActionTryTheDemo'
 import Task from './types/Task'
 import TemplateDimension from './types/TemplateDimension'
 import TemplateScale from './types/TemplateScale'
-import TimelineEvent from './types/TimelineEvent'
 
 export type RethinkSchema = {
   AgendaItem: {
@@ -187,10 +186,6 @@ export type RethinkSchema = {
     type: TemplateScale
     index: 'teamId'
   }
-  TimelineEvent: {
-    type: TimelineEvent
-    index: 'userIdCreatedAt' | 'meetingId'
-  }
 }
 
 export type DBType = {

packages/server/dataloader/foreignKeyLoaderMakers.ts

@@ -60,3 +60,17 @@ export const retroReflectionsByGroupId = foreignKeyLoaderMaker(
       .execute()
   }
 )
+
+export const timelineEventsByMeetingId = foreignKeyLoaderMaker(
+  'timelineEvents',
+  'meetingId',
+  async (meetingIds) => {
+    const pg = getKysely()
+    return pg
+      .selectFrom('TimelineEvent')
+      .selectAll()
+      .where('meetingId', 'in', meetingIds)
+      .where('isActive', '=', true)
+      .execute()
+  }
+)

packages/server/dataloader/primaryKeyLoaderMakers.ts

@@ -53,3 +53,7 @@ export const selectRetroReflections = () =>
 export const retroReflections = primaryKeyLoaderMaker((ids: readonly string[]) => {
   return selectRetroReflections().where('id', 'in', ids).execute()
 })
+
+export const timelineEvents = primaryKeyLoaderMaker((ids: readonly string[]) => {
+  return getKysely().selectFrom('TimelineEvent').selectAll().where('id', 'in', ids).execute()
+})

packages/server/dataloader/rethinkForeignKeyLoaderMakers.ts

@@ -1,5 +1,3 @@
-import TimelineEventCheckinComplete from 'parabol-server/database/types/TimelineEventCheckinComplete'
-import TimelineEventRetroComplete from 'parabol-server/database/types/TimelineEventRetroComplete'
 import getRethink from '../database/rethinkDriver'
 import {RDatum} from '../database/stricterR'
 import RethinkForeignKeyLoaderMaker from './RethinkForeignKeyLoaderMaker'
@@ -182,18 +180,6 @@ export const templateDimensionsByTemplateId = new RethinkForeignKeyLoaderMaker(
     )
   }
 )
-export const timelineEventsByMeetingId = new RethinkForeignKeyLoaderMaker(
-  'timelineEvents',
-  'meetingId',
-  async (meetingIds) => {
-    const r = await getRethink()
-    return r
-      .table('TimelineEvent')
-      .getAll(r.args(meetingIds), {index: 'meetingId'})
-      .filter({isActive: true})
-      .run() as Promise<TimelineEventCheckinComplete[] | TimelineEventRetroComplete[]>
-  }
-)
 
 export const slackAuthByUserId = new RethinkForeignKeyLoaderMaker(
   'slackAuths',

packages/server/dataloader/rethinkPrimaryKeyLoaderMakers.ts

@@ -23,4 +23,3 @@ export const tasks = new RethinkPrimaryKeyLoaderMaker('Task')
 export const teamMembers = new RethinkPrimaryKeyLoaderMaker('TeamMember')
 export const teamInvitations = new RethinkPrimaryKeyLoaderMaker('TeamInvitation')
 export const templateDimensions = new RethinkPrimaryKeyLoaderMaker('TemplateDimension')
-export const timelineEvents = new RethinkPrimaryKeyLoaderMaker('TimelineEvent')

packages/server/graphql/mutations/archiveTimelineEvent.ts

@@ -2,7 +2,6 @@ import {GraphQLID, GraphQLNonNull} from 'graphql'
 import {SubscriptionChannel} from 'parabol-client/types/constEnums'
 import TimelineEventCheckinComplete from 'parabol-server/database/types/TimelineEventCheckinComplete'
 import TimelineEventRetroComplete from 'parabol-server/database/types/TimelineEventRetroComplete'
-import getRethink from '../../database/rethinkDriver'
 import {TimelineEventEnum} from '../../database/types/TimelineEvent'
 import getKysely from '../../postgres/getKysely'
 import {getUserId, isTeamMember} from '../../utils/authorization'
@@ -25,7 +24,6 @@ const archiveTimelineEvent = {
     {timelineEventId}: {timelineEventId: string},
     {authToken, dataLoader, socketId: mutatorId}: GQLContext
   ) => {
-    const r = await getRethink()
     const operationId = dataLoader.share()
     const subOptions = {mutatorId, operationId}
     const viewerId = getUserId(authToken)
@@ -59,14 +57,11 @@ const archiveTimelineEvent = {
         .load(meetingId)
       const eventIds = meetingTimelineEvents.map(({id}) => id)
       const pg = getKysely()
-      await Promise.all([
-        pg
-          .updateTable('TimelineEvent')
-          .set({isActive: false})
-          .where('id', 'in', eventIds)
-          .execute(),
-        r.table('TimelineEvent').getAll(r.args(eventIds)).update({isActive: false}).run()
-      ])
+      await pg
+        .updateTable('TimelineEvent')
+        .set({isActive: false})
+        .where('id', 'in', eventIds)
+        .execute()
       meetingTimelineEvents.map((event) => {
         const {id: timelineEventId, userId} = event
         publish(

packages/server/graphql/mutations/endCheckIn.ts

@@ -246,10 +246,7 @@ export default {
     )
     const timelineEventId = events[0]!.id
     const pg = getKysely()
-    await Promise.all([
-      pg.insertInto('TimelineEvent').values(events).execute(),
-      r.table('TimelineEvent').insert(events).run()
-    ])
+    await pg.insertInto('TimelineEvent').values(events).execute()
     if (team.isOnboardTeam) {
       const teamLeadUserId = await r
         .table('TeamMember')

packages/server/graphql/mutations/endSprintPoker.ts

@@ -129,10 +129,7 @@ export default {
         })
     )
     const pg = getKysely()
-    await Promise.all([
-      pg.insertInto('TimelineEvent').values(events).execute(),
-      r.table('TimelineEvent').insert(events).run()
-    ])
+    await pg.insertInto('TimelineEvent').values(events).execute()
 
     const data = {
       meetingId,

packages/server/graphql/mutations/helpers/bootstrapNewUser.ts

@@ -71,8 +71,7 @@ const bootstrapNewUser = async (
       )
       .insertInto('TimelineEvent')
       .values(joinEvent)
-      .execute(),
-    r.table('TimelineEvent').insert(joinEvent).run()
+      .execute()
   ])
 
   // Identify the user so user properties are set before any events are sent

packages/server/graphql/mutations/helpers/createTeamAndLeader.ts

@@ -47,7 +47,6 @@ export default async function createTeamAndLeader(user: IUser, newTeam: ValidNew
     r.table('MeetingSettings').insert(meetingSettings).run(),
     // denormalize common fields to team member
     insertNewTeamMember(user, teamId),
-    r.table('TimelineEvent').insert(timelineEvent).run(),
     addTeamIdToTMS(userId, teamId)
   ])
 }

packages/server/graphql/mutations/helpers/safeEndRetrospective.ts

@@ -165,10 +165,7 @@ const safeEndRetrospective = async ({
   )
   const timelineEventId = events[0]!.id
   const pg = getKysely()
-  await Promise.all([
-    pg.insertInto('TimelineEvent').values(events).execute(),
-    r.table('TimelineEvent').insert(events).run()
-  ])
+  await pg.insertInto('TimelineEvent').values(events).execute()
 
   if (team.isOnboardTeam) {
     const teamLeadUserId = await r

packages/server/graphql/mutations/helpers/safeEndTeamPrompt.ts

@@ -104,10 +104,7 @@ const safeEndTeamPrompt = async ({
   )
   const timelineEventId = events[0]!.id
   const pg = getKysely()
-  await Promise.all([
-    pg.insertInto('TimelineEvent').values(events).execute(),
-    r.table('TimelineEvent').insert(events).run()
-  ])
+  await pg.insertInto('TimelineEvent').values(events).execute()
   summarizeTeamPrompt(meeting, context)
   analytics.teamPromptEnd(completedTeamPrompt, meetingMembers, responses, dataLoader)
   checkTeamsLimit(team.orgId, dataLoader)

packages/server/graphql/private/mutations/backupOrganization.ts

@@ -256,17 +256,7 @@ const backupOrganization: MutationResolvers['backupOrganization'] = async (_sour
               r.or(row('teamId').default(null).eq(null), r(teamIds).contains(row('teamId')))
             )
             .coerceTo('array')
-            .do((items: RValue) => r.db(DESTINATION).table('SuggestedAction').insert(items)),
-          timelineEvent: (
-            r
-              .table('TimelineEvent')
-              .filter((row: RDatum) => r(userIds).contains(row('userId'))) as any
-          )
-            .filter((row: RValue) =>
-              r.branch(row('teamId'), r(teamIds).contains(row('teamId')), true)
-            )
-            .coerceTo('array')
-            .do((items: RValue) => r.db(DESTINATION).table('TimelineEvent').insert(items))
+            .do((items: RValue) => r.db(DESTINATION).table('SuggestedAction').insert(items))
         })
       }),
     meetingIds: r

packages/server/graphql/private/mutations/hardDeleteUser.ts

@@ -117,12 +117,6 @@ const hardDeleteUser: MutationResolvers['hardDeleteUser'] = async (
       .getAll(r.args(teamIds), {index: 'teamId'})
       .filter((row: RValue) => row('createdBy').eq(userIdToDelete))
       .delete(),
-    timelineEvent: r
-      .table('TimelineEvent')
-      .between([userIdToDelete, r.minval], [userIdToDelete, r.maxval], {
-        index: 'userIdCreatedAt'
-      })
-      .delete(),
     agendaItem: r
       .table('AgendaItem')
       .getAll(r.args(teamIds), {index: 'teamId'})

packages/server/graphql/types/User.ts

@@ -15,13 +15,11 @@ import {
   MAX_RESULT_GROUP_SIZE
 } from '../../../client/utils/constants'
 import groupReflections from '../../../client/utils/smartGroup/groupReflections'
-import getRethink from '../../database/rethinkDriver'
-import {RDatum} from '../../database/stricterR'
 import MeetingMemberType from '../../database/types/MeetingMember'
 import OrganizationType from '../../database/types/Organization'
 import OrganizationUserType from '../../database/types/OrganizationUser'
 import SuggestedActionType from '../../database/types/SuggestedAction'
-import TimelineEvent from '../../database/types/TimelineEvent'
+import getKysely from '../../postgres/getKysely'
 import {getUserId, isSuperUser, isTeamMember} from '../../utils/authorization'
 import getMonthlyStreak from '../../utils/getMonthlyStreak'
 import getRedis from '../../utils/getRedis'
@@ -217,7 +215,6 @@ const User: GraphQLObjectType<any, GQLContext> = new GraphQLObjectType<any, GQLC
         {after, first, teamIds, eventTypes},
         {authToken, dataLoader}: GQLContext
       ) => {
-        const r = await getRethink()
         const viewerId = getUserId(authToken)
 
         // VALIDATE
@@ -244,21 +241,23 @@ const User: GraphQLObjectType<any, GQLContext> = new GraphQLObjectType<any, GQLC
           : accessibleTeamIds
 
         if (viewerId !== id && !isSuperUser(authToken)) return null
-        const dbAfter = after ? new Date(after) : r.maxval
-        const events = await r
-          .table('TimelineEvent')
-          .between([viewerId, r.minval], [viewerId, dbAfter], {
-            index: 'userIdCreatedAt'
-          })
-          .filter({isActive: true})
-          .filter((t: RDatum<TimelineEvent>) =>
-            eventTypes ? r.expr(eventTypes).contains(t('type')) : true
-          )
-          .filter((t: RDatum) => r.expr(validTeamIds).contains(t('teamId')))
-          .orderBy(r.desc('createdAt'))
+        const dbAfter = after ? new Date(after) : new Date('3000-01-01')
+        const minVal = new Date(0)
+
+        const pg = getKysely()
+        const events = await pg
+          .selectFrom('TimelineEvent')
+          .selectAll()
+          .where('userId', '=', viewerId)
+          .where('createdAt', '>', minVal)
+          .where('createdAt', '<=', dbAfter)
+          .where('isActive', '=', true)
+          .where('teamId', 'in', validTeamIds)
+          .$if(!!eventTypes, (db) => db.where('type', 'in', eventTypes))
+          .orderBy('createdAt')
           .limit(first + 1)
-          .coerceTo('array')
-          .run()
+          .execute()
+
         const edges = events.slice(0, first).map((node) => ({
           cursor: node.createdAt,
           node