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
|
// Validates that auto rebalancing works and reaches quiescent state when replica sets are used as
// shards
(function() {
'use strict';
var st = new ShardingTest({ name: 'auto_rebalance_rs',
mongos: 1,
shards: 2,
chunksize: 1,
rs: {
nodes: 3
}
});
assert.writeOK(st.getDB( "config" ).settings.update(
{ _id: "balancer" },
{ $set: { "_secondaryThrottle" : false } },
{ upsert: true }));
st.getDB("admin").runCommand({enableSharding : "TestDB_auto_rebalance_rs"});
st.getDB("admin").runCommand({shardCollection : "TestDB_auto_rebalance_rs.foo", key : {x : 1}});
var dbTest = st.getDB("TestDB_auto_rebalance_rs");
var num = 100000;
var bulk = dbTest.foo.initializeUnorderedBulkOp();
for (var i = 0; i < num; i++) {
bulk.insert({ _id: i, x: i, abc: "defg", date: new Date(), str: "all the talk on the market" });
}
assert.writeOK(bulk.execute());
// Wait for the rebalancing to kick in
st.startBalancer(60000);
assert.soon(function() {
var s1Chunks = st.getDB("config").chunks.count({shard : "auto_rebalance_rs-rs0"});
var s2Chunks = st.getDB("config").chunks.count({shard : "auto_rebalance_rs-rs1"});
var total = st.getDB("config").chunks.count({ns : "TestDB_auto_rebalance_rs.foo"});
print("chunks: " + s1Chunks + " " + s2Chunks + " " + total);
return s1Chunks > 0 && s2Chunks > 0 && (s1Chunks + s2Chunks == total);
},
"Chunks failed to balance",
60000,
5000);
// Ensure the range deleter quiesces
st.rs0.awaitReplication(120000);
st.rs1.awaitReplication(120000);
st.stop();
})();
|
