SERVER-21630 run CheckReplDBHash on csrs and shard replset

1 files changed, 165 insertions, 80 deletions

diff --git a/jstests/hooks/run_check_repl_dbhash.js b/jstests/hooks/run_check_repl_dbhash.js
index 5a9606e20b1..344595de46e 100644
--- a/jstests/hooks/run_check_repl_dbhash.js
+++ b/jstests/hooks/run_check_repl_dbhash.js
@@ -3,96 +3,181 @@
 'use strict';
 
 (function() {
-    // A thin wrapper around master/slave nodes that provides the getHashes(), getPrimary(),
-    // awaitReplication(), and nodeList() methods.
-    // DEPRECATED: this wrapper only supports nodes started through resmoke's masterslave.py
-    // fixture. Please do not use it with other master/slave clusters.
-    var MasterSlaveDBHashTest = function(primaryHost) {
-        var master = new Mongo(primaryHost);
-        var masterPort = master.host.split(':')[1];
-        var slave = new Mongo('localhost:' + String(parseInt(masterPort) + 1));
-
-        this.nodeList = function() {
-            return [master.host, slave.host];
+    load('jstests/libs/parallelTester.js');
+
+    function isMasterSlave(uri) {
+        let mongo = new Mongo(uri);
+        jsTest.authenticate(mongo);
+        let cmdLineOpts = mongo.getDB("admin").adminCommand('getCmdLineOpts');
+        assert.commandWorked(cmdLineOpts);
+        return cmdLineOpts.parsed.master === true;
+    }
+
+    function isReplSet(uri, checkMultiNode) {
+        let mongo = new Mongo(uri);
+        jsTest.authenticate(mongo);
+        let isMaster = mongo.getDB('admin').isMaster();
+        return 'setName' in isMaster && isMaster.hosts.length > (checkMultiNode ? 1 : 0);
+    }
+
+    // adds the uri and description (replset or master-slave) if server needs dbhash check.
+    function checkAndAddServerDesc(uri, out) {
+        if (isReplSet(uri, true /* checkMultiNode */)) {
+            out.push({type: "replset", uri: uri});
+        } else if (isMasterSlave(uri)) {
+            out.push({type: "master-slave", uri: uri});
+        }
+    }
+
+    function checkReplDataHashThread(serverDesc, testData, excludedDBs) {
+        // A thin wrapper around master/slave nodes that provides the getHashes(), getPrimary(),
+        // awaitReplication(), and nodeList() methods.
+        // DEPRECATED: this wrapper only supports nodes started through resmoke's masterslave.py
+        // fixture. Please do not use it with other master/slave clusters.
+        let MasterSlaveDBHashTest = function(primaryHost) {
+            let master = new Mongo(primaryHost);
+            let masterPort = master.host.split(':')[1];
+            let slave = new Mongo('localhost:' + String(parseInt(masterPort) + 1));
+
+            this.nodeList = function() {
+                return [master.host, slave.host];
+            };
+
+            this.getHashes = function(db) {
+                let combinedRes = {};
+                let res = master.getDB(db).runCommand("dbhash");
+                assert.commandWorked(res);
+                combinedRes.master = res;
+
+                res = slave.getDB(db).runCommand("dbhash");
+                assert.commandWorked(res);
+                combinedRes.slaves = [res];
+
+                return combinedRes;
+            };
+
+            this.getPrimary = function() {
+                slave.setSlaveOk();
+                this.liveNodes = {master: master, slaves: [slave]};
+
+                return master;
+            };
+
+            this.getSecondaries = function() {
+                return [slave];
+            };
+
+            this.awaitReplication = function() {
+                assert.commandWorked(master.adminCommand({fsyncUnlock: 1}),
+                                     'failed to unlock the primary');
+
+                print('Starting fsync on master to flush all pending writes');
+                assert.commandWorked(master.adminCommand({fsync: 1}));
+                print('fsync on master completed');
+
+                let timeout = 60 * 1000 * 5;  // 5min timeout
+                let dbNames = master.getDBNames();
+                print('Awaiting replication of inserts into ' + dbNames);
+                for (let dbName of dbNames) {
+                    if (dbName === 'local')
+                        continue;
+                    assert.writeOK(
+                        master.getDB(dbName).await_repl.insert(
+                            {awaiting: 'repl'}, {writeConcern: {w: 2, wtimeout: timeout}}),
+                        'Awaiting replication failed');
+                }
+                print('Finished awaiting replication');
+                assert.commandWorked(master.adminCommand({fsync: 1, lock: 1}),
+                                     'failed to re-lock the primary');
+            };
+
+            this.checkReplicatedDataHashes = function() {
+                ReplSetTest({nodes: 0}).checkReplicatedDataHashes.apply(this, ["test", [], true]);
+            };
+
+            this.checkReplicaSet = function() {
+                ReplSetTest({nodes: 0}).checkReplicaSet.apply(this, arguments);
+            };
+
+            this.dumpOplog = function() {
+                print('master-slave cannot dump oplog');
+            };
         };
 
-        this.getHashes = function(db) {
-            var combinedRes = {};
-            var res = master.getDB(db).runCommand("dbhash");
-            assert.commandWorked(res);
-            combinedRes.master = res;
-
-            res = slave.getDB(db).runCommand("dbhash");
-            assert.commandWorked(res);
-            combinedRes.slaves = [res];
-
-            return combinedRes;
-        };
-
-        this.getPrimary = function() {
-            slave.setSlaveOk();
-            this.liveNodes = {master: master, slaves: [slave]};
-
-            return master;
-        };
-
-        this.getSecondaries = function() {
-            return [slave];
-        };
+        TestData = testData;
+
+        // Since UUIDs aren't explicitly replicated in master-slave deployments, we ignore the UUID
+        // in the output of the "listCollections" command to avoid reporting a known data
+        // inconsistency issue from checkReplicatedDataHashes().
+        let ignoreUUIDs = serverDesc.type === "master-slave";
+        let fixture = null;
+        if (serverDesc.type === "replset") {
+            fixture = new ReplSetTest(serverDesc.uri);
+        } else if (serverDesc.type === "master-slave") {
+            fixture = new MasterSlaveDBHashTest(serverDesc.uri);
+        } else {
+            throw "unrecognized server type " + serverDesc.type;
+        }
+        fixture.checkReplicatedDataHashes(undefined, excludedDBs, ignoreUUIDs);
+    }
+
+    let startTime = Date.now();
+    assert.neq(typeof db, 'undefined', 'No `db` object, is the shell connected to a mongod?');
 
-        this.awaitReplication = function() {
-            assert.commandWorked(master.adminCommand({fsyncUnlock: 1}),
-                                 'failed to unlock the primary');
-
-            print('Starting fsync on master to flush all pending writes');
-            assert.commandWorked(master.adminCommand({fsync: 1}));
-            print('fsync on master completed');
-
-            var timeout = 60 * 1000 * 5;  // 5min timeout
-            var dbNames = master.getDBNames();
-            print('Awaiting replication of inserts into ' + dbNames);
-            for (var dbName of dbNames) {
-                if (dbName === 'local')
-                    continue;
-                assert.writeOK(master.getDB(dbName).await_repl.insert(
-                                   {awaiting: 'repl'}, {writeConcern: {w: 2, wtimeout: timeout}}),
-                               'Awaiting replication failed');
-            }
-            print('Finished awaiting replication');
-            assert.commandWorked(master.adminCommand({fsync: 1, lock: 1}),
-                                 'failed to re-lock the primary');
-        };
+    // stores each server type (master/slave or replset) and uri.
+    let serversNeedingReplDataHashCheck = [];
 
-        this.checkReplicatedDataHashes = function() {
-            ReplSetTest({nodes: 0}).checkReplicatedDataHashes.apply(this, arguments);
-        };
+    let primaryInfo = db.isMaster();
 
-        this.checkReplicaSet = function() {
-            ReplSetTest({nodes: 0}).checkReplicaSet.apply(this, arguments);
-        };
-    };
+    let isMongos = primaryInfo.msg === 'isdbgrid';
 
-    var startTime = Date.now();
-    assert.neq(typeof db, 'undefined', 'No `db` object, is the shell connected to a mongod?');
+    let uri = db.getMongo();
 
-    var primaryInfo = db.isMaster();
 
     assert(primaryInfo.ismaster,
            'shell is not connected to the primary or master node: ' + tojson(primaryInfo));
 
-    var cmdLineOpts = db.adminCommand('getCmdLineOpts');
-    assert.commandWorked(cmdLineOpts);
-    var isMasterSlave = cmdLineOpts.parsed.master === true;
-    var testFixture = isMasterSlave ? new MasterSlaveDBHashTest(db.getMongo().host)
-                                    : new ReplSetTest(db.getMongo().host);
-    var excludedDBs = jsTest.options().excludedDBsFromDBHash || [];
-
-    // Since UUIDs aren't explicitly replicated in master-slave deployments, we ignore the UUID in
-    // the output of the "listCollections" command to avoid reporting a known data inconsistency
-    // issue from checkReplicatedDataHashes().
-    var ignoreUUIDs = isMasterSlave;
-    testFixture.checkReplicatedDataHashes(undefined, excludedDBs, ignoreUUIDs);
-
-    var totalTime = Date.now() - startTime;
+    assert(isMongos || isReplSet(uri) || isMasterSlave(uri),
+           "not replset, master/slave, or sharded cluster");
+
+    if (isMongos) {
+        // add shards and config server if they are replica sets.
+        let res = db.adminCommand('getShardMap');
+        assert.commandWorked(res);
+        let csURI = res.map.config;
+        res = db.adminCommand('listShards');
+        assert.commandWorked(res);
+        let shardURIs = res.shards.map((shard) => shard.host);
+
+        checkAndAddServerDesc(csURI, serversNeedingReplDataHashCheck);
+        shardURIs.forEach((shardURI) => {
+            checkAndAddServerDesc(shardURI, serversNeedingReplDataHashCheck);
+        });
+    } else {
+        let uri = db.getMongo().host;
+        checkAndAddServerDesc(uri, serversNeedingReplDataHashCheck);
+    }
+
+    let threads = [];
+    let excludedDBs = jsTest.options().excludedDBsFromDBHash || [];
+    serversNeedingReplDataHashCheck.forEach((serverDesc) => {
+        const thread = new ScopedThread(checkReplDataHashThread, serverDesc, TestData, excludedDBs);
+        threads.push({serverDesc: serverDesc, handle: thread});
+        thread.start();
+    });
+
+    let failedChecks = [];
+    threads.forEach(thread => {
+        thread.handle.join();
+        if (thread.handle.hasFailed()) {
+            failedChecks.push(thread.serverDesc.uri + " (" + thread.serverDesc.type + ")");
+        }
+    });
+
+    assert.eq(failedChecks.length,
+              0,
+              "dbhash check failed for the following hosts: " + failedChecks.join(","));
+
+    let totalTime = Date.now() - startTime;
     print('Finished consistency checks of cluster in ' + totalTime + ' ms.');
 })();