UNPKG

mingo

Version:

MongoDB query language for in-memory objects

119 lines (118 loc) 4.27 kB
var __defProp = Object.defineProperty; var __getOwnPropDesc = Object.getOwnPropertyDescriptor; var __getOwnPropNames = Object.getOwnPropertyNames; var __hasOwnProp = Object.prototype.hasOwnProperty; var __export = (target, all) => { for (var name in all) __defProp(target, name, { get: all[name], enumerable: true }); }; var __copyProps = (to, from, except, desc) => { if (from && typeof from === "object" || typeof from === "function") { for (let key of __getOwnPropNames(from)) if (!__hasOwnProp.call(to, key) && key !== except) __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable }); } return to; }; var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod); var query_exports = {}; __export(query_exports, { Query: () => Query }); module.exports = __toCommonJS(query_exports); var import_internal = require("./core/_internal"); var import_cursor = require("./cursor"); var import_util = require("./util"); const TOP_LEVEL_RE = /^\$(and|or|nor|expr|jsonSchema)$/; class Query { #compiled; #condition; #options; /** * Creates an instance of the query with the specified condition and options. * This object is preloaded with all query and projection operators. * * @param condition - The query condition object used to define the criteria for matching documents. * @param options - Optional configuration settings to customize the query behavior. */ constructor(condition, options) { this.#condition = (0, import_util.cloneDeep)(condition); this.#options = import_internal.ComputeOptions.init(options).update({ condition }); this.#compiled = []; this.compile(); } compile() { (0, import_util.assert)( (0, import_util.isObject)(this.#condition), `query criteria must be an object: ${JSON.stringify(this.#condition)}` ); const whereOperator = {}; const conditions = Object.entries(this.#condition); for (const [field, expr] of conditions) { if ("$where" === field) { (0, import_util.assert)( this.#options.scriptEnabled, "$where operator requires 'scriptEnabled' option to be true." ); Object.assign(whereOperator, { field, expr }); } else if (TOP_LEVEL_RE.test(field)) { this.processOperator(field, field, expr); } else { (0, import_util.assert)(!(0, import_util.isOperator)(field), `unknown top level operator: ${field}`); for (const [operator, val] of Object.entries( (0, import_util.normalize)(expr) )) { this.processOperator(field, operator, val); } } if (whereOperator.field) { this.processOperator( whereOperator.field, whereOperator.field, whereOperator.expr ); } } } processOperator(field, operator, value) { const call = this.#options.context.getOperator( import_internal.OpType.QUERY, operator ); (0, import_util.assert)(!!call, `unknown query operator ${operator}`); this.#compiled.push(call(field, value, this.#options)); } /** * Tests whether the given object satisfies all compiled predicates. * * @template T - The type of the object to test. * @param obj - The object to be tested against the compiled predicates. * @returns `true` if the object satisfies all predicates, otherwise `false`. */ test(obj) { return this.#compiled.every((p) => p(obj)); } /** * Returns a cursor for iterating over the items in the given collection that match the query criteria. * * @typeParam T - The type of the items in the resulting cursor. * @param collection - The source collection to search through. * @param projection - An optional object specifying fields to include or exclude * in the returned items. * @returns A `Cursor` instance for iterating over the matching items. */ find(collection, projection) { return new import_cursor.Cursor( collection, (o) => this.test(o), projection || {}, this.#options ); } } // Annotate the CommonJS export names for ESM import in node: 0 && (module.exports = { Query });