Adding session support

If the document is associated with a session, run the count query in that session as well.
CampbellSoftwareSolutions · Apr 23, 2020 · 7067339 · 7067339
1 parent 36e63b3
commit 7067339
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lib/id-validator.js b/lib/id-validator.js
@@ -119,6 +119,10 @@ function validateId (
     }
     var refModel = connection.model(refModelName)
     var query = refModel.countDocuments({_id: value})
+    var session = doc.$session && doc.$session()
+    if (session) {
+        query.session(session)
+    }
     executeQuery(query, conditions, 1, resolve, reject)
 }
 
@@ -140,6 +144,10 @@ function validateIdArray (
 
     var refModel = connection.model(refModelName)
     var query = refModel.countDocuments().where('_id')['in'](checkValues)
+    var session = doc.$session && doc.$session()
+    if (session) {
+        query.session(session)
+    }
 
     executeQuery(query, conditions, checkValues.length, resolve, reject)
 }