feat(admin_ngram): compute an ngram field for all admin data

pelias · Feb 19, 2019 · 196760b · 196760b
1 parent b3b5478
commit 196760b
Show file tree

Hide file tree

Showing 6 changed files with 5,004 additions and 471 deletions.
diff --git a/mappings/partial/admin.json b/mappings/partial/admin.json
@@ -1,4 +1,14 @@
 {
   "type": "string",
-  "analyzer": "peliasAdmin"
-}
+  "analyzer": "peliasAdmin",
+  "fields": {
+    "ngram": {
+      "type": "string",
+      "analyzer": "peliasIndexOneEdgeGram",
+      "doc_values": false,
+      "fielddata": {
+        "format": "disabled"
+      }
+    }
+  }
+}
diff --git a/mappings/partial/postalcode.json b/mappings/partial/postalcode.json
@@ -1,4 +1,14 @@
 {
   "type": "string",
-  "analyzer": "peliasZip"
+  "analyzer": "peliasZip",
+  "fields": {
+    "ngram": {
+      "type": "string",
+      "analyzer": "peliasIndexOneEdgeGram",
+      "doc_values": false,
+      "fielddata": {
+        "format": "disabled"
+      }
+    }
+  }
 }
diff --git a/package.json b/package.json
@@ -35,6 +35,7 @@
     "colors": "^1.1.2",
     "elasticsearch": "^15.0.0",
     "joi": "^14.0.0",
+    "lodash.has": "^4.5.2",
     "lodash.merge": "^4.6.0",
     "pelias-config": "^3.5.0",
     "pelias-logger": "^1.3.0"

diff --git a/test/compile.js b/test/compile.js
@@ -65,6 +65,7 @@ module.exports.tests.current_schema = function(test, common) {
 
     // code intentionally commented to allow quick debugging of expected.json
     // common.diff(schemaCopy, fixture);
+    // console.error( JSON.stringify( schemaCopy, null, 2 ) );
 
     t.deepEqual(schemaCopy, fixture);
     t.end();

diff --git a/test/document.js b/test/document.js
@@ -1,4 +1,5 @@
-var schema = require('../mappings/document');
+const schema = require('../mappings/document');
+const has = require('lodash.has');
 
 module.exports.tests = {};
 
@@ -85,24 +86,26 @@ module.exports.tests.address_analysis = function(test, common) {
 // should contain the correct parent field definitions
 module.exports.tests.parent_fields = function(test, common) {
   var fields = [
-    'continent',      'continent_a',      'continent_id',
-    'ocean',          'ocean_a',          'ocean_id',
-    'empire',         'empire_a',         'empire_id',
-    'country',        'country_a',        'country_id',
-    'dependency',     'dependency_a',     'dependency_id',
-    'marinearea',     'marinearea_a',     'marinearea_id',
-    'macroregion',    'macroregion_a',    'macroregion_id',
-    'region',         'region_a',         'region_id',
-    'macrocounty',    'macrocounty_a',    'macrocounty_id',
-    'county',         'county_a',         'county_id',
-    'locality',       'locality_a',       'locality_id',
-    'borough',        'borough_a',        'borough_id',
-    'localadmin',     'localadmin_a',     'localadmin_id',
-    'neighbourhood',  'neighbourhood_a',  'neighbourhood_id',
-    'postalcode',     'postalcode_a',     'postalcode_id'
+    'continent',      'continent_a',      'continent_id',     'continent.fields.ngram',
+    'ocean',          'ocean_a',          'ocean_id',         'ocean.fields.ngram',
+    'empire',         'empire_a',         'empire_id',        'empire.fields.ngram',
+    'country',        'country_a',        'country_id',       'country.fields.ngram',
+    'dependency',     'dependency_a',     'dependency_id',    'dependency.fields.ngram',
+    'marinearea',     'marinearea_a',     'marinearea_id',    'marinearea.fields.ngram',
+    'macroregion',    'macroregion_a',    'macroregion_id',   'macroregion.fields.ngram',
+    'region',         'region_a',         'region_id',        'region.fields.ngram',
+    'macrocounty',    'macrocounty_a',    'macrocounty_id',   'macrocounty.fields.ngram',
+    'county',         'county_a',         'county_id',        'county.fields.ngram',
+    'locality',       'locality_a',       'locality_id',      'locality.fields.ngram',
+    'borough',        'borough_a',        'borough_id',       'borough.fields.ngram',
+    'localadmin',     'localadmin_a',     'localadmin_id',    'localadmin.fields.ngram',
+    'neighbourhood',  'neighbourhood_a',  'neighbourhood_id', 'neighbourhood.fields.ngram',
+    'postalcode',     'postalcode_a',     'postalcode_id',    'postalcode.fields.ngram'
   ];
   test('parent fields specified', function(t) {
-    t.deepEqual(Object.keys(schema.properties.parent.properties), fields);
+    fields.forEach( expected => {
+      t.true( has( schema.properties.parent.properties, expected ), expected );
+    });
     t.end();
   });
 };
@@ -111,7 +114,11 @@ module.exports.tests.parent_fields = function(test, common) {
 // ref: https://github.com/pelias/schema/pull/95
 module.exports.tests.parent_analysis = function(test, common) {
   var prop = schema.properties.parent.properties;
-  var fields = ['country','region','county','locality','localadmin','neighbourhood'];
+  var fields = [
+    'continent', 'ocean', 'empire', 'country', 'dependency', 'marinearea',
+    'macroregion', 'region', 'macrocounty', 'county', 'locality', 'borough',
+    'localadmin', 'neighbourhood'
+  ];
   fields.forEach( function( field ){
     test(field, function(t) {
       t.equal(prop[field].type, 'string');
@@ -121,6 +128,10 @@ module.exports.tests.parent_analysis = function(test, common) {
       t.equal(prop[field+'_id'].type, 'string');
       t.equal(prop[field+'_id'].index, 'not_analyzed');
 
+      // subfields
+      t.equal(prop[field].fields.ngram.type, 'string');
+      t.equal(prop[field].fields.ngram.analyzer, 'peliasIndexOneEdgeGram');
+
       t.end();
     });
   });