@@ -12,6 +12,7 @@ import { ClickHouseRunsRepository } from "./clickhouseRunsRepository.server";
1212import { PostgresRunsRepository } from "./postgresRunsRepository.server" ;
1313import { FEATURE_FLAG , makeFlags } from "~/v3/featureFlags.server" ;
1414import { startActiveSpan } from "~/v3/tracer.server" ;
15+ import { logger } from "../logger.server" ;
1516
1617export type RunsRepositoryOptions = {
1718 clickhouse : ClickHouse ;
@@ -119,10 +120,16 @@ export interface IRunsRepository {
119120export class RunsRepository implements IRunsRepository {
120121 private readonly clickHouseRunsRepository : ClickHouseRunsRepository ;
121122 private readonly postgresRunsRepository : PostgresRunsRepository ;
123+ private readonly defaultRepository : "clickhouse" | "postgres" ;
122124
123- constructor ( private readonly options : RunsRepositoryOptions ) {
125+ constructor (
126+ private readonly options : RunsRepositoryOptions & {
127+ defaultRepository ?: "clickhouse" | "postgres" ;
128+ }
129+ ) {
124130 this . clickHouseRunsRepository = new ClickHouseRunsRepository ( options ) ;
125131 this . postgresRunsRepository = new PostgresRunsRepository ( options ) ;
132+ this . defaultRepository = options . defaultRepository ?? "clickhouse" ;
126133 }
127134
128135 get name ( ) {
@@ -134,11 +141,13 @@ export class RunsRepository implements IRunsRepository {
134141 const getFlag = makeFlags ( this . options . prisma ) ;
135142 const runsListRepository = await getFlag ( {
136143 key : FEATURE_FLAG . runsListRepository ,
137- defaultValue : "clickhouse" ,
144+ defaultValue : this . defaultRepository ,
138145 } ) ;
139146
140147 span . setAttribute ( "repository.name" , runsListRepository ) ;
141148
149+ logger . log ( "runsListRepository" , { runsListRepository } ) ;
150+
142151 switch ( runsListRepository ) {
143152 case "postgres" :
144153 return this . postgresRunsRepository ;
0 commit comments