summaryrefslogtreecommitdiff
path: root/jstests/sharding/sort1.js
blob: dc423ac14d772caf499674b89aaab73caaf6a69b (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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
(function() {
    'use strict';

    var s = new ShardingTest({name: "sort1", shards: 2, mongos: 2});

    s.adminCommand({enablesharding: "test"});
    s.ensurePrimaryShard('test', s.shard1.shardName);
    s.adminCommand({shardcollection: "test.data", key: {'sub.num': 1}});

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

    const N = 100;

    var forward = [];
    var backward = [];
    for (var i = 0; i < N; i++) {
        db.data.insert({_id: i, sub: {num: i, x: N - i}});
        forward.push(i);
        backward.push((N - 1) - i);
    }

    s.adminCommand({split: "test.data", middle: {'sub.num': 33}});
    s.adminCommand({split: "test.data", middle: {'sub.num': 66}});

    s.adminCommand({
        movechunk: "test.data",
        find: {'sub.num': 50},
        to: s.getOther(s.getPrimaryShard("test")).name,
        _waitForDelete: true
    });

    assert.lte(3, s.config.chunks.find({ns: 'test.data'}).itcount(), "A1");

    var temp = s.config.chunks.find({ns: 'test.data'}).sort({min: 1}).toArray();
    temp.forEach(printjsononeline);

    var z = 0;
    for (; z < temp.length; z++)
        if (temp[z].min["sub.num"] <= 50 && temp[z].max["sub.num"] > 50)
            break;

    assert.eq(temp[z - 1].shard, temp[z + 1].shard, "A2");
    assert.neq(temp[z - 1].shard, temp[z].shard, "A3");

    temp = db.data.find().sort({'sub.num': 1}).toArray();
    assert.eq(N, temp.length, "B1");
    for (i = 0; i < 100; i++) {
        assert.eq(i, temp[i].sub.num, "B2");
    }

    db.data.find().sort({'sub.num': 1}).toArray();
    s.getPrimaryShard("test").getDB("test").data.find().sort({'sub.num': 1}).toArray();

    var a = Date.timeFunc(function() {
        z = db.data.find().sort({'sub.num': 1}).toArray();
    }, 200);
    assert.eq(100, z.length, "C1");

    var b = 1.5 * Date.timeFunc(function() {
        z = s.getPrimaryShard("test").getDB("test").data.find().sort({'sub.num': 1}).toArray();
    }, 200);
    assert.eq(67, z.length, "C2");

    print("a: " + a + " b:" + b + " mongos slow down: " + Math.ceil(100 * ((a - b) / b)) + "%");

    // -- secondary index sorting

    function getSorted(by, dir, proj) {
        var s = {};
        s[by] = dir || 1;
        printjson(s);
        var cur = db.data.find({}, proj || {}).sort(s);
        return terse(cur.map(function(z) {
            return z.sub.num;
        }));
    }

    function terse(a) {
        var s = "";
        for (var i = 0; i < a.length; i++) {
            if (i > 0)
                s += ",";
            s += a[i];
        }
        return s;
    }

    forward = terse(forward);
    backward = terse(backward);

    assert.eq(forward, getSorted("sub.num", 1), "D1");
    assert.eq(backward, getSorted("sub.num", -1), "D2");

    assert.eq(backward, getSorted("sub.x", 1), "D3");
    assert.eq(forward, getSorted("sub.x", -1), "D4");

    assert.eq(backward, getSorted("sub.x", 1, {'sub.num': 1}), "D5");
    assert.eq(forward, getSorted("sub.x", -1, {'sub.num': 1}), "D6");

    assert.eq(backward, getSorted("sub.x", 1, {'sub': 1}), "D7");
    assert.eq(forward, getSorted("sub.x", -1, {'sub': 1}), "D8");

    assert.eq(backward, getSorted("sub.x", 1, {'_id': 0}), "D9");
    assert.eq(forward, getSorted("sub.x", -1, {'_id': 0}), "D10");

    assert.eq(backward, getSorted("sub.x", 1, {'_id': 0, 'sub.num': 1}), "D11");
    assert.eq(forward, getSorted("sub.x", -1, {'_id': 0, 'sub.num': 1}), "D12");

    s.stop();
})();