99 lines
2.9 KiB
JavaScript
99 lines
2.9 KiB
JavaScript
// 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');
|
|
})();
|