feat: add public flags to opportunity (batchSize) (#3346)

Author: rebelchris

package.json

@@ -36,7 +36,7 @@
     "@connectrpc/connect-fastify": "^1.6.1",
     "@connectrpc/connect-node": "^1.6.1",
     "@dailydotdev/graphql-redis-subscriptions": "^2.4.3",
-    "@dailydotdev/schema": "0.2.55",
+    "@dailydotdev/schema": "0.2.58",
     "@dailydotdev/ts-ioredis-pool": "^1.0.2",
     "@fastify/cookie": "^11.0.2",
     "@fastify/cors": "^11.1.0",

pnpm-lock.yaml

@@ -28,8 +28,11 @@ export type OpportunityFlags = Partial<{
     userIds: string[];
     totalCount: number;
   };
+  batchSize: number;
 }>;
 
+export type OpportunityFlagsPublic = Pick<OpportunityFlags, 'batchSize'>;
+
 @Entity()
 @TableInheritance({ column: { type: 'text', name: 'type' } })
 export class Opportunity {

src/entity/opportunities/Opportunity.ts

@@ -42,7 +42,12 @@ import {
 import { GQLComment } from '../schema/comments';
 import { GQLUserPost } from '../schema/posts';
 import { UserComment } from '../entity/user/UserComment';
-import { type ContentLanguage, type I18nRecord, UserVote } from '../types';
+import {
+  type ContentLanguage,
+  type I18nRecord,
+  opportunityMatchBatchSize,
+  UserVote,
+} from '../types';
 import { whereVordrFilter } from '../common/vordr';
 import { UserCompany, Post } from '../entity';
 import {
@@ -66,7 +71,10 @@ import { OrganizationLinkType } from '../common/schema/organizations';
 import type { GCSBlob } from '../common/schema/userCandidate';
 import { QuestionType } from '../entity/questions/types';
 import { snotraClient } from '../integrations/snotra';
-import type { Opportunity } from '../entity/opportunities/Opportunity';
+import type {
+  Opportunity,
+  OpportunityFlagsPublic,
+} from '../entity/opportunities/Opportunity';
 import { SubscriptionStatus } from '../common/plus';
 
 const existsByUserAndPost =
@@ -1586,6 +1594,14 @@ const obj = new GraphORM({
           return value?.status || SubscriptionStatus.None;
         },
       },
+      flags: {
+        jsonType: true,
+        transform: (value: OpportunityFlagsPublic): OpportunityFlagsPublic => {
+          return {
+            batchSize: value?.batchSize ?? opportunityMatchBatchSize,
+          };
+        },
+      },
     },
   },
   OpportunityScreeningQuestion: {

src/graphorm/index.ts

@@ -96,7 +96,10 @@ import { ConnectionArguments } from 'graphql-relay';
 import { ProfileResponse, snotraClient } from '../integrations/snotra';
 import { slackClient } from '../common/slack';
 import { fileTypeFromBuffer } from 'file-type';
-import { acceptedOpportunityFileTypes } from '../types';
+import {
+  acceptedOpportunityFileTypes,
+  opportunityMatchBatchSize,
+} from '../types';
 import { getBrokkrClient } from '../common/brokkr';
 import { garmScraperService } from '../common/scraper';
 import { Storage } from '@google-cloud/storage';
@@ -110,7 +113,14 @@ import type { GQLSource } from './sources';
 export interface GQLOpportunity
   extends Pick<
     Opportunity,
-    'id' | 'type' | 'state' | 'title' | 'tldr' | 'content' | 'keywords'
+    | 'id'
+    | 'type'
+    | 'state'
+    | 'title'
+    | 'tldr'
+    | 'content'
+    | 'keywords'
+    | 'flags'
   > {
   createdAt: Date;
   updatedAt: Date;
@@ -273,6 +283,13 @@ export const typeDefs = /* GraphQL */ `
     edges: [OpportunityEdge!]!
   }
 
+  """
+  Flags for the opportunity
+  """
+  type OpportunityFlagsPublic {
+    batchSize: Int
+  }
+
   type Opportunity {
     id: ID!
     type: ProtoEnumValue!
@@ -288,6 +305,7 @@ export const typeDefs = /* GraphQL */ `
     questions: [OpportunityScreeningQuestion]!
     feedbackQuestions: [OpportunityFeedbackQuestion]!
     subscriptionStatus: SubscriptionStatus!
+    flags: OpportunityFlagsPublic
   }
 
   type OpportunityMatchDescription {
@@ -2349,6 +2367,8 @@ export const resolvers: IResolvers<unknown, BaseContext> = traceResolvers<
               flags.anonUserId = ctx.trackingId; // save tracking id to attribute later
             }
 
+            flags.batchSize = opportunityMatchBatchSize;
+
             const opportunity = await entityManager
               .getRepository(OpportunityJob)
               .save(

src/schema/opportunity.ts

@@ -317,3 +317,5 @@ export type TLocation = {
 export const acceptedOpportunityFileTypes = acceptedResumeFileTypes;
 
 export const acceptedOpportunityExtensions = acceptedResumeExtensions;
+
+export const opportunityMatchBatchSize = 50;