File tree Expand file tree Collapse file tree 2 files changed +20
-3
lines changed Expand file tree Collapse file tree 2 files changed +20
-3
lines changed Original file line number Diff line number Diff line change @@ -38,6 +38,24 @@ export async function getEventRepository(
3838 return { repository : eventRepository , store : getTaskEventStore ( ) } ;
3939}
4040
41+ export async function getV3EventRepository (
42+ parentStore : string | undefined
43+ ) : Promise < { repository : IEventRepository ; store : string } > {
44+ if ( typeof parentStore === "string" ) {
45+ if ( parentStore === "clickhouse" ) {
46+ return { repository : clickhouseEventRepository , store : "clickhouse" } ;
47+ } else {
48+ return { repository : eventRepository , store : getTaskEventStore ( ) } ;
49+ }
50+ }
51+
52+ if ( env . EVENT_REPOSITORY_DEFAULT_STORE === "clickhouse" ) {
53+ return { repository : clickhouseEventRepository , store : "clickhouse" } ;
54+ } else {
55+ return { repository : eventRepository , store : getTaskEventStore ( ) } ;
56+ }
57+ }
58+
4159async function resolveTaskEventRepositoryFlag (
4260 featureFlags : Record < string , unknown > | undefined
4361) : Promise < "clickhouse" | "postgres" > {
Original file line number Diff line number Diff line change @@ -22,7 +22,7 @@ import { parseDelay } from "~/utils/delays";
2222import { resolveIdempotencyKeyTTL } from "~/utils/idempotencyKeys.server" ;
2323import { handleMetadataPacket } from "~/utils/packets" ;
2424import { marqs } from "~/v3/marqs/index.server" ;
25- import { getEventRepository } from "../eventRepository/index.server" ;
25+ import { getV3EventRepository } from "../eventRepository/index.server" ;
2626import { generateFriendlyId } from "../friendlyIdentifiers" ;
2727import { findCurrentWorkerFromEnvironment } from "../models/workerDeployment.server" ;
2828import { guardQueueSizeLimitsForEnv } from "../queueSizeLimits.server" ;
@@ -291,8 +291,7 @@ export class TriggerTaskServiceV1 extends BaseService {
291291 } )
292292 : undefined ;
293293
294- const { repository, store } = await getEventRepository (
295- environment . organization . featureFlags as Record < string , unknown > ,
294+ const { repository, store } = await getV3EventRepository (
296295 dependentAttempt ?. taskRun . taskEventStore ??
297296 parentAttempt ?. taskRun . taskEventStore ??
298297 dependentBatchRun ?. dependentTaskAttempt ?. taskRun . taskEventStore
You can’t perform that action at this time.
0 commit comments