Skip to content

Commit e62b119

Browse files
committed
Pass the default repository in, so we can try Postgres in the tests
1 parent e1c9367 commit e62b119

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

apps/webapp/app/services/runsRepository/runsRepository.server.ts

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ import { ClickHouseRunsRepository } from "./clickhouseRunsRepository.server";
1212
import { PostgresRunsRepository } from "./postgresRunsRepository.server";
1313
import { FEATURE_FLAG, makeFlags } from "~/v3/featureFlags.server";
1414
import { startActiveSpan } from "~/v3/tracer.server";
15+
import { logger } from "../logger.server";
1516

1617
export type RunsRepositoryOptions = {
1718
clickhouse: ClickHouse;
@@ -119,10 +120,16 @@ export interface IRunsRepository {
119120
export 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

Comments
 (0)