summaryrefslogtreecommitdiff
path: root/jstests/sharding/migrateBig.js
blob: 60306797dfda9f3b4ff6a7c3333d628fc09e5bbe (plain)
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
(function() {
'use strict';

var s = new ShardingTest({name: "migrateBig", shards: 2, other: {chunkSize: 1}});

assert.commandWorked(
    s.config.settings.update({_id: "balancer"}, {$set: {_waitForDelete: true}}, true));
assert.commandWorked(s.s0.adminCommand({enablesharding: "test"}));
s.ensurePrimaryShard('test', s.shard1.shardName);
assert.commandWorked(s.s0.adminCommand({shardcollection: "test.foo", key: {x: 1}}));

var db = s.getDB("test");
var coll = db.foo;

var big = "";
while (big.length < 10000)
    big += "eliot";

var bulk = coll.initializeUnorderedBulkOp();
for (var x = 0; x < 100; x++) {
    bulk.insert({x: x, big: big});
}
assert.commandWorked(bulk.execute());

assert.commandWorked(s.s0.adminCommand({split: "test.foo", middle: {x: 30}}));
assert.commandWorked(s.s0.adminCommand({split: "test.foo", middle: {x: 66}}));
assert.commandWorked(s.s0.adminCommand(
    {movechunk: "test.foo", find: {x: 90}, to: s.getOther(s.getPrimaryShard("test")).name}));

s.printShardingStatus();

print("YO : " + s.getPrimaryShard("test").host);
var direct = new Mongo(s.getPrimaryShard("test").host);
print("direct : " + direct);

var directDB = direct.getDB("test");

for (var done = 0; done < 2 * 1024 * 1024; done += big.length) {
    assert.commandWorked(directDB.foo.insert({x: 50 + Math.random(), big: big}));
}

s.printShardingStatus();

// This is a large chunk, which should not be able to move
assert.commandFailed(s.s0.adminCommand(
    {movechunk: "test.foo", find: {x: 50}, to: s.getOther(s.getPrimaryShard("test")).name}));

for (var i = 0; i < 20; i += 2) {
    try {
        assert.commandWorked(s.s0.adminCommand({split: "test.foo", middle: {x: i}}));
    } catch (e) {
        // We may have auto split on some of these, which is ok
        print(e);
    }
}

s.printShardingStatus();

s.startBalancer();

assert.soon(function() {
    var x = s.chunkDiff("foo", "test");
    print("chunk diff: " + x);
    return x < 2;
}, "no balance happened", 8 * 60 * 1000, 2000);

s.stop();
})();