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