@opra/mongodb 0.33.13 → 1.0.0-alpha.10

package/esm/mongo-entity-service.js

@@ -0,0 +1,330 @@
+import * as assert from 'node:assert';
+import { InternalServerError } from '@opra/common';
+import omit from 'lodash.omit';
+import { MongoAdapter } from './mongo-adapter.js';
+import { MongoService } from './mongo-service.js';
+/**
+ * @class MongoEntityService
+ * @template T - The type of the documents in the collection.
+ */
+export class MongoEntityService extends MongoService {
+    /**
+     * Constructs a new instance
+     *
+     * @param {Type | string} dataType - The data type of the array elements.
+     * @param {MongoEntityService.Options} [options] - The options for the array service.
+     * @constructor
+     */
+    constructor(dataType, options) {
+        super(dataType, options);
+    }
+    /**
+     * Creates a new document in the MongoDB collection.
+     *
+     * @param {PartialDTO<T>} input
+     * @param {MongoEntityService.CreateOptions} options
+     * @protected
+     */
+    async _create(input, options) {
+        const inputCodec = this.getInputCodec('create');
+        const doc = inputCodec(input);
+        assert.ok(doc._id, 'You must provide the "_id" field');
+        const r = await this._dbInsertOne(doc, options);
+        if (r.insertedId) {
+            if (!options)
+                return doc;
+            const out = await this._findById(doc._id, omit(options, 'filter'));
+            if (out)
+                return out;
+        }
+        /* istanbul ignore next */
+        throw new InternalServerError(`Unknown error while creating document for "${this.getResourceName()}"`);
+    }
+    /**
+     * Returns the count of documents in the collection based on the provided options.
+     *
+     * @param {MongoEntityService.CountOptions<T>} options - The options for the count operation.
+     * @return {Promise<number>} - A promise that resolves to the count of documents in the collection.
+     */
+    async _count(options) {
+        const filter = MongoAdapter.prepareFilter(options?.filter);
+        return this._dbCountDocuments(filter, omit(options, 'filter'));
+    }
+    /**
+     * Deletes a document from the collection.
+     *
+     * @param {MongoAdapter.AnyId} id - The ID of the document to delete.
+     * @param {MongoEntityService.DeleteOptions<T>} [options] - Optional delete options.
+     * @return {Promise<number>} - A Promise that resolves to the number of documents deleted.
+     */
+    async _delete(id, options) {
+        assert.ok(id, 'You must provide an id');
+        const filter = MongoAdapter.prepareFilter([MongoAdapter.prepareKeyValues(id, ['_id']), options?.filter]);
+        const r = await this._dbDeleteOne(filter, options);
+        return r.deletedCount;
+    }
+    /**
+     * Deletes multiple documents from the collection that meet the specified filter criteria.
+     *
+     * @param {MongoEntityService.DeleteManyOptions<T>} options - The options for the delete operation.
+     * @return {Promise<number>} - A promise that resolves to the number of documents deleted.
+     */
+    async _deleteMany(options) {
+        const filter = MongoAdapter.prepareFilter(options?.filter);
+        const r = await this._dbDeleteMany(filter, omit(options, 'filter'));
+        return r.deletedCount;
+    }
+    /**
+     * The distinct command returns a list of distinct values for the given key across a collection.
+     * @param {string} field
+     * @param {MongoEntityService.DistinctOptions<T>} options
+     * @protected
+     */
+    async _distinct(field, options) {
+        const filter = MongoAdapter.prepareFilter(options?.filter);
+        return await this._dbDistinct(field, filter, omit(options, 'filter'));
+    }
+    /**
+     * Finds a document by its ID.
+     *
+     * @param {MongoAdapter.AnyId} id - The ID of the document.
+     * @param {MongoEntityService.FindOneOptions<T>} [options] - The options for the find query.
+     * @return {Promise<PartialDTO<T | undefined>>} - A promise resolving to the found document, or undefined if not found.
+     */
+    async _findById(id, options) {
+        const filter = MongoAdapter.prepareFilter([MongoAdapter.prepareKeyValues(id, ['_id']), options?.filter]);
+        const mongoOptions = {
+            ...options,
+            projection: MongoAdapter.prepareProjection(this.dataType, options?.projection),
+            limit: undefined,
+            skip: undefined,
+            sort: undefined,
+        };
+        const out = await this._dbFindOne(filter, mongoOptions);
+        const outputCodec = this.getOutputCodec('find');
+        if (out)
+            return outputCodec(out);
+    }
+    /**
+     * Finds a document in the collection that matches the specified options.
+     *
+     * @param {MongoEntityService.FindOneOptions} [options] - The options for the query.
+     * @return {Promise<PartialDTO<T> | undefined>} A promise that resolves with the found document or undefined if no document is found.
+     */
+    async _findOne(options) {
+        const filter = MongoAdapter.prepareFilter(options?.filter);
+        const mongoOptions = {
+            ...omit(options, 'filter'),
+            sort: options?.sort ? MongoAdapter.prepareSort(options.sort) : undefined,
+            projection: MongoAdapter.prepareProjection(this.dataType, options?.projection),
+            limit: undefined,
+        };
+        const out = await this._dbFindOne(filter, mongoOptions);
+        const outputCodec = this.getOutputCodec('find');
+        if (out)
+            return outputCodec(out);
+    }
+    /**
+     * Finds multiple documents in the MongoDB collection.
+     *
+     * @param {MongoEntityService.FindManyOptions<T>} [options] - The options for the find operation.
+     * @return A Promise that resolves to an array of partial outputs of type T.
+     */
+    async _findMany(options) {
+        const mongoOptions = {
+            ...omit(options, ['projection', 'sort', 'skip', 'limit', 'filter']),
+        };
+        const limit = options?.limit || 10;
+        const stages = [];
+        let filter;
+        if (options?.filter)
+            filter = MongoAdapter.prepareFilter(options?.filter);
+        if (filter)
+            stages.push({ $match: filter });
+        if (options?.skip)
+            stages.push({ $skip: options.skip });
+        if (options?.sort) {
+            const sort = MongoAdapter.prepareSort(options.sort);
+            if (sort)
+                stages.push({ $sort: sort });
+        }
+        stages.push({ $limit: limit });
+        const dataType = this.dataType;
+        const projection = MongoAdapter.prepareProjection(dataType, options?.projection);
+        if (projection)
+            stages.push({ $project: projection });
+        const cursor = await this._dbAggregate(stages, mongoOptions);
+        /** Execute db command */
+        try {
+            /** Fetch the cursor and decode the result objects */
+            const outputCodec = this.getOutputCodec('find');
+            return (await cursor.toArray()).map((r) => outputCodec(r));
+        }
+        finally {
+            if (!cursor.closed)
+                await cursor.close();
+        }
+    }
+    /**
+     * Finds multiple documents in the collection and returns both records (max limit)
+     * and total count that matched the given criteria
+     *
+     * @param {MongoEntityService.FindManyOptions<T>} [options] - The options for the find operation.
+     * @return A Promise that resolves to an array of partial outputs of type T.
+     */
+    async _findManyWithCount(options) {
+        const mongoOptions = {
+            ...omit(options, ['projection', 'sort', 'skip', 'limit', 'filter']),
+        };
+        const limit = options?.limit || 10;
+        let filter;
+        if (options?.filter)
+            filter = MongoAdapter.prepareFilter(options?.filter);
+        const dataStages = [];
+        const countStages = [];
+        if (filter)
+            countStages.push({ $match: filter });
+        countStages.push({ $count: 'totalMatches' });
+        const stages = [
+            {
+                $facet: {
+                    data: dataStages,
+                    count: countStages,
+                },
+            },
+        ];
+        if (filter)
+            dataStages.push({ $match: filter });
+        if (options?.skip)
+            dataStages.push({ $skip: options.skip });
+        if (options?.sort) {
+            const sort = MongoAdapter.prepareSort(options.sort);
+            if (sort)
+                dataStages.push({ $sort: sort });
+        }
+        dataStages.push({ $limit: limit });
+        const dataType = this.dataType;
+        const projection = MongoAdapter.prepareProjection(dataType, options?.projection);
+        if (projection)
+            dataStages.push({ $project: projection });
+        const outputCodec = this.getOutputCodec('find');
+        /** Execute db command */
+        const cursor = await this._dbAggregate(stages, mongoOptions);
+        try {
+            /** Fetch the cursor and decode the result objects */
+            const facetResult = await cursor.toArray();
+            return {
+                count: facetResult[0].count[0]?.totalMatches || 0,
+                items: facetResult[0].data?.map((r) => outputCodec(r)),
+            };
+        }
+        finally {
+            if (!cursor.closed)
+                await cursor.close();
+        }
+    }
+    /**
+     * Updates a document with the given id in the collection.
+     *
+     * @param {AnyId} id - The id of the document to update.
+     * @param {PatchDTO<T>|UpdateFilter<T>} input - The partial input object containing the fields to update.
+     * @param {MongoEntityService.UpdateOptions<T>} [options] - The options for the update operation.
+     * @returns {Promise<PartialDTO<T> | undefined>} A promise that resolves to the updated document or
+     * undefined if the document was not found.
+     */
+    async _update(id, input, options) {
+        const isUpdateFilter = Array.isArray(input) || !!Object.keys(input).find(x => x.startsWith('$'));
+        const isDocument = !Array.isArray(input) && !!Object.keys(input).find(x => !x.startsWith('$'));
+        if (isUpdateFilter && isDocument) {
+            throw new TypeError('You must pass one of MongoDB UpdateFilter or a partial document, not both');
+        }
+        let update;
+        if (isDocument) {
+            const inputCodec = this.getInputCodec('update');
+            const doc = inputCodec(input);
+            delete doc._id;
+            update = MongoAdapter.preparePatch(doc);
+            update.$set = update.$set || {};
+        }
+        else
+            update = input;
+        const filter = MongoAdapter.prepareFilter([MongoAdapter.prepareKeyValues(id, ['_id']), options?.filter]);
+        const mongoOptions = {
+            ...options,
+            includeResultMetadata: false,
+            upsert: undefined,
+            projection: MongoAdapter.prepareProjection(this.dataType, options?.projection),
+        };
+        const out = await this._dbFindOneAndUpdate(filter, update, mongoOptions);
+        const outputCodec = this.getOutputCodec('update');
+        if (out)
+            return outputCodec(out);
+    }
+    /**
+     * Updates a document in the collection with the specified ID.
+     *
+     * @param {MongoAdapter.AnyId} id - The ID of the document to update.
+     * @param {PatchDTO<T>|UpdateFilter<T>} input - The partial input data to update the document with.
+     * @param {MongoEntityService.UpdateOptions<T>} [options] - The options for updating the document.
+     * @returns {Promise<number>} - A promise that resolves to the number of documents modified.
+     */
+    async _updateOnly(id, input, options) {
+        const isUpdateFilter = Array.isArray(input) || !!Object.keys(input).find(x => x.startsWith('$'));
+        const isDocument = !Array.isArray(input) && !!Object.keys(input).find(x => !x.startsWith('$'));
+        if (isUpdateFilter && isDocument) {
+            throw new TypeError('You must pass one of MongoDB UpdateFilter or a partial document, not both');
+        }
+        let update;
+        if (isDocument) {
+            const inputCodec = this.getInputCodec('update');
+            const doc = inputCodec(input);
+            delete doc._id;
+            update = MongoAdapter.preparePatch(doc);
+            if (!Object.keys(doc).length)
+                return 0;
+        }
+        else
+            update = input;
+        const filter = MongoAdapter.prepareFilter([MongoAdapter.prepareKeyValues(id, ['_id']), options?.filter]);
+        const mongoOptions = {
+            ...options,
+            includeResultMetadata: false,
+            upsert: undefined,
+            projection: MongoAdapter.prepareProjection(this.dataType, options?.projection),
+        };
+        const out = await this._dbUpdateOne(filter, update, mongoOptions);
+        return out.matchedCount;
+    }
+    /**
+     * Updates multiple documents in the collection based on the specified input and options.
+     *
+     * @param {PatchDTO<T>|UpdateFilter<T>} input - The partial input to update the documents with.
+     * @param {MongoEntityService.UpdateManyOptions<T>} [options] - The options for updating the documents.
+     * @return {Promise<number>} - A promise that resolves to the number of documents matched and modified.
+     */
+    async _updateMany(input, options) {
+        const isUpdateFilter = Array.isArray(input) || !!Object.keys(input).find(x => x.startsWith('$'));
+        const isDocument = !Array.isArray(input) && !!Object.keys(input).find(x => !x.startsWith('$'));
+        if (isUpdateFilter && isDocument) {
+            throw new TypeError('You must pass one of MongoDB UpdateFilter or a partial document, not both');
+        }
+        let update;
+        if (isDocument) {
+            const inputCodec = this.getInputCodec('update');
+            const doc = inputCodec(input);
+            delete doc._id;
+            update = MongoAdapter.preparePatch(doc);
+            if (!Object.keys(doc).length)
+                return 0;
+        }
+        else
+            update = input;
+        const mongoOptions = {
+            ...omit(options, 'filter'),
+            upsert: undefined,
+        };
+        const filter = MongoAdapter.prepareFilter(options?.filter);
+        const r = await this._dbUpdateMany(filter, update, mongoOptions);
+        return r.matchedCount;
+    }
+}