feat: expose countDocuments in mongodb (#10314)

FxEasy · web-flow · commit ebd61d14400e · 2023-09-30T15:09:58.000+05:00
diff --git a/docs/mongodb.md b/docs/mongodb.md
@@ -284,6 +284,10 @@ Perform a bulkWrite operation without a fluent API.
 
 Count number of matching documents in the db to a query.
 
+### `countDocuments`
+
+Count number of matching documents in the db to a query.
+
 #### `createCollectionIndex`
 
 Creates an index on the db and collection.
diff --git a/src/entity-manager/MongoEntityManager.ts b/src/entity-manager/MongoEntityManager.ts
@@ -54,6 +54,7 @@ import {
     IndexInformationOptions,
     ObjectId,
     FilterOperators,
+    CountDocumentsOptions,
 } from "../driver/mongodb/typings"
 import { DataSource } from "../data-source/DataSource"
 import { MongoFindManyOptions } from "../find-options/mongodb/MongoFindManyOptions"
@@ -499,6 +500,22 @@ export class MongoEntityManager extends EntityManager {
         return this.mongoQueryRunner.count(metadata.tableName, query, options)
     }
 
+    /**
+     * Count number of matching documents in the db to a query.
+     */
+    countDocuments<Entity>(
+        entityClassOrName: EntityTarget<Entity>,
+        query: Filter<Document> = {},
+        options: CountDocumentsOptions = {},
+    ): Promise<number> {
+        const metadata = this.connection.getMetadata(entityClassOrName)
+        return this.mongoQueryRunner.countDocuments(
+            metadata.tableName,
+            query,
+            options,
+        )
+    }
+
     /**
      * Count number of matching documents in the db to a query.
      */
diff --git a/src/repository/MongoRepository.ts b/src/repository/MongoRepository.ts
@@ -43,6 +43,7 @@ import {
     UpdateFilter,
     UpdateOptions,
     UpdateResult,
+    CountDocumentsOptions,
 } from "../driver/mongodb/typings"
 import { FindManyOptions } from "../find-options/FindManyOptions"
 
@@ -244,6 +245,20 @@ export class MongoRepository<
         return this.manager.count(this.metadata.target, query || {}, options)
     }
 
+    /**
+     * Count number of matching documents in the db to a query.
+     */
+    countDocuments(
+        query?: ObjectLiteral,
+        options?: CountDocumentsOptions,
+    ): Promise<number> {
+        return this.manager.countDocuments(
+            this.metadata.target,
+            query || {},
+            options,
+        )
+    }
+
     /**
      * Count number of matching documents in the db to a query.
      */
