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
|
/**
* Tests that the addShard process initializes sharding awareness on an added standalone or
* replica set shard that was started with --shardsvr.
*/
(function() {
"use strict";
var waitForIsMaster = function(conn) {
assert.soon(function() {
var res = conn.getDB('admin').runCommand({isMaster: 1});
return res.ismaster;
});
};
var checkShardingStateInitialized = function(conn, configConnStr, shardName, clusterId) {
var res = conn.getDB('admin').runCommand({shardingState: 1});
assert.commandWorked(res);
assert(res.enabled);
assert.eq(configConnStr, res.configServer);
assert.eq(shardName, res.shardName);
assert(clusterId.equals(res.clusterId),
'cluster id: ' + tojson(clusterId) + ' != ' + tojson(res.clusterId));
};
var checkShardMarkedAsShardAware = function(mongosConn, shardName) {
var res = mongosConn.getDB('config').getCollection('shards').findOne({_id: shardName});
assert.neq(null, res, "Could not find new shard " + shardName + " in config.shards");
assert.eq(1, res.state);
};
// Create the cluster to test adding shards to.
var st = new ShardingTest({shards: 1});
var clusterId = st.s.getDB('config').getCollection('version').findOne().clusterId;
// Add a shard that is a standalone mongod.
var standaloneConn = MongoRunner.runMongod({shardsvr: ''});
waitForIsMaster(standaloneConn);
jsTest.log("Going to add standalone as shard: " + standaloneConn);
var newShardName = "newShard";
assert.commandWorked(st.s.adminCommand({addShard: standaloneConn.name, name: newShardName}));
checkShardingStateInitialized(standaloneConn, st.configRS.getURL(), newShardName, clusterId);
checkShardMarkedAsShardAware(st.s, newShardName);
MongoRunner.stopMongod(standaloneConn);
// Add a shard that is a replica set.
var replTest = new ReplSetTest({nodes: 1});
replTest.startSet({shardsvr: ''});
replTest.initiate();
waitForIsMaster(replTest.getPrimary());
jsTest.log("Going to add replica set as shard: " + tojson(replTest));
assert.commandWorked(st.s.adminCommand({addShard: replTest.getURL(), name: replTest.getURL()}));
checkShardingStateInitialized(
replTest.getPrimary(), st.configRS.getURL(), replTest.getURL(), clusterId);
checkShardMarkedAsShardAware(st.s, newShardName);
replTest.stopSet();
st.stop();
})();
|
