UNPKG

mingo

Version:

MongoDB query language for in-memory objects

110 lines (109 loc) 3.83 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 internal_exports = {}; __export(internal_exports, { QueryImpl: () => QueryImpl }); module.exports = __toCommonJS(internal_exports); var import_core = require("../core"); var import_cursor = require("../cursor"); var import_util = require("../util"); const TOP_LEVEL_RE = /^\$(and|or|nor|expr|jsonSchema)$/; class QueryImpl { #compiled; #options; #condition; constructor(condition, options) { this.#condition = (0, import_util.cloneDeep)(condition); this.#options = options; 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 = {}; for (const [field, expr] of Object.entries(this.#condition)) { 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 = (0, import_core.getOperator)( import_core.OpType.QUERY, operator, this.#options ); (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 = { QueryImpl });