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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
// server-7781 $geoNear pipeline stage
load('jstests/libs/geo_near_random.js');
load('jstests/aggregation/extras/utils.js');
var coll = 'server7781';
db[coll].drop();
db[coll].insert({loc:[0,0]});
// $geoNear is only allowed as the first stage in a pipeline, nowhere else.
assertErrorCode(db[coll],
[{$match: {x:1}}, {$geoNear:{near: [1,1], spherical: true, distanceField: 'dis'}}],
16602);
function checkOutput(cmdOut, aggOut, expectedNum) {
assert.commandWorked(cmdOut, "geoNear command");
// the output arrays are accessed differently
cmdOut = cmdOut.results;
aggOut = aggOut.toArray();
assert.eq(cmdOut.length, expectedNum);
assert.eq(aggOut.length, expectedNum);
var allSame = true;
var massaged; // massage geoNear command output to match output from agg pipeline
for (var i=0; i < cmdOut.length; i++) {
massaged = {};
Object.extend(massaged, cmdOut[i].obj, /*deep=*/true)
massaged.stats = {'dis': cmdOut[i].dis,
'loc': cmdOut[i].loc};
if (!friendlyEqual(massaged, aggOut[i])) {
allSame = false; // don't bail yet since we want to print all differences
print("Difference detected at index " + i + " of " + expectedNum);
print("from geoNear command:" + tojson(massaged));
print("from aggregate command:" + tojson(aggOut[i]));
}
}
assert(allSame);
}
// We use this to generate points. Using a single global to avoid reseting RNG in each pass.
var pointMaker = new GeoNearRandomTest(coll);
function test(db, sharded, indexType) {
db[coll].drop();
if (sharded) { // sharded setup
var shards = [];
var config = db.getSiblingDB("config");
config.shards.find().forEach(function(shard) { shards.push(shard._id) });
db.adminCommand({shardCollection: db[coll].getFullName(), key: {rand:1}});
for (var i=1; i < 10; i++) {
// split at 0.1, 0.2, ... 0.9
db.adminCommand({split: db[coll].getFullName(), middle: {rand: i/10}});
db.adminCommand({moveChunk: db[coll].getFullName(), find: {rand: i/10},
to: shards[i%shards.length]});
}
assert.eq(config.chunks.count({'ns': db[coll].getFullName()}), 10);
}
// insert points
var numPts = 10*1000;
var bulk = db[coll].initializeUnorderedBulkOp();
for (var i=0; i < numPts; i++) {
bulk.insert({ rand: Math.random(), loc: pointMaker.mkPt() });
}
assert.writeOK(bulk.execute());
assert.eq(db[coll].count(), numPts);
db[coll].ensureIndex({loc: indexType});
// test with defaults
var queryPoint = pointMaker.mkPt(0.25); // stick to center of map
geoCmd = {geoNear: coll, near: queryPoint, includeLocs: true, spherical: true};
aggCmd = {$geoNear: {near: queryPoint, includeLocs: 'stats.loc', distanceField: 'stats.dis', spherical: true}};
checkOutput(db.runCommand(geoCmd), db[coll].aggregate(aggCmd), 100);
// test with num
queryPoint = pointMaker.mkPt(0.25);
geoCmd.num = 75;
geoCmd.near = queryPoint;
aggCmd.$geoNear.num = 75;
aggCmd.$geoNear.near = queryPoint;
checkOutput(db.runCommand(geoCmd), db[coll].aggregate(aggCmd), 75);
// test with limit instead of num (they mean the same thing, but want to test both)
queryPoint = pointMaker.mkPt(0.25);
geoCmd.near = queryPoint;
delete geoCmd.num;
geoCmd.limit = 70;
aggCmd.$geoNear.near = queryPoint;
delete aggCmd.$geoNear.num;
aggCmd.$geoNear.limit = 70;
checkOutput(db.runCommand(geoCmd), db[coll].aggregate(aggCmd), 70);
// test spherical
queryPoint = pointMaker.mkPt(0.25);
geoCmd.spherical = true;
geoCmd.near = queryPoint;
aggCmd.$geoNear.spherical = true;
aggCmd.$geoNear.near = queryPoint;
checkOutput(db.runCommand(geoCmd), db[coll].aggregate(aggCmd), 70);
// test $geoNear + $limit coalescing
queryPoint = pointMaker.mkPt(0.25);
geoCmd.num = 40;
geoCmd.near = queryPoint;
aggCmd.$geoNear.near = queryPoint;
aggArr = [aggCmd, {$limit: 50}, {$limit:60}, {$limit:40}];
checkOutput(db.runCommand(geoCmd), db[coll].aggregate(aggArr), 40);
}
test(db, false, '2d');
test(db, false, '2dsphere');
var sharded = new ShardingTest({shards: 3, verbose: 0, mongos: 1});
sharded.stopBalancer();
sharded.adminCommand( { enablesharding : "test" } );
test(sharded.getDB('test'), true, '2d');
test(sharded.getDB('test'), true, '2dsphere');
sharded.stop();
|
