wehub-resource-sync
40 lines
1.7 KiB
TypeScript
40 lines
1.7 KiB
TypeScript
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;
|
|
}
|
|
}
|