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
|
/**
* Tests that $geoNear allows for variable in the 'near' argument.
* $geoNear is not allowed in a facet, even in a lookup.
* @tags: [
* requires_pipeline_optimization,
* do_not_wrap_aggregations_in_facets,
* ]
*/
(function() {
"use strict";
load("jstests/libs/fixture_helpers.js");
const collName = jsTest.name();
const coll = db[collName];
coll.drop();
assert.commandWorked(coll.insert({_id: 0, location: {type: "Point", coordinates: [10, 20]}}));
assert.commandWorked(coll.insert({_id: 1, location: {type: "Point", coordinates: [8, 22]}}));
assert.commandWorked(coll.createIndex({location: "2dsphere"}));
/**
* Compare pipelines with constant and variable 'near' argument.
*/
function compareResults(nearArgument) {
const pipeline = [{$geoNear: {near: "$$pt", distanceField: "distance"}}];
const res = coll.aggregate(pipeline, {let : {pt: nearArgument}}).toArray();
const constRes =
coll.aggregate([{$geoNear: {near: nearArgument, distanceField: "distance"}}]).toArray();
assert.eq(res.length, constRes.length);
for (let i = 0; i < res.length; i++) {
assert.eq(res[i].distance, constRes[i].distance);
}
}
compareResults({type: "Point", coordinates: [10, 22]});
// Let variable in $lookup pipeline.
const geo2 = db.geonear_let2;
geo2.drop();
assert.commandWorked(geo2.insert({_id: 5, location: {type: "Point", coordinates: [10, 21]}}));
assert.commandWorked(geo2.insert({_id: 6, location: {type: "Point", coordinates: [11, 21]}}));
let pipelineLookup = [
{$lookup: {
from: coll.getName(),
let: {pt: "$location"},
pipeline: [{$geoNear: {near: "$$pt", distanceField: "distance"}}, {$match: {_id: 0}}],
as: "joinedField"
}}];
const lookupRes = geo2.aggregate(pipelineLookup).toArray();
assert.eq(lookupRes.length, 2);
// Make sure the computed distance uses the location field in the current document in the outer
// collection.
assert.neq(lookupRes[0].joinedField[0].distance, lookupRes[1].joinedField[0].distance);
// With legacy geometry.
coll.drop();
assert.commandWorked(coll.insert({_id: 0, location: [10, 20]}));
assert.commandWorked(coll.insert({_id: 1, location: [8, 22]}));
coll.createIndex({location: "2d"});
compareResults([10, 22]);
// Error checks.
let err = assert.throws(
() => coll.aggregate([{$geoNear: {near: "$$pt", distanceField: "distance"}}], {let : {pt: 5}}));
assert.commandFailedWithCode(err, 5860401);
err = assert.throws(() => coll.aggregate([{$geoNear: {near: "$$pt", distanceField: "distance"}}],
{let : {pt: "abc"}}));
assert.commandFailedWithCode(err, 5860401);
}());
|
