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
|
// Tests changing the zones on a shard at runtime results in a correct distribution of chunks across
// the cluster
(function() {
'use strict';
var st = new ShardingTest({shards: 3, mongos: 1});
assert.commandWorked(st.s0.adminCommand({enablesharding: 'test'}));
st.ensurePrimaryShard('test', st.shard1.shardName);
var testDB = st.s0.getDB('test');
var configDB = st.s0.getDB('config');
var bulk = testDB.foo.initializeUnorderedBulkOp();
for (var i = 0; i < 9; i++) {
bulk.insert({_id: i, x: i});
}
assert.commandWorked(bulk.execute());
assert.commandWorked(st.s0.adminCommand({shardCollection: 'test.foo', key: {_id: 1}}));
// Produce 9 chunks with min value at the documents just inserted
for (var i = 0; i < 8; i++) {
assert.commandWorked(st.s0.adminCommand({split: 'test.foo', middle: {_id: i}}));
}
/**
* Waits for the balancer state described by the checking logic function (checkFunc) to be
* reached and ensures that it does not change from that state at the next balancer round.
*/
function assertBalanceCompleteAndStable(checkFunc, stepName) {
st.printShardingStatus(true);
assert.soon(checkFunc, 'Balance at step ' + stepName + ' did not happen', 3 * 60 * 1000, 2000);
st.waitForBalancer(true, 60000);
st.printShardingStatus(true);
assert(checkFunc());
jsTestLog('Completed step ' + stepName);
}
/**
* Checker function to be used with assertBalanceCompleteAndStable, which ensures that the
* cluster is evenly balanced.
*/
function checkClusterEvenlyBalanced() {
var maxChunkDiff = st.chunkDiff('foo', 'test');
return maxChunkDiff <= 1;
}
st.startBalancer();
// Initial balance
assertBalanceCompleteAndStable(checkClusterEvenlyBalanced, 'initial');
// Spread chunks correctly across zones
st.addShardTag(st.shard0.shardName, 'a');
st.addShardTag(st.shard1.shardName, 'a');
st.addTagRange('test.foo', {_id: -100}, {_id: 100}, 'a');
st.addShardTag(st.shard2.shardName, 'b');
st.addTagRange('test.foo', {_id: MinKey}, {_id: -100}, 'b');
st.addTagRange('test.foo', {_id: 100}, {_id: MaxKey}, 'b');
assertBalanceCompleteAndStable(function() {
var chunksOnShard2 =
configDB.chunks.find({ns: 'test.foo', shard: st.shard2.shardName}).sort({min: 1}).toArray();
jsTestLog('Chunks on shard2: ' + tojson(chunksOnShard2));
if (chunksOnShard2.length != 2) {
return false;
}
return chunksOnShard2[0].min._id == MinKey && chunksOnShard2[0].max._id == -100 &&
chunksOnShard2[1].min._id == 100 && chunksOnShard2[1].max._id == MaxKey;
}, 'chunks to zones a and b');
// Tag the entire collection to shard0 and wait for everything to move to that shard
st.removeTagRange('test.foo', {_id: -100}, {_id: 100}, 'a');
st.removeTagRange('test.foo', {_id: MinKey}, {_id: -100}, 'b');
st.removeTagRange('test.foo', {_id: 100}, {_id: MaxKey}, 'b');
st.removeShardTag(st.shard1.shardName, 'a');
st.removeShardTag(st.shard2.shardName, 'b');
st.addTagRange('test.foo', {_id: MinKey}, {_id: MaxKey}, 'a');
assertBalanceCompleteAndStable(function() {
var counts = st.chunkCounts('foo');
printjson(counts);
return counts[st.shard0.shardName] == 11 && counts[st.shard1.shardName] == 0 &&
counts[st.shard2.shardName] == 0;
}, 'all chunks to zone a');
// Remove all zones and ensure collection is correctly redistributed
st.removeShardTag(st.shard0.shardName, 'a');
st.removeTagRange('test.foo', {_id: MinKey}, {_id: MaxKey}, 'a');
assertBalanceCompleteAndStable(checkClusterEvenlyBalanced, 'final');
st.stop();
})();
|
