SERVER-51362 Ensure update and delete shell helpers support 'let' variables

1 files changed, 10 insertions, 0 deletions

diff --git a/src/mongo/shell/bulk_api.js b/src/mongo/shell/bulk_api.js
index 61b89819b69..139f0bc631b 100644
--- a/src/mongo/shell/bulk_api.js
+++ b/src/mongo/shell/bulk_api.js
@@ -538,6 +538,7 @@ var _bulk_api_module = (function() {
         var maxBatchSizeBytes = 1024 * 1024 * 16;
         var maxNumberOfDocsInBatch = (TestData && TestData.disableBatchWrites) ? 1 : 1000;
         var writeConcern = null;
+        var letParams = null;
         var currentOp;
 
         // Final results
@@ -808,6 +809,10 @@ var _bulk_api_module = (function() {
             return findOperations;
         };
 
+        this.setLetParams = function(userLet) {
+            letParams = userLet;
+        };
+
         //
         // Merge write command result into aggregated results object
         var mergeBatchResults = function(batch, bulkResult, result) {
@@ -897,6 +902,11 @@ var _bulk_api_module = (function() {
                 cmd.writeConcern = writeConcern;
             }
 
+            // If we have let parameters, add them to the command.
+            if (letParams) {
+                cmd.let = letParams;
+            }
+
             {
                 const kWireVersionSupportingRetryableWrites = 6;
                 const serverSupportsRetryableWrites =