fix(query): maxTimeMS in count, countDocuments, distinct

Automattic · Nov 15, 2020 · ffb4208 · ffb4208
1 parent 222a69f
commit ffb4208
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lib/query.js b/lib/query.js
@@ -2217,6 +2217,8 @@ Query.prototype._count = wrapThunk(function(callback) {
     return callback(this.error());
   }
 
+  applyGlobalMaxTimeMS(this.options, this.model);
+
   const conds = this._conditions;
   const options = this._optionsForExec();
 
@@ -2242,6 +2244,8 @@ Query.prototype._countDocuments = wrapThunk(function(callback) {
     return callback(this.error());
   }
 
+  applyGlobalMaxTimeMS(this.options, this.model);
+
   const conds = this._conditions;
   const options = this._optionsForExec();
 
@@ -2447,6 +2451,8 @@ Query.prototype.__distinct = wrapThunk(function __distinct(callback) {
     return null;
   }
 
+  applyGlobalMaxTimeMS(this.options, this.model);
+
   const options = this._optionsForExec();
 
   // don't pass in the conditions because we already merged them in