[ML] Enabling lat_long agg in advanced wizard

Author: jgowdyelastic

x-pack/legacy/plugins/ml/server/models/job_service/new_job_caps/field_service.ts

@@ -31,6 +31,8 @@ const supportedTypes: string[] = [
   ES_FIELD_TYPES.SCALED_FLOAT,
   ES_FIELD_TYPES.SHORT,
   ES_FIELD_TYPES.IP,
+  ES_FIELD_TYPES.GEO_POINT,
+  ES_FIELD_TYPES.GEO_SHAPE,
 ];
 
 export function fieldServiceProvider(
@@ -135,24 +137,24 @@ async function combineFieldsAndAggs(
   const keywordFields = getKeywordFields(fields);
   const numericalFields = getNumericalFields(fields);
   const ipFields = getIpFields(fields);
+  const geoFields = getGeoFields(fields);
 
   const isRollup = Object.keys(rollupFields).length > 0;
   const mix = mixFactory(isRollup, rollupFields);
 
   aggs.forEach(a => {
     if (a.type === METRIC_AGG_TYPE && a.fields !== undefined) {
       switch (a.id) {
+        case ML_JOB_AGGREGATION.LAT_LONG:
+          geoFields.forEach(f => mix(f, a));
+          break;
         case ML_JOB_AGGREGATION.DISTINCT_COUNT:
         case ML_JOB_AGGREGATION.HIGH_DISTINCT_COUNT:
         case ML_JOB_AGGREGATION.LOW_DISTINCT_COUNT:
           // distinct count (i.e. cardinality) takes keywords, ips
           // as well as numerical fields
-          keywordFields.forEach(f => {
-            mix(f, a);
-          });
-          ipFields.forEach(f => {
-            mix(f, a);
-          });
+          keywordFields.forEach(f => mix(f, a));
+          ipFields.forEach(f => mix(f, a));
         // note, no break to fall through to add numerical fields.
         default:
           // all other aggs take numerical fields
@@ -235,3 +237,9 @@ function getNumericalFields(fields: Field[]): Field[] {
       f.type === ES_FIELD_TYPES.SCALED_FLOAT
   );
 }
+
+function getGeoFields(fields: Field[]): Field[] {
+  return fields.filter(
+    f => f.type === ES_FIELD_TYPES.GEO_POINT || f.type === ES_FIELD_TYPES.GEO_SHAPE
+  );
+}