summaryrefslogtreecommitdiff
path: root/jstests/sharding/shard_targeting.js
blob: f1d52768f3cb4c93afa910c27fc864ad9afd76f2 (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
// SERVER-13702
// Some commands allow optional query, e.g. count, mapreduce.
// If the optional query is not given, mongos will wrongly use the command
// BSONObj itself as the query to target shards, which could return wrong
// shards if the shard key happens to be one of the fields in the command object.
(function() {
    'use strict';

    var s = new ShardingTest({shards: 2});
    assert.commandWorked(s.s0.adminCommand({enablesharding: "test"}));
    s.ensurePrimaryShard('test', 'shard0001');

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

    //
    // Target count command
    //

    // Shard key is the same with command name.
    s.shardColl("foo", {count: 1}, {count: ""});

    for (var i = 0; i < 50; i++) {
        db.foo.insert({count: i});       // chunk [MinKey, ""), including numbers
        db.foo.insert({count: "" + i});  // chunk ["", MaxKey]
    }

    var theOtherShard = s.getOther(s.getPrimaryShard("test")).name;
    s.printShardingStatus();

    // Count documents on both shards

    // "count" commnad with "query" option { }.
    assert.eq(db.foo.count(), 100);
    // Optional "query" option is not given.
    res = db.foo.runCommand("count");
    assert.eq(res.n, 100);

    //
    // Target mapreduce command
    //
    db.foo.drop();

    // Shard key is the same with command name.
    s.shardColl("foo", {mapReduce: 1}, {mapReduce: ""});

    for (var i = 0; i < 50; i++) {
        db.foo.insert({mapReduce: i});       // to the chunk including number
        db.foo.insert({mapReduce: "" + i});  // to the chunk including string
    }

    s.printShardingStatus();

    function m() {
        emit("total", 1);
    }
    function r(k, v) {
        return Array.sum(v);
    }
    res = db.foo.runCommand({mapReduce: "foo", map: m, reduce: r, out: {inline: 1}});

    // Count documents on both shards
    assert.eq(res.results[0].value, 100);

    s.stop();

})();