chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,39 @@
|
||||
import { type ClickHouse } from "@internal/clickhouse";
|
||||
import { type PrismaClientOrTransaction } from "~/db.server";
|
||||
import { RunsRepository } from "~/services/runsRepository/runsRepository.server";
|
||||
import { type RunListFilter, type RunListResolver } from "./runReader.server";
|
||||
|
||||
export type ClickHouseRunListResolverOptions = {
|
||||
/** Resolves the per-organization ClickHouse client (multi-tenant routing). */
|
||||
getClickhouse: (organizationId: string) => Promise<ClickHouse>;
|
||||
prisma: PrismaClientOrTransaction;
|
||||
};
|
||||
|
||||
/**
|
||||
* Resolves the realtime tag/list filter into matching run ids via ClickHouse `listRunIds` (filter-only;
|
||||
* rows hydrated from Postgres by id afterward). Tag matching is contains-ALL, byte-matching Electric's
|
||||
* `runTags @> ARRAY[...]` shape.
|
||||
*/
|
||||
export class ClickHouseRunListResolver implements RunListResolver {
|
||||
constructor(private readonly options: ClickHouseRunListResolverOptions) {}
|
||||
|
||||
async resolveMatchingRunIds(filter: RunListFilter): Promise<string[]> {
|
||||
const clickhouse = await this.options.getClickhouse(filter.organizationId);
|
||||
const repository = new RunsRepository({ clickhouse, prisma: this.options.prisma });
|
||||
|
||||
const { runIds } = await repository.listRunIds({
|
||||
organizationId: filter.organizationId,
|
||||
projectId: filter.projectId,
|
||||
environmentId: filter.environmentId,
|
||||
tags: filter.tags && filter.tags.length > 0 ? filter.tags : undefined,
|
||||
// Contains-ALL, matching the Electric shape's `runTags @> ARRAY[...]` semantics.
|
||||
tagsMatch: "all",
|
||||
batchId: filter.batchId,
|
||||
from: filter.createdAtAfter?.getTime(),
|
||||
page: { size: filter.limit },
|
||||
});
|
||||
|
||||
// listRunIds is keyset-paginated; runIds is already capped to page.size (= limit).
|
||||
return runIds;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user