From 64196a04b52e5f74d5cf5f4656dafcbdd6469ebe Mon Sep 17 00:00:00 2001
From: nreese <reese.nathan@elastic.co>
Date: Fri, 20 May 2022 12:16:33 -0600
Subject: [PATCH] [Maps] create MVT_VECTOR when using choropleth wizard

---
 .../create_choropleth_layer_descriptor.ts     | 61 +++++++++++--------
 1 file changed, 36 insertions(+), 25 deletions(-)

diff --git a/x-pack/plugins/maps/public/classes/layers/wizards/choropleth_layer_wizard/create_choropleth_layer_descriptor.ts b/x-pack/plugins/maps/public/classes/layers/wizards/choropleth_layer_wizard/create_choropleth_layer_descriptor.ts
index 92045f5911176..36e07d7383d18 100644
--- a/x-pack/plugins/maps/public/classes/layers/wizards/choropleth_layer_wizard/create_choropleth_layer_descriptor.ts
+++ b/x-pack/plugins/maps/public/classes/layers/wizards/choropleth_layer_wizard/create_choropleth_layer_descriptor.ts
@@ -10,6 +10,7 @@ import {
   AGG_TYPE,
   COLOR_MAP_TYPE,
   FIELD_ORIGIN,
+  LAYER_TYPE,
   SCALING_TYPES,
   SOURCE_TYPES,
   STYLE_TYPE,
@@ -21,10 +22,11 @@ import {
   CountAggDescriptor,
   EMSFileSourceDescriptor,
   ESSearchSourceDescriptor,
+  JoinDescriptor,
   VectorStylePropertiesDescriptor,
 } from '../../../../../common/descriptor_types';
 import { VectorStyle } from '../../../styles/vector/vector_style';
-import { GeoJsonVectorLayer } from '../../vector_layer';
+import { GeoJsonVectorLayer, MvtVectorLayer } from '../../vector_layer';
 import { EMSFileSource } from '../../../sources/ems_file_source';
 // @ts-ignore
 import { ESSearchSource } from '../../../sources/es_search_source';
@@ -38,14 +40,14 @@ function createChoroplethLayerDescriptor({
   rightIndexPatternId,
   rightIndexPatternTitle,
   rightTermField,
-  setLabelStyle,
+  layerType,
 }: {
   sourceDescriptor: EMSFileSourceDescriptor | ESSearchSourceDescriptor;
   leftField: string;
   rightIndexPatternId: string;
   rightIndexPatternTitle: string;
   rightTermField: string;
-  setLabelStyle: boolean;
+  layerType: LAYER_TYPE.GEOJSON_VECTOR | LAYER_TYPE.MVT_VECTOR;
 }) {
   const metricsDescriptor: CountAggDescriptor = { type: AGG_TYPE.COUNT };
   const joinId = uuid();
@@ -75,7 +77,8 @@ function createChoroplethLayerDescriptor({
       },
     },
   };
-  if (setLabelStyle) {
+  // Styling label by join metric with MVT is not supported
+  if (layerType === LAYER_TYPE.GEOJSON_VECTOR) {
     styleProperties[VECTOR_STYLES.LABEL_TEXT] = {
       type: STYLE_TYPE.DYNAMIC,
       options: {
@@ -88,26 +91,34 @@ function createChoroplethLayerDescriptor({
     };
   }
 
-  return GeoJsonVectorLayer.createDescriptor({
-    joins: [
-      {
-        leftField,
-        right: {
-          type: SOURCE_TYPES.ES_TERM_SOURCE,
-          id: joinId,
-          indexPatternId: rightIndexPatternId,
-          indexPatternTitle: rightIndexPatternTitle,
-          term: rightTermField,
-          metrics: [metricsDescriptor],
-          applyGlobalQuery: true,
-          applyGlobalTime: true,
-          applyForceRefresh: true,
-        },
+  const joins = [
+    {
+      leftField,
+      right: {
+        type: SOURCE_TYPES.ES_TERM_SOURCE,
+        id: joinId,
+        indexPatternId: rightIndexPatternId,
+        indexPatternTitle: rightIndexPatternTitle,
+        term: rightTermField,
+        metrics: [metricsDescriptor],
+        applyGlobalQuery: true,
+        applyGlobalTime: true,
+        applyForceRefresh: true,
       },
-    ],
-    sourceDescriptor,
-    style: VectorStyle.createDescriptor(styleProperties),
-  });
+    } as JoinDescriptor,
+  ];
+
+  return layerType === LAYER_TYPE.MVT_VECTOR
+    ? MvtVectorLayer.createDescriptor({
+        joins,
+        sourceDescriptor,
+        style: VectorStyle.createDescriptor(styleProperties),
+      })
+    : GeoJsonVectorLayer.createDescriptor({
+        joins,
+        sourceDescriptor,
+        style: VectorStyle.createDescriptor(styleProperties),
+      });
 }
 
 export function createEmsChoroplethLayerDescriptor({
@@ -132,7 +143,7 @@ export function createEmsChoroplethLayerDescriptor({
     rightIndexPatternId,
     rightIndexPatternTitle,
     rightTermField,
-    setLabelStyle: true,
+    layerType: LAYER_TYPE.GEOJSON_VECTOR,
   });
 }
 
@@ -165,6 +176,6 @@ export function createEsChoroplethLayerDescriptor({
     rightIndexPatternId,
     rightIndexPatternTitle,
     rightTermField,
-    setLabelStyle: false, // Styling label by join metric with MVT is not supported
+    layerType: LAYER_TYPE.MVT_VECTOR,
   });
 }