mingo
Version:
MongoDB query language for in-memory objects
110 lines (109 loc) • 3.83 kB
JavaScript
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
});