From a1f7c461e9f635e085f93a804f3c820ccc9d614d Mon Sep 17 00:00:00 2001
From: Max Hirschhorn <max.hirschhorn@mongodb.com>
Date: Sun, 11 Feb 2018 18:54:58 -0500
Subject: SERVER-33219 Add backpressure to CRUD client in backup_restore.js.

Also fixes the FSM client by changing the concurrency suite to handle
when TestData isn't defined.

(cherry picked from commit fdc23cf6a586a607299ec8a880574e95071e8f39)
---
 jstests/concurrency/fsm_libs/worker_thread.js |  6 +++-
 jstests/noPassthrough/libs/backup_restore.js  | 42 ++++++++++++++++++++-------
 2 files changed, 36 insertions(+), 12 deletions(-)

(limited to 'jstests')

diff --git a/jstests/concurrency/fsm_libs/worker_thread.js b/jstests/concurrency/fsm_libs/worker_thread.js
index 83f0384562a..8a5611414c0 100644
--- a/jstests/concurrency/fsm_libs/worker_thread.js
+++ b/jstests/concurrency/fsm_libs/worker_thread.js
@@ -26,7 +26,11 @@ var workerThread = (function() {
         var configs = {};
 
         globalAssertLevel = args.globalAssertLevel;
-        TestData = args.testData;
+
+        // The global 'TestData' object may still be undefined if the concurrency suite isn't being
+        // run by resmoke.py (e.g. if it is being run via a parallel shell in the backup/restore
+        // tests).
+        TestData = (args.testData !== undefined) ? args.testData : {};
 
         try {
             if (Cluster.isStandalone(args.clusterOptions)) {
diff --git a/jstests/noPassthrough/libs/backup_restore.js b/jstests/noPassthrough/libs/backup_restore.js
index 9d2e9c23a16..4cdddd434b6 100644
--- a/jstests/noPassthrough/libs/backup_restore.js
+++ b/jstests/noPassthrough/libs/backup_restore.js
@@ -41,17 +41,36 @@ var BackupRestoreTest = function(options) {
     /**
      * Starts a client that will run a CRUD workload.
      */
-    function _crudClient(host, dbName, collectionName) {
+    function _crudClient(host, dbName, collectionName, numNodes) {
         // Launch CRUD client
-        var crudClientCmds = function(dbName, collectionName) {
+        var crudClientCmds = function(dbName, collectionName, numNodes) {
             var bulkNum = 1000;
             var baseNum = 100000;
+
+            let iteration = 0;
+
             var coll = db.getSiblingDB(dbName).getCollection(collectionName);
             coll.ensureIndex({x: 1});
+
             var largeValue = new Array(1024).join('L');
+
             Random.setRandomSeed();
+
             // Run indefinitely.
             while (true) {
+                ++iteration;
+
+                // We periodically use a write concern of w='numNodes' as a backpressure mechanism
+                // to prevent the secondaries from falling off the primary's oplog. The CRUD client
+                // inserts ~1KB documents 1000 at a time, so in the worst case we'll have rolled the
+                // primary's oplog over every ~1000 iterations. We use 100 iterations for the
+                // frequency of when to use a write concern of w='numNodes' to lessen the risk of
+                // being unlucky as a result of running concurrently with the FSM client. Note that
+                // although the updates performed by the CRUD client may in the worst case modify
+                // every document, the oplog entries produced as a result are 10x smaller than the
+                // document itself.
+                const writeConcern = (iteration % 100 === 0) ? {w: numNodes} : {w: 1};
+
                 try {
                     var op = Random.rand();
                     var match = Math.floor(Random.rand() * baseNum);
@@ -64,10 +83,10 @@ var BackupRestoreTest = function(options) {
                                 doc: largeValue.substring(0, match % largeValue.length),
                             });
                         }
-                        assert.writeOK(bulk.execute());
+                        assert.writeOK(bulk.execute(writeConcern));
                     } else if (op < 0.4) {
                         // 20% of the operations: update docs.
-                        var updateOpts = {upsert: true, multi: true};
+                        var updateOpts = {upsert: true, multi: true, writeConcern: writeConcern};
                         assert.writeOK(coll.update({x: {$gte: match}},
                                                    {$inc: {x: baseNum}, $set: {n: 'hello'}},
                                                    updateOpts));
@@ -77,7 +96,8 @@ var BackupRestoreTest = function(options) {
                         coll.find({x: {$gte: match}}).itcount();
                     } else {
                         // 10% of the operations: remove matching docs.
-                        assert.writeOK(coll.remove({x: {$gte: match}}));
+                        assert.writeOK(
+                            coll.remove({x: {$gte: match}}, {writeConcern: writeConcern}));
                     }
                 } catch (e) {
                     if (e instanceof ReferenceError || e instanceof TypeError) {
@@ -89,11 +109,11 @@ var BackupRestoreTest = function(options) {
 
         // Returns the pid of the started mongo shell so the CRUD test client can be terminated
         // without waiting for its execution to finish.
-        return startMongoProgramNoConnect(
-            'mongo',
-            '--eval',
-            '(' + crudClientCmds + ')("' + dbName + '", "' + collectionName + '")',
-            host);
+        return startMongoProgramNoConnect('mongo',
+                                          '--eval',
+                                          '(' + crudClientCmds + ')("' + dbName + '", "' +
+                                              collectionName + '", ' + numNodes + ')',
+                                          host);
     }
 
     /**
@@ -222,7 +242,7 @@ var BackupRestoreTest = function(options) {
         // Launch CRUD client
         var crudDb = "crud";
         var crudColl = "backuprestore";
-        var crudPid = _crudClient(primary.host, crudDb, crudColl);
+        var crudPid = _crudClient(primary.host, crudDb, crudColl, numNodes);
 
         // Launch FSM client
         var fsmPid = _fsmClient(primary.host, crudDb, numNodes);
-- 
cgit v1.2.1