summaryrefslogtreecommitdiff
path: root/jstests/sharding/migrateBig.js
blob: 5cdc03d292e8e63c65540851290f8152c12ee1fd (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.writeOK(
        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.writeOK(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.writeOK(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();
})();