mirror of https://github.com/pelias/api.git
Julian Simioni
9 years ago
13 changed files with 676 additions and 102 deletions
@ -0,0 +1,35 @@ |
|||||||
|
|
||||||
|
var peliasQuery = require('pelias-query'); |
||||||
|
|
||||||
|
/** |
||||||
|
This view is the same as `peliasQuery.view.focus` with one exception: |
||||||
|
|
||||||
|
if the view is generated successfully, we add a 'filter' clause which |
||||||
|
restricts the targeted '_type' to be in the list specified below. |
||||||
|
|
||||||
|
documents which are not in the '_type' list below will simply score 0 for |
||||||
|
this section of the query. |
||||||
|
**/ |
||||||
|
|
||||||
|
module.exports = function( subview ){ |
||||||
|
return function( vs ){ |
||||||
|
|
||||||
|
if( !subview ){ return null; } // subview validation failed
|
||||||
|
var macroView = peliasQuery.view.focus( subview ); |
||||||
|
if( !macroView ){ return null; } // macroView validation failed
|
||||||
|
var view = macroView( vs ); |
||||||
|
|
||||||
|
if( view && view.hasOwnProperty('function_score') ){ |
||||||
|
view.function_score.filter = { |
||||||
|
'or': [ |
||||||
|
{ 'type': { 'value': 'osmnode' } }, |
||||||
|
{ 'type': { 'value': 'osmway' } }, |
||||||
|
{ 'type': { 'value': 'osmaddress' } }, |
||||||
|
{ 'type': { 'value': 'openaddresses' } } |
||||||
|
] |
||||||
|
}; |
||||||
|
} |
||||||
|
|
||||||
|
return view; |
||||||
|
}; |
||||||
|
}; |
@ -0,0 +1,37 @@ |
|||||||
|
|
||||||
|
var peliasQuery = require('pelias-query'), |
||||||
|
ngrams_strict = require('./ngrams_strict'); |
||||||
|
|
||||||
|
/** |
||||||
|
Ngrams view which trims the 'input:name' and only uses the LAST TOKEN. |
||||||
|
|
||||||
|
eg. if the input was "100 foo str", then 'input:name' would only be 'str' |
||||||
|
note: it is assumed that the rest of the input is matched using another view. |
||||||
|
|
||||||
|
there is an additional flag 'input:name:isComplete' used to disable this view |
||||||
|
selectively, see that section for more info. |
||||||
|
|
||||||
|
code notes: this view makes a copy of the $vs object in order to change their |
||||||
|
values without mutating the original values, which may be expected in their |
||||||
|
unaltered form by other views. |
||||||
|
**/ |
||||||
|
|
||||||
|
module.exports = function( vs ){ |
||||||
|
|
||||||
|
// Totally disable this view when bool value 'input:name:isComplete' is true.
|
||||||
|
// This is the case when the user has typed a comma, so we can assume
|
||||||
|
// that the 'name' part of the query is now complete.
|
||||||
|
if( vs.var('input:name:isComplete').get() ){ return null; } |
||||||
|
|
||||||
|
// make a copy Vars so we don't mutate the original
|
||||||
|
var vsCopy = new peliasQuery.Vars( vs.export() ); |
||||||
|
|
||||||
|
// get the input 'name' variable
|
||||||
|
var name = vs.var('input:name').get(); |
||||||
|
|
||||||
|
// set the 'name' variable in the copy to only the last token
|
||||||
|
vsCopy.var('input:name').set( name.substr( name.lastIndexOf(' ')+1 ) ); |
||||||
|
|
||||||
|
// return the view rendered using the copy
|
||||||
|
return ngrams_strict( vsCopy ); |
||||||
|
}; |
@ -0,0 +1,19 @@ |
|||||||
|
|
||||||
|
var peliasQuery = require('pelias-query'); |
||||||
|
|
||||||
|
/** |
||||||
|
Ngrams view with the additional properties to enable: |
||||||
|
type:phrase -> tokens MUST appear in the same order in BOTH query and index |
||||||
|
operator:and -> ALL tokens are mandatory, missing any single token will cause |
||||||
|
a query failure. |
||||||
|
**/ |
||||||
|
|
||||||
|
module.exports = function( vs ){ |
||||||
|
|
||||||
|
var view = peliasQuery.view.ngrams( vs ); |
||||||
|
|
||||||
|
view.match['name.default'].type = 'phrase'; |
||||||
|
view.match['name.default'].operator = 'and'; |
||||||
|
|
||||||
|
return view; |
||||||
|
}; |
@ -0,0 +1,44 @@ |
|||||||
|
|
||||||
|
var peliasQuery = require('pelias-query'); |
||||||
|
|
||||||
|
/** |
||||||
|
Phrase view which trims the 'input:name' and uses ALL BUT the last token. |
||||||
|
|
||||||
|
eg. if the input was "100 foo str", then 'input:name' would only be '100 foo' |
||||||
|
note: it is assumed that the rest of the input is matched using another view. |
||||||
|
|
||||||
|
there is an additional flag 'input:name:isComplete' used to disable this view |
||||||
|
selectively, see that section for more info. |
||||||
|
|
||||||
|
code notes: this view makes a copy of the $vs object in order to change their |
||||||
|
values without mutating the original values, which may be expected in their |
||||||
|
unaltered form by other views. |
||||||
|
**/ |
||||||
|
|
||||||
|
module.exports = function( vs ){ |
||||||
|
|
||||||
|
// Don't mutate the name variable when 'input:name:isComplete' is true.
|
||||||
|
// This is the case when the user has typed a comma, so we can assume
|
||||||
|
// that the 'name' part of the query is now complete.
|
||||||
|
if( vs.var('input:name:isComplete').get() ){ |
||||||
|
// return the view rendered using the original vars
|
||||||
|
return peliasQuery.view.phrase( vs ); |
||||||
|
} |
||||||
|
|
||||||
|
// make a copy Vars so we don't mutate the original
|
||||||
|
var vsCopy = new peliasQuery.Vars( vs.export() ); |
||||||
|
|
||||||
|
// get the input 'name' variable and split in to tokens
|
||||||
|
var name = vs.var('input:name').get(), |
||||||
|
tokens = name.split(' '); |
||||||
|
|
||||||
|
// single token only, abort (we don't want the *last* token)
|
||||||
|
// return null here will completely disable the view.
|
||||||
|
if( tokens.length < 2 ){ return null; } |
||||||
|
|
||||||
|
// set the 'name' variable in the copy to all but the last token
|
||||||
|
vsCopy.var('input:name').set( name.substr( 0, name.lastIndexOf(' ') ) ); |
||||||
|
|
||||||
|
// return the view rendered using the copy
|
||||||
|
return peliasQuery.view.phrase( vsCopy ); |
||||||
|
}; |
@ -0,0 +1,84 @@ |
|||||||
|
|
||||||
|
module.exports = { |
||||||
|
'query': { |
||||||
|
'filtered': { |
||||||
|
'query': { |
||||||
|
'bool': { |
||||||
|
'must': [{ |
||||||
|
'match': { |
||||||
|
'name.default': { |
||||||
|
'analyzer': 'peliasPhrase', |
||||||
|
'boost': 100, |
||||||
|
'query': 'one', |
||||||
|
'type': 'phrase', |
||||||
|
'operator': 'and' |
||||||
|
} |
||||||
|
} |
||||||
|
}], |
||||||
|
'should':[{ |
||||||
|
'function_score': { |
||||||
|
'query': { |
||||||
|
'match': { |
||||||
|
'name.default': { |
||||||
|
'analyzer': 'peliasPhrase', |
||||||
|
'boost': 100, |
||||||
|
'query': 'one', |
||||||
|
'type': 'phrase', |
||||||
|
'operator': 'and' |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
'max_boost': 20, |
||||||
|
'score_mode': 'first', |
||||||
|
'boost_mode': 'replace', |
||||||
|
'filter': { |
||||||
|
'exists': { |
||||||
|
'field': 'popularity' |
||||||
|
} |
||||||
|
}, |
||||||
|
'functions': [{ |
||||||
|
'field_value_factor': { |
||||||
|
'modifier': 'log1p', |
||||||
|
'field': 'popularity' |
||||||
|
}, |
||||||
|
'weight': 1 |
||||||
|
}] |
||||||
|
} |
||||||
|
},{ |
||||||
|
'function_score': { |
||||||
|
'query': { |
||||||
|
'match': { |
||||||
|
'name.default': { |
||||||
|
'analyzer': 'peliasPhrase', |
||||||
|
'boost': 100, |
||||||
|
'query': 'one', |
||||||
|
'type': 'phrase', |
||||||
|
'operator': 'and' |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
'max_boost': 20, |
||||||
|
'score_mode': 'first', |
||||||
|
'boost_mode': 'replace', |
||||||
|
'filter': { |
||||||
|
'exists': { |
||||||
|
'field': 'population' |
||||||
|
} |
||||||
|
}, |
||||||
|
'functions': [{ |
||||||
|
'field_value_factor': { |
||||||
|
'modifier': 'log1p', |
||||||
|
'field': 'population' |
||||||
|
}, |
||||||
|
'weight': 2 |
||||||
|
}] |
||||||
|
} |
||||||
|
}] |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
'sort': [ '_score' ], |
||||||
|
'size': 20, |
||||||
|
'track_scores': true |
||||||
|
}; |
@ -0,0 +1,95 @@ |
|||||||
|
|
||||||
|
module.exports = { |
||||||
|
'query': { |
||||||
|
'filtered': { |
||||||
|
'query': { |
||||||
|
'bool': { |
||||||
|
'must': [{ |
||||||
|
'match': { |
||||||
|
'phrase.default': { |
||||||
|
'analyzer': 'peliasPhrase', |
||||||
|
'type': 'phrase', |
||||||
|
'boost': 1, |
||||||
|
'slop': 2, |
||||||
|
'query': 'one two' |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
{ |
||||||
|
'match': { |
||||||
|
'name.default': { |
||||||
|
'analyzer': 'peliasPhrase', |
||||||
|
'boost': 100, |
||||||
|
'query': 'three', |
||||||
|
'type': 'phrase', |
||||||
|
'operator': 'and' |
||||||
|
} |
||||||
|
} |
||||||
|
}], |
||||||
|
'should':[{ |
||||||
|
'function_score': { |
||||||
|
'query': { |
||||||
|
'match': { |
||||||
|
'name.default': { |
||||||
|
'analyzer': 'peliasPhrase', |
||||||
|
'boost': 100, |
||||||
|
'query': 'one two three', |
||||||
|
'type': 'phrase', |
||||||
|
'operator': 'and' |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
'max_boost': 20, |
||||||
|
'score_mode': 'first', |
||||||
|
'boost_mode': 'replace', |
||||||
|
'filter': { |
||||||
|
'exists': { |
||||||
|
'field': 'popularity' |
||||||
|
} |
||||||
|
}, |
||||||
|
'functions': [{ |
||||||
|
'field_value_factor': { |
||||||
|
'modifier': 'log1p', |
||||||
|
'field': 'popularity' |
||||||
|
}, |
||||||
|
'weight': 1 |
||||||
|
}] |
||||||
|
} |
||||||
|
},{ |
||||||
|
'function_score': { |
||||||
|
'query': { |
||||||
|
'match': { |
||||||
|
'name.default': { |
||||||
|
'analyzer': 'peliasPhrase', |
||||||
|
'boost': 100, |
||||||
|
'query': 'one two three', |
||||||
|
'type': 'phrase', |
||||||
|
'operator': 'and' |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
'max_boost': 20, |
||||||
|
'score_mode': 'first', |
||||||
|
'boost_mode': 'replace', |
||||||
|
'filter': { |
||||||
|
'exists': { |
||||||
|
'field': 'population' |
||||||
|
} |
||||||
|
}, |
||||||
|
'functions': [{ |
||||||
|
'field_value_factor': { |
||||||
|
'modifier': 'log1p', |
||||||
|
'field': 'population' |
||||||
|
}, |
||||||
|
'weight': 2 |
||||||
|
}] |
||||||
|
} |
||||||
|
}] |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
'sort': [ '_score' ], |
||||||
|
'size': 20, |
||||||
|
'track_scores': true |
||||||
|
}; |
@ -0,0 +1,158 @@ |
|||||||
|
|
||||||
|
module.exports = { |
||||||
|
'query': { |
||||||
|
'filtered': { |
||||||
|
'query': { |
||||||
|
'bool': { |
||||||
|
'must': [ |
||||||
|
{ |
||||||
|
'match': { |
||||||
|
'phrase.default': { |
||||||
|
'analyzer': 'peliasPhrase', |
||||||
|
'type': 'phrase', |
||||||
|
'boost': 1, |
||||||
|
'slop': 2, |
||||||
|
'query': 'one two' |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
], |
||||||
|
'should': [ |
||||||
|
{ |
||||||
|
'match': { |
||||||
|
'admin0': { |
||||||
|
'analyzer': 'peliasAdmin', |
||||||
|
'boost': 800, |
||||||
|
'query': 'three' |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
{ |
||||||
|
'match': { |
||||||
|
'admin1': { |
||||||
|
'analyzer': 'peliasAdmin', |
||||||
|
'boost': 600, |
||||||
|
'query': 'three' |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
{ |
||||||
|
'match': { |
||||||
|
'admin1_abbr': { |
||||||
|
'analyzer': 'peliasAdmin', |
||||||
|
'boost': 600, |
||||||
|
'query': 'three' |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
{ |
||||||
|
'match': { |
||||||
|
'admin2': { |
||||||
|
'analyzer': 'peliasAdmin', |
||||||
|
'boost': 400, |
||||||
|
'query': 'three' |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
{ |
||||||
|
'match': { |
||||||
|
'local_admin': { |
||||||
|
'analyzer': 'peliasAdmin', |
||||||
|
'boost': 200, |
||||||
|
'query': 'three' |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
{ |
||||||
|
'match': { |
||||||
|
'locality': { |
||||||
|
'analyzer': 'peliasAdmin', |
||||||
|
'boost': 200, |
||||||
|
'query': 'three' |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
{ |
||||||
|
'match': { |
||||||
|
'neighborhood': { |
||||||
|
'analyzer': 'peliasAdmin', |
||||||
|
'boost': 200, |
||||||
|
'query': 'three' |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
{ |
||||||
|
'function_score': { |
||||||
|
'query': { |
||||||
|
'match': { |
||||||
|
'name.default': { |
||||||
|
'analyzer': 'peliasPhrase', |
||||||
|
'boost': 100, |
||||||
|
'query': 'one two', |
||||||
|
'type': 'phrase', |
||||||
|
'operator': 'and' |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
'max_boost': 20, |
||||||
|
'functions': [ |
||||||
|
{ |
||||||
|
'field_value_factor': { |
||||||
|
'modifier': 'log1p', |
||||||
|
'field': 'popularity' |
||||||
|
}, |
||||||
|
'weight': 1 |
||||||
|
} |
||||||
|
], |
||||||
|
'score_mode': 'first', |
||||||
|
'boost_mode': 'replace', |
||||||
|
'filter': { |
||||||
|
'exists': { |
||||||
|
'field': 'popularity' |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
{ |
||||||
|
'function_score': { |
||||||
|
'query': { |
||||||
|
'match': { |
||||||
|
'name.default': { |
||||||
|
'analyzer': 'peliasPhrase', |
||||||
|
'boost': 100, |
||||||
|
'query': 'one two', |
||||||
|
'type': 'phrase', |
||||||
|
'operator': 'and' |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
'max_boost': 20, |
||||||
|
'functions': [ |
||||||
|
{ |
||||||
|
'field_value_factor': { |
||||||
|
'modifier': 'log1p', |
||||||
|
'field': 'population' |
||||||
|
}, |
||||||
|
'weight': 2 |
||||||
|
} |
||||||
|
], |
||||||
|
'score_mode': 'first', |
||||||
|
'boost_mode': 'replace', |
||||||
|
'filter': { |
||||||
|
'exists': { |
||||||
|
'field': 'population' |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
] |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
}, |
||||||
|
'size': 20, |
||||||
|
'track_scores': true, |
||||||
|
'sort': [ |
||||||
|
'_score' |
||||||
|
] |
||||||
|
}; |
Loading…
Reference in new issue