summaryrefslogtreecommitdiff
path: root/jstests/sharding/limit_push.js
blob: b681f328298b22e76cbfd2aa69559e3970241c21 (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
// This test is to ensure that limit() clauses are pushed down to the shards and evaluated
// See: http://jira.mongodb.org/browse/SERVER-1896
(function() {

    var s = new ShardingTest({name: "limit_push", shards: 2, mongos: 1});
    var db = s.getDB("test");

    // Create some data
    for (i = 0; i < 100; i++) {
        db.limit_push.insert({_id: i, x: i});
    }
    db.limit_push.ensureIndex({x: 1});
    assert.eq(100, db.limit_push.find().length(), "Incorrect number of documents");

    // Shard the collection
    s.adminCommand({enablesharding: "test"});
    s.ensurePrimaryShard('test', 'shard0001');
    s.adminCommand({shardcollection: "test.limit_push", key: {x: 1}});

    // Now split the and move the data between the shards
    s.adminCommand({split: "test.limit_push", middle: {x: 50}});
    s.adminCommand({
        moveChunk: "test.limit_push",
        find: {x: 51},
        to: s.getOther(s.getPrimaryShard("test")).name,
        _waitForDelete: true
    });

    // Check that the chunck have split correctly
    assert.eq(2, s.config.chunks.count(), "wrong number of chunks");

    // The query is asking for the maximum value below a given value
    // db.limit_push.find( { x : { $lt : 60} } ).sort( { x:-1} ).limit(1)
    q = {x: {$lt: 60}};

    // Make sure the basic queries are correct
    assert.eq(60, db.limit_push.find(q).count(), "Did not find 60 documents");
    // rs = db.limit_push.find( q ).sort( { x:-1} ).limit(1)
    // assert.eq( rs , { _id : "1" , x : 59 } , "Did not find document with value 59" );

    // Now make sure that the explain shos that each shard is returning a single document as
    // indicated
    // by the "n" element for each shard
    exp = db.limit_push.find(q).sort({x: -1}).limit(1).explain("executionStats");
    printjson(exp);

    var execStages = exp.executionStats.executionStages;
    assert.eq("SHARD_MERGE_SORT", execStages.stage, "Expected SHARD_MERGE_SORT as root stage");

    var k = 0;
    for (var j in execStages.shards) {
        assert.eq(1,
                  execStages.shards[j].executionStages.nReturned,
                  "'n' is not 1 from shard000" + k.toString());
        k++;
    }

    s.stop();

})();