1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
/**
* @tags: [
* multiversion_incompatible,
* ]
*/
// Checking UUID and index consistency involves talking to a shard node, which in this
// test is shutdown.
TestData.skipCheckingUUIDsConsistentAcrossCluster = true;
TestData.skipCheckingIndexesConsistentAcrossCluster = true;
TestData.skipCheckOrphans = true;
(function() {
'use strict';
function runTest(setParams, connPoolStatsCheck, extraOptions) {
const test = new ShardingTest({shards: 2, mongosOptions: setParams});
var db = test.getDB("test");
const shardCommand = {shardcollection: "test.foo", key: {num: 1}};
// shard
assert.commandWorked(test.s0.adminCommand({enablesharding: "test"}));
assert.commandWorked(test.s0.adminCommand(shardCommand));
var primary;
var pId;
if (extraOptions !== undefined) {
const resp = extraOptions(test);
primary = resp.connString;
pId = resp.nodeId;
}
test.restartMongos(0);
db = test.getDB("admin");
const shardDocs = db.runCommand({listShards: 1})["shards"];
assert.soon(() => {
let connPoolStats = db.runCommand({connPoolStats: 1});
for (const shardDoc of shardDocs) {
let currentShard = shardDoc["host"].split("/")[1];
if (!connPoolStatsCheck(connPoolStats, currentShard, primary)) {
return false;
}
}
return true;
});
if (extraOptions !== undefined) {
test.rs0.restart(pId);
}
test.stop();
}
jsTest.log("Tests basic warm up of connection pool");
var testWarmUpParams = {};
var testWarmUpConnPoolStatsCheck = function(connPoolStats, currentShard) {
return connPoolStats["hosts"][currentShard]["inUse"] +
connPoolStats["hosts"][currentShard]["available"] +
connPoolStats["hosts"][currentShard]["refreshing"] >=
1;
};
runTest(testWarmUpParams, testWarmUpConnPoolStatsCheck);
jsTest.log("Tests does not warm up connection pool when parameter is disabled");
var warmUpDisabledParams = {
setParameter: {warmMinConnectionsInShardingTaskExecutorPoolOnStartup: false}
};
var warmUpDisabledConnPoolStatsCheck = function(connPoolStats, currentShard) {
return undefined === connPoolStats["hosts"][currentShard];
};
runTest(warmUpDisabledParams, warmUpDisabledConnPoolStatsCheck);
jsTest.log("Tests establishes more connections when parameter is set.");
// Increase the amount of time to establish more connections to avoid timing out
// before establishing the expected number of connections.
var biggerPoolSizeParams = {
setParameter: {
ShardingTaskExecutorPoolMinSize: 3,
warmMinConnectionsInShardingTaskExecutorPoolOnStartupWaitMS: 60000
}
};
var biggerPoolSizeConnPoolStatsCheck = function(connPoolStats, currentShard) {
return connPoolStats["hosts"][currentShard]["inUse"] +
connPoolStats["hosts"][currentShard]["available"] +
connPoolStats["hosts"][currentShard]["refreshing"] >=
3;
};
runTest(biggerPoolSizeParams, biggerPoolSizeConnPoolStatsCheck);
jsTest.log("Tests establishes connections it can and ignores the ones it can't");
var shutdownNodeParams = {};
var shutdownNodeConnPoolStatsCheck = function(connPoolStats, currentShard, primary) {
if (currentShard == primary) {
return connPoolStats["hosts"][currentShard]["inUse"] +
connPoolStats["hosts"][currentShard]["available"] +
connPoolStats["hosts"][currentShard]["refreshing"] ===
0;
}
return connPoolStats["hosts"][currentShard]["inUse"] +
connPoolStats["hosts"][currentShard]["available"] +
connPoolStats["hosts"][currentShard]["refreshing"] ===
1;
};
var shutdownNodeExtraOptions = function(test) {
const nodeList = test.rs0.nodeList();
const primary = test.rs0.getPrimary();
var pId = test.rs0.getNodeId(primary);
test.rs0.stop(pId);
return {connString: nodeList[pId], nodeId: pId};
};
runTest(shutdownNodeParams, shutdownNodeConnPoolStatsCheck, shutdownNodeExtraOptions);
})();
|
