query-or.ts
No OneTemporary
Actions

Size

1 KB

Referenced Files

None

Subscribers

None

query-or.ts
View Options

	import Query from "./query";
	import QueryStage, { MatchBody } from "./query-stage";
	import QueryStep, { Lookup, Match } from "./query-step";

	export default class Or extends Query {
	lookup_steps: QueryStep[] = [];
	constructor(...queries: Query[]) {
	super();
	for (const query of queries) {
	this.addQuery(query);
	}
	}

	addQuery(query: Query): void {
	const steps = query.dump();
	this.lookup_steps.push(
	...(steps.filter((step) => step instanceof Lookup) as Lookup[])
	);
	const match_stage_bodies = steps
	.filter((step) => step instanceof Match)
	.map((match: Match) => match.body);

	if (match_stage_bodies.length) {
	const match_stage: MatchBody =
	match_stage_bodies.length > 1
	? { $and: match_stage_bodies }
	: match_stage_bodies[0];
	this.steps.push(new Match(match_stage));
	}
	}

	dump(): QueryStep[] {
	return this.lookup_steps.concat(
	new Match({ $or: this._getMatchExpressions() })
	);
	}

	toPipeline(): QueryStage[] {
	const lookups = this.lookup_steps
	.map((step) => step.toPipeline())
	.reduce((acc, cur) => acc.concat(cur), []);
	return lookups.concat({ $match: { $or: this._getMatchExpressions() } });
	}

	_getMatchExpressions(): MatchBody[] {
	return this.steps
	.filter((step) => step instanceof Match)
	.map((step) => step.body as MatchBody);
	}

	prefix(prefix: string): this {
	super.prefix(prefix);
	this.lookup_steps = this.lookup_steps.map((step) =>
	step.prefix(prefix)
	);
	return this;
	}
	}