summaryrefslogtreecommitdiff
path: root/jstests/sharding/query/geo_near_sharded.js
blob: d68dcdec140a2f72a7ccfd2e3c3cdedc93d50aae (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
// SERVER-7906
(function() {
'use strict';

var coll = 'points';

function test(st, db, sharded, indexType) {
    printjson(db);

    if (sharded) {
        var shards = [st.shard0, st.shard1, st.shard2];

        assert.commandWorked(
            st.s0.adminCommand({shardCollection: db[coll].getFullName(), key: {rand: 1}}));
        for (var i = 1; i < 10; i++) {
            // split at 0.1, 0.2, ... 0.9
            assert.commandWorked(
                st.s0.adminCommand({split: db[coll].getFullName(), middle: {rand: i / 10}}));
            assert.commandWorked(st.s0.adminCommand({
                moveChunk: db[coll].getFullName(),
                find: {rand: i / 10},
                to: shards[i % shards.length].shardName
            }));
        }

        var config = db.getSiblingDB("config");
        assert.eq(config.chunks.count({'ns': db[coll].getFullName()}), 10);
    }

    Random.setRandomSeed();

    var bulk = db[coll].initializeUnorderedBulkOp();
    var numPts = 10 * 1000;
    for (var i = 0; i < numPts; i++) {
        var lat = 90 - Random.rand() * 180;
        var lng = 180 - Random.rand() * 360;
        bulk.insert({rand: Math.random(), loc: [lng, lat]});
    }
    assert.commandWorked(bulk.execute());
    assert.eq(db[coll].count(), numPts);

    assert.commandWorked(db[coll].createIndex({loc: indexType}));

    let res = assert.commandWorked(db.runCommand({
        aggregate: coll,
        cursor: {},
        pipeline: [{
            $geoNear: {
                near: [0, 0],
                spherical: true,
                includeLocs: "match",
                distanceField: "dist",
            }
        }]
    }),
                                   tojson({sharded: sharded, indexType: indexType}));
    assert.gt(res.cursor.firstBatch.length, 0, tojson(res));
}

var st = new ShardingTest({shards: 3, mongos: 1});
assert.commandWorked(st.s0.adminCommand({enablesharding: "test"}));
st.ensurePrimaryShard('test', st.shard1.shardName);

test(st, st.getDB('test'), true, '2dsphere');
st.stop();
})();