perf: added match stage before facet to use indexing

sliit-foss · Jul 5, 2024 · 881fe64 · 881fe64
1 parent 419db51
commit 881fe64
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/lib/mongoose-aggregate-paginate.js b/lib/mongoose-aggregate-paginate.js
@@ -89,7 +89,7 @@ function aggregatePaginate(query, options, callback) {
   const allowDiskUse = options.allowDiskUse || false;
   const isPaginationEnabled = options.pagination === false ? false : true;
 
-  const q = this.aggregate();
+  let q = this.aggregate();
 
   if (query.options) q.options = query.options;
 
@@ -125,7 +125,13 @@ function aggregatePaginate(query, options, callback) {
 
   let promise;
   if (options.useFacet && !options.countQuery) {
-    const [pipeline, countPipeline] = constructPipelines();
+    let [pipeline, countPipeline] = constructPipelines();
+    const match = pipeline[0]?.$match;
+    if (match) {
+      pipeline.shift();
+      countPipeline.shift();
+      q = q.match(match);
+    }
     promise = q
       .facet({
         docs: pipeline,