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(
38
38
return { repository : eventRepository , store : getTaskEventStore ( ) } ;
39
39
}
40
40
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
+
41
59
async function resolveTaskEventRepositoryFlag (
42
60
featureFlags : Record < string , unknown > | undefined
43
61
) : Promise < "clickhouse" | "postgres" > {
Original file line number Diff line number Diff line change @@ -22,7 +22,7 @@ import { parseDelay } from "~/utils/delays";
22
22
import { resolveIdempotencyKeyTTL } from "~/utils/idempotencyKeys.server" ;
23
23
import { handleMetadataPacket } from "~/utils/packets" ;
24
24
import { marqs } from "~/v3/marqs/index.server" ;
25
- import { getEventRepository } from "../eventRepository/index.server" ;
25
+ import { getV3EventRepository } from "../eventRepository/index.server" ;
26
26
import { generateFriendlyId } from "../friendlyIdentifiers" ;
27
27
import { findCurrentWorkerFromEnvironment } from "../models/workerDeployment.server" ;
28
28
import { guardQueueSizeLimitsForEnv } from "../queueSizeLimits.server" ;
@@ -291,8 +291,7 @@ export class TriggerTaskServiceV1 extends BaseService {
291
291
} )
292
292
: undefined ;
293
293
294
- const { repository, store } = await getEventRepository (
295
- environment . organization . featureFlags as Record < string , unknown > ,
294
+ const { repository, store } = await getV3EventRepository (
296
295
dependentAttempt ?. taskRun . taskEventStore ??
297
296
parentAttempt ?. taskRun . taskEventStore ??
298
297
dependentBatchRun ?. dependentTaskAttempt ?. taskRun . taskEventStore
You can’t perform that action at this time.
0 commit comments