fix(read-operations): send sessions on all read operations

Somewhere during the wire protocol refactor we lost support for sending `lsid` on read operations. This commit reintroduces support for sessions in all of these code paths.
mongodb · Aug 13, 2019 · 4d45c8a · 4d45c8a
1 parent 6078bcd
commit 4d45c8a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/lib/core/wireprotocol/get_more.js b/lib/core/wireprotocol/get_more.js
@@ -80,6 +80,10 @@ function getMore(server, ns, cursorState, batchSize, options, callback) {
     options
   );
 
+  if (cursorState.session) {
+    commandOptions.session = cursorState.session;
+  }
+
   command(server, ns, getMoreCmd, commandOptions, queryCallback);
 }
 

diff --git a/lib/core/wireprotocol/query.js b/lib/core/wireprotocol/query.js
@@ -46,7 +46,14 @@ function query(server, ns, cmd, cursorState, options, callback) {
     options
   );
 
-  if (cmd.readPreference) commandOptions.readPreference = readPreference;
+  if (cmd.readPreference) {
+    commandOptions.readPreference = readPreference;
+  }
+
+  if (cursorState.session) {
+    commandOptions.session = cursorState.session;
+  }
+
   command(server, ns, findCmd, commandOptions, callback);
 }