MAX_PARTNER_LINKS_FOR_LOCAL_FILTERING

Author: steven-tey

apps/web/app/(ee)/api/partner-profile/programs/[programId]/analytics/route.ts

@@ -2,6 +2,7 @@ import { getAnalytics } from "@/lib/analytics/get-analytics";
 import { DubApiError } from "@/lib/api/errors";
 import { getProgramEnrollmentOrThrow } from "@/lib/api/programs/get-program-enrollment-or-throw";
 import { withPartnerProfile } from "@/lib/auth/partner";
+import { MAX_PARTNER_LINKS_FOR_LOCAL_FILTERING } from "@/lib/constants/partner-profile";
 import {
   LARGE_PROGRAM_IDS,
   LARGE_PROGRAM_MIN_TOTAL_COMMISSIONS_CENTS,
@@ -52,7 +53,11 @@ export const GET = withPartnerProfile(
         ? { event: rest.event, groupBy: "count", interval: "all" }
         : rest),
       workspaceId: program.workspaceId,
-      ...(linkId ? { linkId } : { linkIds: links.map((link) => link.id) }),
+      ...(linkId
+        ? { linkId }
+        : links.length > MAX_PARTNER_LINKS_FOR_LOCAL_FILTERING
+          ? { partnerId: partner.id }
+          : { linkIds: links.map((link) => link.id) }),
       dataAvailableFrom: program.startedAt ?? program.createdAt,
     });
 

apps/web/app/(ee)/api/partner-profile/programs/[programId]/events/route.ts

@@ -2,6 +2,7 @@ import { getEvents } from "@/lib/analytics/get-events";
 import { DubApiError } from "@/lib/api/errors";
 import { getProgramEnrollmentOrThrow } from "@/lib/api/programs/get-program-enrollment-or-throw";
 import { withPartnerProfile } from "@/lib/auth/partner";
+import { MAX_PARTNER_LINKS_FOR_LOCAL_FILTERING } from "@/lib/constants/partner-profile";
 import {
   LARGE_PROGRAM_IDS,
   LARGE_PROGRAM_MIN_TOTAL_COMMISSIONS_CENTS,
@@ -68,7 +69,11 @@ export const GET = withPartnerProfile(
     const events = await getEvents({
       ...rest,
       workspaceId: program.workspaceId,
-      ...(linkId ? { linkId } : { linkIds: links.map((link) => link.id) }),
+      ...(linkId
+        ? { linkId }
+        : links.length > MAX_PARTNER_LINKS_FOR_LOCAL_FILTERING
+          ? { partnerId: partner.id }
+          : { linkIds: links.map((link) => link.id) }),
       dataAvailableFrom: program.startedAt ?? program.createdAt,
     });
 

apps/web/lib/constants/partner-profile.ts

@@ -1,2 +1,3 @@
 export const MAX_INVITES_PER_REQUEST = 5;
 export const MAX_PARTNER_USERS = 10;
+export const MAX_PARTNER_LINKS_FOR_LOCAL_FILTERING = 100; // if over 100 links we should filter on TB directly