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