summaryrefslogtreecommitdiff
path: root/jstests/sharding/migrateBig.js
blob: 01260123b67d69d4be8aadab1c197dd3e8dd49b2 (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
69
70
71
72
73
74
75
76
77
(function() {

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

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

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

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

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

    s.printShardingStatus();

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

    s.printShardingStatus();

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

    directDB = direct.getDB("test");

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

    s.printShardingStatus();

    assert.throws(function() {
        s.adminCommand(
            {movechunk: "test.foo", find: {x: 50}, to: s.getOther(s.getPrimaryShard("test")).name});
    }, [], "move should fail");

    for (i = 0; i < 20; i += 2) {
        try {
            s.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.config.settings.update({_id: "balancer"}, {$set: {stopped: false}}, true);

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

    assert.soon(function() {
        return !s.isAnyBalanceInFlight();
    });

    assert.eq(coll.count(), coll.find().itcount());

    s.stop();

})();