SERVER-46567 Update shell methods Mongo.setReadPref and cursor.readPref to support hedging

1 files changed, 6 insertions, 2 deletions

diff --git a/src/mongo/shell/query.js b/src/mongo/shell/query.js
index cbdbeda2929..cb72bf0fff4 100644
--- a/src/mongo/shell/query.js
+++ b/src/mongo/shell/query.js
@@ -489,17 +489,21 @@ DBQuery.prototype.allowDiskUse = function() {
  *
  * @param mode {string} read preference mode to use.
  * @param tagSet {Array.<Object>} optional. The list of tags to use, order matters.
- *     Note that this object only keeps a shallow copy of this array.
+ * @param hedgeOptions {<Object>} optional. The hedge options of the form {enabled: <bool>}.
  *
  * @return this cursor
  */
-DBQuery.prototype.readPref = function(mode, tagSet) {
+DBQuery.prototype.readPref = function(mode, tagSet, hedgeOptions) {
     var readPrefObj = {mode: mode};
 
     if (tagSet) {
         readPrefObj.tags = tagSet;
     }
 
+    if (hedgeOptions) {
+        readPrefObj.hedge = hedgeOptions;
+    }
+
     return this._addSpecial("$readPreference", readPrefObj);
 };