mirror of https://github.com/pelias/api.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
2.1 KiB
80 lines
2.1 KiB
|
|
var geojsonify = require('../helper/geojsonify').suggest; |
|
var async = require('async'); |
|
|
|
function setup( backend, query ){ |
|
|
|
function controller( req, res, next ){ |
|
|
|
// combine the 2 queries |
|
|
|
// allow overriding of dependencies |
|
backend = backend || require('../src/backend'); |
|
var query_admin = require('../query/suggest_admin'); |
|
var query_poi = require('../query/suggest_poi'); |
|
var cmd = { |
|
index: 'pelias' |
|
}; |
|
var query_backend = function(cmd, callback) { |
|
// query backend |
|
backend().client.suggest( cmd, function( err, data ){ |
|
|
|
var docs = []; |
|
|
|
// handle backend errors |
|
if( err ){ return next( err ); } |
|
|
|
// map response to a valid FeatureCollection |
|
if( data && Array.isArray( data.pelias ) && data.pelias.length ){ |
|
docs = data['pelias'][0].options || []; |
|
} |
|
|
|
callback(null, docs); |
|
}); |
|
}; |
|
|
|
async.parallel({ |
|
admin: function(callback){ |
|
cmd.body = query_admin( req.clean ); |
|
query_backend(cmd, callback); |
|
}, |
|
poi: function(callback){ |
|
cmd.body = query_poi( req.clean ); |
|
query_backend(cmd, callback); |
|
}, |
|
poi1: function(callback){ |
|
cmd.body = query_poi( req.clean, 1 ); |
|
query_backend(cmd, callback); |
|
} |
|
}, |
|
function(err, results) { |
|
var splice_length = req.clean.size / 3; |
|
// results is now equals to: {admin: docs, poi: docs, poi1: docs} |
|
var combined = results.poi.splice(0,splice_length).concat(results.admin.splice(0,splice_length)).concat(results.poi1.splice(0,splice_length)); |
|
|
|
//dedup |
|
var unique_ids = []; |
|
combined = combined.filter(function(item, pos) { |
|
if (unique_ids.indexOf(item.payload.id) == -1) { |
|
unique_ids.push(item.payload.id); |
|
return true; |
|
} |
|
return false; |
|
}); |
|
|
|
// convert docs to geojson |
|
var geojson = geojsonify( combined ); |
|
|
|
// response envelope |
|
geojson.date = new Date().getTime(); |
|
|
|
// respond |
|
return res.status(200).json( geojson ); |
|
}); |
|
|
|
} |
|
|
|
return controller; |
|
} |
|
|
|
module.exports = setup; |