summaryrefslogtreecommitdiff
path: root/jstests/slow1/sharding_multiple_collections.js
blob: 46ade14a0add70fb58d579e0cbff7e28a508508a (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
(function() {

    var s = new ShardingTest({
        name: "multcollections",
        shards: 2,
        mongos: 1,
        verbose: 1,
        other: {chunkSize: 1, enableBalancer: true}
    });

    s.adminCommand({enablesharding: "test"});
    db = s.getDB("test");
    s.ensurePrimaryShard('test', 'shard0001');

    N = 100000;

    S = "";
    while (S.length < 500)
        S += "123123312312";

    var bulk = db.foo.initializeUnorderedBulkOp();
    var bulk2 = db.bar.initializeUnorderedBulkOp();
    for (i = 0; i < N; i++) {
        bulk.insert({_id: i, s: S});
        bulk2.insert({_id: i, s: S, s2: S});
    }
    assert.writeOK(bulk.execute());
    assert.writeOK(bulk2.execute());

    s.printShardingStatus();

    function mytest(coll, i, loopNumber) {
        x = coll.find({_id: i}).explain();
        if (x)
            return;
        throw Error("can't find " + i + " in " + coll.getName() + " on loopNumber: " + loopNumber +
                    " explain: " + tojson(x));
    }

    loopNumber = 0;
    while (1) {
        for (i = 0; i < N; i++) {
            mytest(db.foo, i, loopNumber);
            mytest(db.bar, i, loopNumber);
            if (i % 1000 == 0)
                print(i);
        }
        s.printShardingStatus();
        loopNumber++;

        if (loopNumber == 1) {
            s.adminCommand({shardcollection: "test.foo", key: {_id: 1}});
            s.adminCommand({shardcollection: "test.bar", key: {_id: 1}});
        }

        assert(loopNumber < 1000, "taking too long");

        if (s.chunkDiff("foo") < 12 && s.chunkDiff("bar") < 12)
            break;
    }

    s.stop();

})();