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
|
// SERVER-7781 $geoNear pipeline stage
// @tags: [
// sbe_incompatible,
// ]
(function() {
'use strict';
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.
assert.throws(
() => db[coll].aggregate(
[{$match: {x: 1}}, {$geoNear: {near: [1, 1], spherical: true, distanceField: 'dis'}}]));
const kDistanceField = "dis";
const kIncludeLocsField = "loc";
/**
* Tests the output of the $geoNear command. This function expects a document with the following
* fields:
* - 'geoNearSpec' is the specification for a $geoNear aggregation stage.
* - 'limit' is an integer limiting the number of pipeline results.
* - 'batchSize', if specified, is the batchSize to use for the aggregation.
*/
function testGeoNearStageOutput({geoNearSpec, limit, batchSize}) {
const aggOptions = batchSize ? {batchSize: batchSize} : {};
const result =
db[coll].aggregate([{$geoNear: geoNearSpec}, {$limit: limit}], aggOptions).toArray();
const errmsg = () => tojson(result);
// Verify that we got the expected number of results.
assert.eq(result.length, limit, errmsg);
// Run though the array, checking for proper sort order and sane computed distances.
result.reduce((lastDist, curDoc) => {
const curDist = curDoc[kDistanceField];
// Verify that distances are in increasing order.
assert.lte(lastDist, curDist, errmsg);
// Verify that the computed distance is correct.
const computed = Geo.sphereDistance(geoNearSpec["near"], curDoc[kIncludeLocsField]);
assert.close(computed, curDist, errmsg);
return curDist;
}, 0);
}
// 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, indexType) {
db[coll].drop();
// 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.commandWorked(bulk.execute());
assert.eq(db[coll].count(), numPts);
db[coll].ensureIndex({loc: indexType});
// Test $geoNear with spherical coordinates.
testGeoNearStageOutput({
geoNearSpec: {
near: pointMaker.mkPt(0.25),
distanceField: kDistanceField,
includeLocs: kIncludeLocsField,
spherical: true,
},
limit: 100
});
// Test $geoNear with an initial batchSize of 1.
testGeoNearStageOutput({
geoNearSpec: {
near: pointMaker.mkPt(0.25),
distanceField: kDistanceField,
includeLocs: kIncludeLocsField,
spherical: true,
},
limit: 70,
batchSize: 1
});
}
test(db, '2d');
test(db, '2dsphere');
})();
|
