mirror of https://github.com/pelias/api.git
Harish Krishna
10 years ago
12 changed files with 128 additions and 429 deletions
@ -1,77 +0,0 @@ |
|||||||
|
|
||||||
var service = { |
|
||||||
suggest: require('../service/suggest_multiple'), |
|
||||||
mget: require('../service/mget') |
|
||||||
}; |
|
||||||
var geojsonify = require('../helper/geojsonify').search; |
|
||||||
var resultsHelper = require('../helper/results'); |
|
||||||
|
|
||||||
function setup( backend, query, query_mixer ){ |
|
||||||
|
|
||||||
// allow overriding of dependencies
|
|
||||||
backend = backend || require('../src/backend'); |
|
||||||
query = query || require('../query/suggest_multiple'); |
|
||||||
query_mixer = query_mixer || require('../helper/queryMixer').suggest; |
|
||||||
|
|
||||||
function controller( req, res, next ){ |
|
||||||
|
|
||||||
// backend command
|
|
||||||
var cmd = { |
|
||||||
index: 'pelias', |
|
||||||
body: query( req.clean, query_mixer ) |
|
||||||
}; |
|
||||||
|
|
||||||
// responder
|
|
||||||
function reply( docs ){ |
|
||||||
|
|
||||||
// convert docs to geojson
|
|
||||||
var geojson = geojsonify( docs ); |
|
||||||
|
|
||||||
// response envelope
|
|
||||||
geojson.date = new Date().getTime(); |
|
||||||
|
|
||||||
// respond
|
|
||||||
return res.status(200).json( geojson ); |
|
||||||
} |
|
||||||
|
|
||||||
// query backend
|
|
||||||
service.suggest( backend, cmd, function( err, suggested ){ |
|
||||||
|
|
||||||
// error handler
|
|
||||||
if( err ){ return next( err ); } |
|
||||||
|
|
||||||
// pick the required number of results
|
|
||||||
suggested = resultsHelper.picker(suggested, req.clean.size); |
|
||||||
|
|
||||||
// no documents suggested, return empty array to avoid ActionRequestValidationException
|
|
||||||
if( !Array.isArray( suggested ) || !suggested.length ){ |
|
||||||
return reply([]); |
|
||||||
} |
|
||||||
|
|
||||||
// map suggester output to mget query
|
|
||||||
var query = suggested.map( function( doc ) { |
|
||||||
var idParts = doc.text.split(':'); |
|
||||||
return { |
|
||||||
_index: 'pelias', |
|
||||||
_type: idParts[0], |
|
||||||
_id: idParts.slice(1).join(':') |
|
||||||
}; |
|
||||||
}); |
|
||||||
|
|
||||||
service.mget( backend, query, function( err, docs ){ |
|
||||||
|
|
||||||
// error handler
|
|
||||||
if( err ){ return next( err ); } |
|
||||||
|
|
||||||
// reply
|
|
||||||
return reply( docs ); |
|
||||||
|
|
||||||
}); |
|
||||||
}); |
|
||||||
|
|
||||||
} |
|
||||||
|
|
||||||
return controller; |
|
||||||
} |
|
||||||
|
|
||||||
module.exports = setup; |
|
@ -1,65 +0,0 @@ |
|||||||
|
|
||||||
var logger = require('../src/logger'); |
|
||||||
|
|
||||||
// Build pelias suggest query
|
|
||||||
function generate( params, query_mixer ){ |
|
||||||
|
|
||||||
var CmdGenerator = function(params){ |
|
||||||
this.params = params; |
|
||||||
this.cmd = { |
|
||||||
'text': params.input |
|
||||||
}; |
|
||||||
};
|
|
||||||
|
|
||||||
CmdGenerator.prototype.get_precision = function() { |
|
||||||
var zoom = this.params.zoom; |
|
||||||
switch (true) { |
|
||||||
case (zoom > 15): |
|
||||||
return 5; // zoom: >= 16
|
|
||||||
case (zoom > 9): |
|
||||||
return 4; // zoom: 10-15
|
|
||||||
case (zoom > 5): |
|
||||||
return 3; // zoom: 6-9
|
|
||||||
case (zoom > 3): |
|
||||||
return 2; // zoom: 4-5
|
|
||||||
default: |
|
||||||
return 1; // zoom: 1-3 or when zoom: undefined
|
|
||||||
}
|
|
||||||
}; |
|
||||||
|
|
||||||
CmdGenerator.prototype.add_suggester = function(name, precision, layers, fuzzy) { |
|
||||||
this.cmd[name] = { |
|
||||||
'completion' : { |
|
||||||
'size' : this.params.size, |
|
||||||
'field' : 'suggest', |
|
||||||
'context': { |
|
||||||
'dataset': layers || this.params.layers, |
|
||||||
'location': { |
|
||||||
'value': [ this.params.lon, this.params.lat ], |
|
||||||
'precision': precision || this.get_precision() |
|
||||||
} |
|
||||||
}, |
|
||||||
'fuzzy': { |
|
||||||
'fuzziness': fuzzy || 0 |
|
||||||
} |
|
||||||
} |
|
||||||
}; |
|
||||||
}; |
|
||||||
|
|
||||||
var cmd = new CmdGenerator(params); |
|
||||||
query_mixer.forEach(function(item, index){ |
|
||||||
if (item.precision && Array.isArray( item.precision ) && item.precision.length ) { |
|
||||||
item.precision.forEach(function(precision) { |
|
||||||
cmd.add_suggester('pelias_'+index, precision, item.layers, item.fuzzy); |
|
||||||
}); |
|
||||||
} else { |
|
||||||
cmd.add_suggester('pelias_'+index, undefined, item.layers, item.fuzzy); |
|
||||||
} |
|
||||||
}); |
|
||||||
|
|
||||||
// logger.log( 'cmd', JSON.stringify( cmd.cmd, null, 2 ) );
|
|
||||||
return cmd.cmd; |
|
||||||
|
|
||||||
} |
|
||||||
|
|
||||||
module.exports = generate; |
|
@ -1,44 +0,0 @@ |
|||||||
|
|
||||||
/** |
|
||||||
|
|
||||||
cmd can be any valid ES suggest command |
|
||||||
|
|
||||||
**/ |
|
||||||
|
|
||||||
function service( backend, cmd, cb ){ |
|
||||||
// query new backend
|
|
||||||
backend().client.suggest( cmd, function( err, data ){ |
|
||||||
// handle backend errors
|
|
||||||
if( err ){ return cb( err ); } |
|
||||||
|
|
||||||
// map returned documents
|
|
||||||
|
|
||||||
var docs = []; |
|
||||||
var unique_ids = []; |
|
||||||
var num_keys = Object.keys(data).length; |
|
||||||
var has_docs = function(obj) { |
|
||||||
return Array.isArray( obj ) && obj.length && obj[0].options && obj[0].options.length; |
|
||||||
}; |
|
||||||
for (var i=0, j=0; i<num_keys && j<num_keys; i++) { |
|
||||||
var keys = 'pelias_'+i; |
|
||||||
if ( has_docs(data[keys]) ){ |
|
||||||
docs[i] = docs[i] || []; |
|
||||||
var res = data[keys][0].options[0]; |
|
||||||
if (unique_ids.indexOf(res.text) === -1) { |
|
||||||
docs[i].push(res); |
|
||||||
unique_ids.push(res.text); |
|
||||||
}
|
|
||||||
data[keys][0].options.splice(0,1); |
|
||||||
} else { |
|
||||||
j++; |
|
||||||
} |
|
||||||
i = i === num_keys-1 ? 1 : i; |
|
||||||
} |
|
||||||
|
|
||||||
// fire callback
|
|
||||||
return cb( null, docs); |
|
||||||
}); |
|
||||||
|
|
||||||
} |
|
||||||
|
|
||||||
module.exports = service; |
|
@ -1,112 +0,0 @@ |
|||||||
|
|
||||||
var setup = require('../../../controller/suggest'), |
|
||||||
mockBackend = require('../mock/backend'), |
|
||||||
mockQuery = require('../mock/query'); |
|
||||||
|
|
||||||
module.exports.tests = {}; |
|
||||||
|
|
||||||
module.exports.tests.interface = function(test, common) { |
|
||||||
test('valid interface', function(t) { |
|
||||||
t.equal(typeof setup, 'function', 'setup is a function'); |
|
||||||
t.equal(typeof setup(), 'function', 'setup returns a controller'); |
|
||||||
t.end(); |
|
||||||
}); |
|
||||||
}; |
|
||||||
|
|
||||||
// functionally test controller (backend success)
|
|
||||||
module.exports.tests.functional_success = function(test, common) { |
|
||||||
|
|
||||||
// expected geojson features for 'client/mget/ok/1' fixture
|
|
||||||
var expected = [{ |
|
||||||
type: 'Feature', |
|
||||||
geometry: { |
|
||||||
type: 'Point', |
|
||||||
coordinates: [ -50.5, 100.1 ] |
|
||||||
}, |
|
||||||
properties: { |
|
||||||
id: 'myid1', |
|
||||||
type: 'mytype1', |
|
||||||
layer: 'mytype1', |
|
||||||
name: 'test name1', |
|
||||||
admin0: 'country1', |
|
||||||
admin1: 'state1', |
|
||||||
admin2: 'city1', |
|
||||||
text: 'test name1, city1, state1' |
|
||||||
} |
|
||||||
}, { |
|
||||||
type: 'Feature', |
|
||||||
geometry: { |
|
||||||
type: 'Point', |
|
||||||
coordinates: [ -51.5, 100.2 ] |
|
||||||
}, |
|
||||||
properties: { |
|
||||||
id: 'myid2', |
|
||||||
type: 'mytype2', |
|
||||||
layer: 'mytype2', |
|
||||||
name: 'test name2', |
|
||||||
admin0: 'country2', |
|
||||||
admin1: 'state2', |
|
||||||
admin2: 'city2', |
|
||||||
text: 'test name2, city2, state2' |
|
||||||
} |
|
||||||
}]; |
|
||||||
|
|
||||||
test('functional success', function(t) { |
|
||||||
var i = 0; |
|
||||||
var backend = mockBackend( 'client/suggest/ok/1', function( cmd ){ |
|
||||||
// the backend executes 2 commands, so we check them both
|
|
||||||
if( ++i === 1 ){ |
|
||||||
t.deepEqual(cmd, { body: { a: 'b' }, index: 'pelias' }, 'correct suggest command'); |
|
||||||
} else { |
|
||||||
t.deepEqual(cmd, {
|
|
||||||
body: { docs: [
|
|
||||||
{ _id: 'mockid1', _index: 'pelias', _type: 'mocktype' },
|
|
||||||
{ _id: 'mockid2', _index: 'pelias', _type: 'mocktype' } ]
|
|
||||||
}
|
|
||||||
}, 'correct mget command'); |
|
||||||
} |
|
||||||
}); |
|
||||||
var controller = setup( backend, mockQuery() ); |
|
||||||
var res = { |
|
||||||
status: function( code ){ |
|
||||||
t.equal(code, 200, 'status set'); |
|
||||||
return res; |
|
||||||
}, |
|
||||||
json: function( json ){ |
|
||||||
t.equal(typeof json, 'object', 'returns json'); |
|
||||||
t.equal(typeof json.date, 'number', 'date set'); |
|
||||||
t.equal(json.type, 'FeatureCollection', 'valid geojson'); |
|
||||||
t.true(Array.isArray(json.features), 'features is array'); |
|
||||||
t.deepEqual(json.features, expected, 'values correctly mapped'); |
|
||||||
t.end(); |
|
||||||
} |
|
||||||
}; |
|
||||||
controller( { clean: { a: 'b' } }, res ); |
|
||||||
}); |
|
||||||
}; |
|
||||||
|
|
||||||
// functionally test controller (backend failure)
|
|
||||||
module.exports.tests.functional_failure = function(test, common) { |
|
||||||
test('functional failure', function(t) { |
|
||||||
var backend = mockBackend( 'client/suggest/fail/1', function( cmd ){ |
|
||||||
t.deepEqual(cmd, { body: { a: 'b' }, index: 'pelias' }, 'correct backend command'); |
|
||||||
}); |
|
||||||
var controller = setup( backend, mockQuery() ); |
|
||||||
var next = function( message ){ |
|
||||||
t.equal(message,'a backend error occurred','error passed to errorHandler'); |
|
||||||
t.end(); |
|
||||||
}; |
|
||||||
controller( { clean: { a: 'b' } }, undefined, next ); |
|
||||||
}); |
|
||||||
}; |
|
||||||
|
|
||||||
module.exports.all = function (tape, common) { |
|
||||||
|
|
||||||
function test(name, testFunction) { |
|
||||||
return tape('GET /suggest/nearby ' + name, testFunction); |
|
||||||
} |
|
||||||
|
|
||||||
for( var testCase in module.exports.tests ){ |
|
||||||
module.exports.tests[testCase](test, common); |
|
||||||
} |
|
||||||
}; |
|
Loading…
Reference in new issue