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
|
// Test balancing all chunks to one shard by tagging the full shard-key range on that collection
var s = new ShardingTest(
{name: "balance_tags2", shards: 3, mongos: 1, other: {chunkSize: 1, enableBalancer: true}});
s.adminCommand({enablesharding: "test"});
s.ensurePrimaryShard('test', 'shard0001');
var db = s.getDB("test");
var bulk = db.foo.initializeUnorderedBulkOp();
for (i = 0; i < 21; i++) {
bulk.insert({_id: i, x: i});
}
assert.writeOK(bulk.execute());
sh.shardCollection("test.foo", {_id: 1});
sh.stopBalancer();
for (i = 0; i < 20; i++) {
sh.splitAt("test.foo", {_id: i});
}
sh.startBalancer();
sh.status(true);
// Wait for the initial balance to happen
assert.soon(function() {
var counts = s.chunkCounts("foo");
printjson(counts);
return counts["shard0000"] == 7 && counts["shard0001"] == 7 && counts["shard0002"] == 7;
}, "balance 1 didn't happen", 1000 * 60 * 10, 1000);
// Tag one shard
sh.addShardTag("shard0000", "a");
assert.eq(["a"], s.config.shards.findOne({_id: "shard0000"}).tags);
// Tag the whole collection (ns) to one shard
sh.addTagRange("test.foo", {_id: MinKey}, {_id: MaxKey}, "a");
// Wait for things to move to that one shard
sh.status(true);
assert.soon(function() {
var counts = s.chunkCounts("foo");
printjson(counts);
return counts["shard0001"] == 0 && counts["shard0002"] == 0;
}, "balance 2 didn't happen", 1000 * 60 * 10, 1000);
printjson(sh.status());
s.stop();
|