SERVER-22688 Provide a setParameter and shell option to control the js heap limit

1 files changed, 13 insertions, 0 deletions

diff --git a/src/mongo/shell/shell_options.cpp b/src/mongo/shell/shell_options.cpp
index 1b378836c35..324ad87fa3d 100644
--- a/src/mongo/shell/shell_options.cpp
+++ b/src/mongo/shell/shell_options.cpp
@@ -200,6 +200,9 @@ Status addMongoShellOptions(moe::OptionSection* options) {
     if (!ret.isOK())
         return ret;
 
+    options->addOptionChaining(
+        "jsHeapMBLimit", "jsHeapMBLimit", moe::Int, "set the js scope's heap size limit");
+
     return Status::OK();
 }
 
@@ -370,6 +373,16 @@ Status storeMongoShellOptions(const moe::Environment& params,
         }
     }
 
+    if (params.count("jsHeapMBLimit")) {
+        int jsHeapMBLimit = params["jsHeapMBLimit"].as<int>();
+        if (jsHeapMBLimit <= 0) {
+            StringBuilder sb;
+            sb << "ERROR: \"jsHeapMBLimit\" needs to be greater than 0";
+            return Status(ErrorCodes::BadValue, sb.str());
+        }
+        shellGlobalParams.jsHeapMBLimit = jsHeapMBLimit;
+    }
+
     if (shellGlobalParams.url == "*") {
         StringBuilder sb;
         sb << "ERROR: "