@mikro-orm/mongodb 7.1.0-dev.1 → 7.1.0-dev.11

package/MongoDriver.d.ts

@@ -15,11 +15,13 @@ export declare class MongoDriver extends DatabaseDriver<MongoConnection> {
         rawResults?: boolean;
     }): AsyncIterableIterator<T>;
     find<T extends object, P extends string = never, F extends string = never, E extends string = never>(entityName: EntityName<T>, where: FilterQuery<T>, options?: FindOptions<T, P, F, E>): Promise<EntityData<T>[]>;
+    private findWithPartitionLimit;
     findOne<T extends object, P extends string = never, F extends string = never, E extends string = never>(entityName: EntityName<T>, where: FilterQuery<T>, options?: FindOneOptions<T, P, F, E>): Promise<EntityData<T> | null>;
     findVirtual<T extends object>(entityName: EntityName<T>, where: FilterQuery<T>, options: FindOptions<T, any, any, any>): Promise<EntityData<T>[]>;
     streamVirtual<T extends object>(entityName: EntityName<T>, where: FilterQuery<T>, options: FindOptions<T, any, any, any>): AsyncIterableIterator<EntityData<T>>;
     count<T extends object>(entityName: EntityName<T>, where: FilterQuery<T>, options?: CountOptions<T>): Promise<number>;
     nativeInsert<T extends object>(entityName: EntityName<T>, data: EntityDictionary<T>, options?: NativeInsertUpdateOptions<T>): Promise<QueryResult<T>>;
+    nativeClone<T extends object>(entityName: EntityName<T>, where: FilterQuery<T>, overrides?: EntityData<T>, options?: NativeInsertUpdateOptions<T>): Promise<QueryResult<T>>;
     nativeInsertMany<T extends object>(entityName: EntityName<T>, data: EntityDictionary<T>[], options?: NativeInsertUpdateManyOptions<T>): Promise<QueryResult<T>>;
     nativeUpdate<T extends object>(entityName: EntityName<T>, where: FilterQuery<T>, data: EntityDictionary<T>, options?: NativeInsertUpdateOptions<T> & UpsertOptions<T>): Promise<QueryResult<T>>;
     nativeUpdateMany<T extends object>(entityName: EntityName<T>, where: FilterQuery<T>[], data: EntityDictionary<T>[], options?: NativeInsertUpdateOptions<T> & UpsertManyOptions<T>): Promise<QueryResult<T>>;
@@ -30,7 +32,8 @@ export declare class MongoDriver extends DatabaseDriver<MongoConnection> {
     streamAggregate<T extends object>(entityName: EntityName<T>, pipeline: any[], ctx?: Transaction<ClientSession>): AsyncIterableIterator<T>;
     getPlatform(): MongoPlatform;
     private buildQueryOptions;
-    private renameFields;
+    /** @internal */
+    renameFields<T extends object>(entityName: EntityName<T>, data: T, dotPaths?: boolean, object?: boolean, root?: boolean): T;
     private convertObjectIds;
     private buildFilterById;
     protected buildFields<T extends object, P extends string = never>(entityName: EntityName<T>, populate: PopulateOptions<T>[], fields?: readonly EntityField<T, P>[], exclude?: string[]): string[] | undefined;

package/MongoDriver.js

@@ -78,6 +78,10 @@ export class MongoDriver extends DatabaseDriver {
             return res.map(r => this.mapResult(r, this.metadata.find(entityName)));
         }
         const orderBy = Utils.asArray(options.orderBy).map(orderBy => this.renameFields(entityName, orderBy, true));
+        const partitionLimit = options._partitionLimit;
+        if (partitionLimit) {
+            return this.findWithPartitionLimit(entityName, where, orderBy, fields ?? [], partitionLimit, options);
+        }
         const res = await this.rethrow(this.getConnection('read').find(entityName, where, {
             orderBy,
             limit: options.limit,
@@ -88,6 +92,48 @@ export class MongoDriver extends DatabaseDriver {
         }));
         return res.map(r => this.mapResult(r, this.metadata.find(entityName)));
     }
+    async findWithPartitionLimit(entityName, where, orderBy, fields, partitionLimit, options) {
+        const meta = this.metadata.find(entityName);
+        const { limit, offset = 0 } = partitionLimit;
+        // Resolve the partition property to its DB field name; callers always pass
+        // a declared property on the target meta (FK of the owning side).
+        const prop = meta.properties[partitionLimit.partitionBy];
+        const partitionField = prop.fieldNames[0];
+        const pipeline = [];
+        pipeline.push({ $match: where });
+        if (orderBy.length > 0) {
+            const sortSpec = {};
+            for (const order of orderBy) {
+                for (const [key, dir] of Object.entries(order)) {
+                    sortSpec[key] = dir === 'ASC' || dir === 'asc' || dir === 1 ? 1 : -1;
+                }
+            }
+            pipeline.push({ $sort: sortSpec });
+        }
+        // $sort before $group ensures $push collects in correct order
+        pipeline.push({
+            $group: {
+                _id: `$${partitionField}`,
+                __docs: { $push: '$$ROOT' },
+            },
+        });
+        pipeline.push({
+            $project: {
+                __docs: { $slice: ['$__docs', offset, limit] },
+            },
+        });
+        pipeline.push({ $unwind: '$__docs' });
+        pipeline.push({ $replaceRoot: { newRoot: '$__docs' } });
+        if (fields.length > 0) {
+            const projection = {};
+            for (const field of fields) {
+                projection[field] = 1;
+            }
+            pipeline.push({ $project: projection });
+        }
+        const res = await this.rethrow(this.getConnection('read').aggregate(entityName, pipeline, options.ctx, options.logging));
+        return res.map(r => this.mapResult(r, meta));
+    }
     async findOne(entityName, where, options = { populate: [], orderBy: {} }) {
         if (this.metadata.find(entityName)?.virtual) {
             const [item] = await this.findVirtual(entityName, where, options);
@@ -148,6 +194,27 @@ export class MongoDriver extends DatabaseDriver {
         data = this.renameFields(entityName, data);
         return this.rethrow(this.getConnection('write').insertOne(entityName, data, options.ctx));
     }
+    async nativeClone(entityName, where, overrides, options = {}) {
+        const meta = this.metadata.find(entityName);
+        const pk = meta.getPrimaryProps()[0].fieldNames[0] ?? '_id';
+        const normalizedWhere = Utils.isPrimaryKey(where) ? { [pk]: where } : where;
+        const renameWhere = this.renameFields(entityName, normalizedWhere);
+        const source = await this.rethrow(this.getConnection('read').find(entityName, renameWhere, { ctx: options.ctx, limit: 1 }));
+        if (!source.length) {
+            throw new Error('Cannot clone: no entity found matching the given condition');
+        }
+        const doc = source[0];
+        delete doc[pk];
+        if (overrides) {
+            const mapped = this.renameFields(entityName, overrides);
+            Object.assign(doc, mapped);
+        }
+        if (meta.versionProperty) {
+            const vProp = meta.properties[meta.versionProperty];
+            doc[vProp.fieldNames[0]] = vProp.runtimeType === 'Date' ? new Date() : 1;
+        }
+        return this.rethrow(this.getConnection('write').insertOne(entityName, doc, options.ctx));
+    }
     async nativeInsertMany(entityName, data, options = {}) {
         data = data.map(item => {
             this.handleVersionProperty(entityName, item);
@@ -247,6 +314,13 @@ export class MongoDriver extends DatabaseDriver {
         if (options.indexHint != null) {
             ret.indexHint = options.indexHint;
         }
+        else if (options.using != null) {
+            const names = Utils.asArray(options.using);
+            if (names.length > 1) {
+                throw new Error('MongoDB only supports a single index hint per query. Provide one index name instead of an array.');
+            }
+            ret.indexHint = names[0];
+        }
         if (options.maxTimeMS != null) {
             ret.maxTimeMS = options.maxTimeMS;
         }
@@ -255,6 +329,7 @@ export class MongoDriver extends DatabaseDriver {
         }
         return ret;
     }
+    /** @internal */
     renameFields(entityName, data, dotPaths = false, object, root = true) {
         if (data == null && root) {
             return {};

package/MongoEntityManager.d.ts

@@ -1,4 +1,4 @@
-import { EntityManager, type EntityName, type EntityRepository, type GetRepository, type TransactionOptions, type StreamOptions, type Loaded } from '@mikro-orm/core';
+import { EntityManager, type CountByOptions, type Dictionary, type EntityKey, type EntityName, type EntityRepository, type GetRepository, type Loaded, type StreamOptions, type TransactionOptions, type WithUsingOptions } from '@mikro-orm/core';
 import type { Collection, Document, TransactionOptions as MongoTransactionOptions } from 'mongodb';
 import type { MongoDriver } from './MongoDriver.js';
 import type { MongoEntityRepository } from './MongoEntityRepository.js';
@@ -17,8 +17,12 @@ export declare class MongoEntityManager<Driver extends MongoDriver = MongoDriver
     /**
      * @inheritDoc
      */
-    stream<Entity extends object, Hint extends string = never, Fields extends string = never, Excludes extends string = never>(entityName: EntityName<Entity>, options?: StreamOptions<NoInfer<Entity>, Hint, Fields, Excludes>): AsyncIterableIterator<Loaded<Entity, Hint, Fields, Excludes>>;
+    stream<Entity extends object, Hint extends string = never, Fields extends string = never, Excludes extends string = never, Using extends string = never>(entityName: EntityName<Entity>, options?: WithUsingOptions<StreamOptions<NoInfer<Entity>, Hint, Fields, Excludes>, NoInfer<Entity>, Using>): AsyncIterableIterator<Loaded<Entity, Hint, Fields, Excludes>>;
     getCollection<T extends Document>(entityOrCollectionName: EntityName<T> | string): Collection<T>;
+    /**
+     * @inheritDoc
+     */
+    countBy<Entity extends object>(entityName: EntityName<Entity>, groupBy: EntityKey<Entity> | readonly EntityKey<Entity>[], options?: CountByOptions<Entity>): Promise<Dictionary<number>>;
     /**
      * @inheritDoc
      */

package/MongoEntityManager.js

@@ -27,6 +27,37 @@ export class MongoEntityManager extends EntityManager {
     getCollection(entityOrCollectionName) {
         return this.getConnection().getCollection(entityOrCollectionName);
     }
+    /**
+     * @inheritDoc
+     */
+    async countBy(entityName, groupBy, options = {}) {
+        const em = this.getContext(false);
+        options = { ...options };
+        em.prepareOptions(options);
+        const meta = em.getMetadata().find(entityName);
+        const fields = Utils.asArray(groupBy);
+        if (options.having) {
+            throw new Error('The `having` option is not supported for MongoDB in `countBy`.');
+        }
+        await em.tryFlush(entityName, options);
+        const rawWhere = options.where;
+        const where = await em.processWhere(entityName, rawWhere ?? {}, options, 'read');
+        const renamedWhere = em.getDriver().renameFields(meta.class, where, true);
+        const fieldNames = fields.map(f => meta.properties[f]?.fieldNames?.[0] ?? f);
+        const groupId = fieldNames.length === 1 ? `$${fieldNames[0]}` : Object.fromEntries(fieldNames.map(f => [f, `$${f}`]));
+        const pipeline = [];
+        if (renamedWhere && Object.keys(renamedWhere).length > 0) {
+            pipeline.push({ $match: renamedWhere });
+        }
+        pipeline.push({ $group: { _id: groupId, count: { $sum: 1 } } });
+        const rows = await em.getDriver().aggregate(meta.class, pipeline, em.getTransactionContext());
+        const results = {};
+        for (const row of rows) {
+            const key = fieldNames.length === 1 ? String(row._id) : fieldNames.map(f => String(row._id[f])).join(Utils.PK_SEPARATOR);
+            results[key] = +row.count;
+        }
+        return results;
+    }
     /**
      * @inheritDoc
      */

package/MongoMikroORM.d.ts

@@ -2,17 +2,17 @@ import { type AnyEntity, type EntityClass, type EntitySchema, MikroORM, type Opt
 import { MongoDriver } from './MongoDriver.js';
 import type { MongoEntityManager } from './MongoEntityManager.js';
 /** Configuration options for the MongoDB driver. */
-export type MongoOptions<EM extends MongoEntityManager = MongoEntityManager, Entities extends (string | EntityClass<AnyEntity> | EntitySchema)[] = (string | EntityClass<AnyEntity> | EntitySchema)[]> = Partial<Options<MongoDriver, EM, Entities>>;
+export type MongoOptions<EM extends MongoEntityManager = MongoEntityManager, Entities extends readonly (string | EntityClass<AnyEntity> | EntitySchema)[] = (string | EntityClass<AnyEntity> | EntitySchema)[]> = Partial<Options<MongoDriver, EM, Entities>>;
 /** Creates a type-safe configuration object for the MongoDB driver. */
-export declare function defineMongoConfig<EM extends MongoEntityManager = MongoEntityManager, Entities extends (string | EntityClass<AnyEntity> | EntitySchema)[] = (string | EntityClass<AnyEntity> | EntitySchema)[]>(options: MongoOptions<EM, Entities>): MongoOptions<EM, Entities>;
+export declare function defineMongoConfig<EM extends MongoEntityManager = MongoEntityManager, Entities extends readonly (string | EntityClass<AnyEntity> | EntitySchema)[] = (string | EntityClass<AnyEntity> | EntitySchema)[]>(options: MongoOptions<EM, Entities>): MongoOptions<EM, Entities>;
 /**
  * @inheritDoc
  */
-export declare class MongoMikroORM<EM extends MongoEntityManager = MongoEntityManager, Entities extends (string | EntityClass<AnyEntity> | EntitySchema)[] = (string | EntityClass<AnyEntity> | EntitySchema)[]> extends MikroORM<MongoDriver, EM, Entities> {
+export declare class MongoMikroORM<EM extends MongoEntityManager = MongoEntityManager, Entities extends readonly (string | EntityClass<AnyEntity> | EntitySchema)[] = (string | EntityClass<AnyEntity> | EntitySchema)[]> extends MikroORM<MongoDriver, EM, Entities> {
     /**
      * @inheritDoc
      */
-    static init<D extends IDatabaseDriver = MongoDriver, EM extends EntityManager<D> = D[typeof EntityManagerType] & EntityManager<D>, Entities extends (string | EntityClass<AnyEntity> | EntitySchema)[] = (string | EntityClass<AnyEntity> | EntitySchema)[]>(options: Partial<Options<D, EM, Entities>>): Promise<MikroORM<D, EM, Entities>>;
+    static init<D extends IDatabaseDriver = MongoDriver, EM extends EntityManager<D> = D[typeof EntityManagerType] & EntityManager<D>, Entities extends readonly (string | EntityClass<AnyEntity> | EntitySchema)[] = (string | EntityClass<AnyEntity> | EntitySchema)[]>(options: Partial<Options<D, EM, Entities>>): Promise<MikroORM<D, EM, Entities>>;
     /**
      * @inheritDoc
      */

package/MongoPlatform.js

@@ -76,6 +76,9 @@ export class MongoPlatform extends Platform {
         if (meta.inheritanceType === 'tpt') {
             throw MetadataError.tptNotSupportedByDriver(meta);
         }
+        if (meta.triggers?.length > 0) {
+            throw MetadataError.triggersNotSupportedByDriver(meta);
+        }
         const pk = meta.getPrimaryProps()[0];
         if (pk && pk.fieldNames?.[0] !== '_id') {
             throw MetadataError.invalidPrimaryKey(meta, pk, '_id');

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mikro-orm/mongodb",
-  "version": "7.1.0-dev.1",
+  "version": "7.1.0-dev.11",
   "description": "TypeScript ORM for Node.js based on Data Mapper, Unit of Work and Identity Map patterns. Supports MongoDB, MySQL, PostgreSQL and SQLite databases as well as usage with vanilla JavaScript.",
   "keywords": [
     "data-mapper",
@@ -53,7 +53,7 @@
     "@mikro-orm/core": "^7.0.11"
   },
   "peerDependencies": {
-    "@mikro-orm/core": "7.1.0-dev.1"
+    "@mikro-orm/core": "7.1.0-dev.11"
   },
   "engines": {
     "node": ">= 22.17.0"