107 lines
3.9 KiB
JavaScript
107 lines
3.9 KiB
JavaScript
// Tests distinct with geospatial field values.
|
|
// 1. Test distinct with geo values for 'key' (SERVER-2135)
|
|
// 2. Test distinct with geo predicates for 'query' (SERVER-13769)
|
|
|
|
var coll = db.geo_distinct;
|
|
var res;
|
|
|
|
//
|
|
// 1. Test distinct with geo values for 'key'.
|
|
//
|
|
|
|
coll.drop();
|
|
coll.insert({loc: {type: 'Point', coordinates: [10, 20]}});
|
|
coll.insert({loc: {type: 'Point', coordinates: [10, 20]}});
|
|
coll.insert({loc: {type: 'Point', coordinates: [20, 30]}});
|
|
coll.insert({loc: {type: 'Point', coordinates: [20, 30]}});
|
|
assert.eq(4, coll.count());
|
|
|
|
// Test distinct on GeoJSON points with/without a 2dsphere index.
|
|
|
|
res = coll.runCommand('distinct', {key: 'loc'});
|
|
assert.commandWorked(res);
|
|
assert.eq(res.values.sort(),
|
|
[{type: 'Point', coordinates: [10, 20]}, {type: 'Point', coordinates: [20, 30]}]);
|
|
|
|
assert.commandWorked(coll.ensureIndex({loc: '2dsphere'}));
|
|
|
|
res = coll.runCommand('distinct', {key: 'loc'});
|
|
assert.commandWorked(res);
|
|
assert.eq(res.values.sort(),
|
|
[{type: 'Point', coordinates: [10, 20]}, {type: 'Point', coordinates: [20, 30]}]);
|
|
|
|
// Test distinct on legacy points with/without a 2d index.
|
|
|
|
// (Note that distinct on a 2d-indexed field doesn't produce a list of coordinate pairs, since
|
|
// distinct logically operates on unique values in an array. Hence, the results are unintuitive and
|
|
// not semantically meaningful.)
|
|
|
|
coll.dropIndexes();
|
|
|
|
res = coll.runCommand('distinct', {key: 'loc.coordinates'});
|
|
assert.commandWorked(res);
|
|
assert.eq(res.values.sort(), [10, 20, 30]);
|
|
|
|
assert.commandWorked(coll.ensureIndex({'loc.coordinates': '2d'}));
|
|
|
|
res = coll.runCommand('distinct', {key: 'loc.coordinates'});
|
|
assert.commandWorked(res);
|
|
assert.eq(res.values.sort(), [10, 20, 30]);
|
|
|
|
//
|
|
// 2. Test distinct with geo predicates for 'query'.
|
|
//
|
|
|
|
coll.drop();
|
|
for (var i = 0; i < 50; ++i) {
|
|
coll.insert({zone: 1, loc: {type: 'Point', coordinates: [-20, -20]}});
|
|
coll.insert({zone: 2, loc: {type: 'Point', coordinates: [-10, -10]}});
|
|
coll.insert({zone: 3, loc: {type: 'Point', coordinates: [0, 0]}});
|
|
coll.insert({zone: 4, loc: {type: 'Point', coordinates: [10, 10]}});
|
|
coll.insert({zone: 5, loc: {type: 'Point', coordinates: [20, 20]}});
|
|
}
|
|
var originGeoJSON = {type: 'Point', coordinates: [0, 0]};
|
|
|
|
// Test distinct with $nearSphere query predicate.
|
|
|
|
// A. Unindexed key, no geo index on query predicate.
|
|
res = coll.runCommand(
|
|
'distinct',
|
|
{key: 'zone', query: {loc: {$nearSphere: {$geometry: originGeoJSON, $maxDistance: 1}}}});
|
|
assert.commandFailed(res);
|
|
// B. Unindexed key, with 2dsphere index on query predicate.
|
|
assert.commandWorked(coll.ensureIndex({loc: '2dsphere'}));
|
|
res = coll.runCommand(
|
|
'distinct',
|
|
{key: 'zone', query: {loc: {$nearSphere: {$geometry: originGeoJSON, $maxDistance: 1}}}});
|
|
assert.commandWorked(res);
|
|
assert.eq(res.values.sort(), [3]);
|
|
// C. Indexed key, with 2dsphere index on query predicate.
|
|
assert.commandWorked(coll.ensureIndex({zone: 1}));
|
|
res = coll.runCommand(
|
|
'distinct',
|
|
{key: 'zone', query: {loc: {$nearSphere: {$geometry: originGeoJSON, $maxDistance: 1}}}});
|
|
assert.commandWorked(res);
|
|
assert.eq(res.values.sort(), [3]);
|
|
|
|
// Test distinct with $near query predicate.
|
|
|
|
coll.dropIndexes();
|
|
|
|
// A. Unindexed key, no geo index on query predicate.
|
|
res = coll.runCommand('distinct',
|
|
{key: 'zone', query: {'loc.coordinates': {$near: [0, 0], $maxDistance: 1}}});
|
|
assert.commandFailed(res);
|
|
// B. Unindexed key, with 2d index on query predicate.
|
|
assert.commandWorked(coll.ensureIndex({'loc.coordinates': '2d'}));
|
|
res = coll.runCommand('distinct',
|
|
{key: 'zone', query: {'loc.coordinates': {$near: [0, 0], $maxDistance: 1}}});
|
|
assert.commandWorked(res);
|
|
assert.eq(res.values.sort(), [3]);
|
|
// C. Indexed key, with 2d index on query predicate.
|
|
assert.commandWorked(coll.ensureIndex({zone: 1}));
|
|
res = coll.runCommand('distinct',
|
|
{key: 'zone', query: {'loc.coordinates': {$near: [0, 0], $maxDistance: 1}}});
|
|
assert.commandWorked(res);
|
|
assert.eq(res.values.sort(), [3]);
|