From e164e7d031a1d4a22a03df3382316f6107ab81c8 Mon Sep 17 00:00:00 2001
From: William Schultz <william.schultz@mongodb.com>
Date: Tue, 26 Nov 2019 16:34:39 +0000
Subject: [PATCH] SERVER-43773 Add log messages in ShardingTest to measure
 total duration of startup and initiation of shards and config server

---
 src/mongo/shell/shardingtest.js | 28 ++++++++++++++++++++++++++--
 1 file changed, 26 insertions(+), 2 deletions(-)

diff --git a/src/mongo/shell/shardingtest.js b/src/mongo/shell/shardingtest.js
index a87e1bfc57b..e0fb313c1a0 100644
--- a/src/mongo/shell/shardingtest.js
+++ b/src/mongo/shell/shardingtest.js
@@ -1049,6 +1049,21 @@ var ShardingTest = function(params) {
         return true;
     }
 
+    /**
+     * Returns the total number of mongod nodes across all shards, excluding config server nodes.
+     * Used only for diagnostic logging.
+     */
+    function totalNumShardNodes(shardsAsReplSets) {
+        // Standalone mongod shards.
+        if (!shardsAsReplSets) {
+            return self._connections.length;
+        }
+
+        // Replica set shards.
+        const numNodesPerReplSet = self._rs.map(r => r.test.nodes.length);
+        return numNodesPerReplSet.reduce((a, b) => a + b, 0);
+    }
+
     // ShardingTest initialization
 
     assert(isObject(params), 'ShardingTest configuration must be a JSON object');
@@ -1208,6 +1223,9 @@ var ShardingTest = function(params) {
         randomSeedAlreadySet = true;
     }
 
+    // Should we start up shards as replica sets.
+    const shardsAsReplSets = (otherParams.rs || otherParams["rs" + i] || startShardsAsRS);
+
     // Start the MongoD servers (shards)
     let startTime = new Date();  // Measure the execution time of startup and initiate.
     for (var i = 0; i < numShards; i++) {
@@ -1449,7 +1467,8 @@ var ShardingTest = function(params) {
 
     rstOptions.nodes = nodeOptions;
 
-    startTime = new Date();  // Measure the execution time of config server startup and initiate.
+    const configServerStartTime =
+        new Date();  // Measure the execution time of config server startup and initiate.
 
     // Start the config server's replica set
     this.configRS = new ReplSetTest(rstOptions);
@@ -1467,7 +1486,12 @@ var ShardingTest = function(params) {
     var csrsPrimary = this.configRS.getPrimary();
 
     print("ShardingTest startup and initiation for the config server took " +
-          (new Date() - startTime) + "ms with " + this.configRS.nodeList().length + " nodes.");
+          (new Date() - configServerStartTime) + "ms with " + this.configRS.nodeList().length +
+          " nodes.");
+
+    print("ShardingTest startup and initiation for all nodes took " + (new Date() - startTime) +
+          "ms with " + this.configRS.nodeList().length + " config server nodes and " +
+          totalNumShardNodes(shardsAsReplSets) + " total shard nodes.");
 
     // If 'otherParams.mongosOptions.binVersion' is an array value, then we'll end up constructing a
     // version iterator.