mirror of https://github.com/pelias/api.git
Harish Krishna
10 years ago
8 changed files with 183 additions and 352 deletions
@ -1,72 +0,0 @@
|
||||
|
||||
var service = { |
||||
suggest: require('../service/suggest'), |
||||
mget: require('../service/mget') |
||||
}; |
||||
var geojsonify = require('../helper/geojsonify').search; |
||||
|
||||
function setup( backend, query ){ |
||||
|
||||
// allow overriding of dependencies
|
||||
backend = backend || require('../src/backend'); |
||||
query = query || require('../query/suggest'); |
||||
|
||||
function controller( req, res, next ){ |
||||
|
||||
// backend command
|
||||
var cmd = { |
||||
index: 'pelias', |
||||
body: query( req.clean ) |
||||
}; |
||||
|
||||
// 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 ); } |
||||
|
||||
// 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; |
||||
@ -0,0 +1,28 @@
|
||||
{ |
||||
"suggest": [ |
||||
{ |
||||
"layers": "all", |
||||
"precision": [5, 3, 1] |
||||
}, |
||||
{ |
||||
"layers": "admin", |
||||
"precision": [] |
||||
}, |
||||
{ |
||||
"layers": "all", |
||||
"precision": [3], |
||||
"fuzzy": "AUTO" |
||||
} |
||||
], |
||||
"suggest_nearby": [ |
||||
{ |
||||
"layers": "all", |
||||
"precision": [] |
||||
}, |
||||
{ |
||||
"layers": "all", |
||||
"precision": [], |
||||
"fuzzy": "AUTO" |
||||
} |
||||
] |
||||
} |
||||
@ -0,0 +1,88 @@
|
||||
var service = { |
||||
suggest: require('../service/suggest'), |
||||
mget: require('../service/mget') |
||||
}; |
||||
var geojsonify = require('../helper/geojsonify').search; |
||||
|
||||
function setup(backend, res, next) { |
||||
var query_backend = function(cmd, callback) { |
||||
// query backend
|
||||
|
||||
service.suggest( backend, cmd, function( err, docs ){ |
||||
|
||||
// error handler
|
||||
if( err ){ return next( err ); } |
||||
|
||||
callback(null, docs); |
||||
}); |
||||
}; |
||||
|
||||
var dedup = function(combined) { |
||||
var unique_ids = []; |
||||
return combined.filter(function(item, pos) { |
||||
if (unique_ids.indexOf(item.text) === -1) { |
||||
unique_ids.push(item.text); |
||||
return true; |
||||
} |
||||
return false; |
||||
}); |
||||
}; |
||||
|
||||
var sort_by_score = function(combined) { |
||||
return combined.sort(function(a,b) { |
||||
return b.score - a.score; |
||||
}); |
||||
}; |
||||
|
||||
var reply = function(res, docs) { |
||||
|
||||
// convert docs to geojson
|
||||
var geojson = geojsonify( docs ); |
||||
|
||||
// response envelope
|
||||
geojson.date = new Date().getTime(); |
||||
|
||||
// respond
|
||||
return res.status(200).json( geojson ); |
||||
}; |
||||
|
||||
var respond = function(data) { |
||||
|
||||
// no documents suggested, return empty array to avoid ActionRequestValidationException
|
||||
if( !Array.isArray( data ) || !data.length ){ |
||||
return reply(res, []); |
||||
} |
||||
|
||||
// map suggester output to mget query
|
||||
var query = data.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( res, docs ); |
||||
|
||||
}); |
||||
|
||||
}; |
||||
|
||||
return { |
||||
query_backend: query_backend, |
||||
dedup: dedup, |
||||
sort_by_score: sort_by_score, |
||||
reply: reply, |
||||
respond: respond |
||||
}; |
||||
|
||||
} |
||||
|
||||
module.exports = setup; |
||||
@ -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