Adding support for new 'resultMeta' object

elastic · Oct 31, 2019 · cb0d08d · cb0d08d
1 parent 50edbd8
commit cb0d08d
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 5 deletions.
diff --git a/x-pack/legacy/plugins/siem/public/components/embeddables/__mocks__/mock.ts b/x-pack/legacy/plugins/siem/public/components/embeddables/__mocks__/mock.ts
@@ -216,6 +216,6 @@ export const mockLayerListDouble = [
 
 export const mockMapLayerEventHandlers: MapLayerEventHandlers = {
   onDataLoad: ({ layerId, dataId }: OnDataLoadProps) => {},
-  onDataLoadEnd: ({ layerId, dataId, featuresCount }: OnDataLoadEndProps) => {},
+  onDataLoadEnd: ({ layerId, dataId, resultMeta }: OnDataLoadEndProps) => {},
   onDataLoadError: ({ layerId, dataId, errorMessage }: OnDataLoadErrorProps) => {},
 };
diff --git a/x-pack/legacy/plugins/siem/public/components/embeddables/hooks/use_map_layers.tsx b/x-pack/legacy/plugins/siem/public/components/embeddables/hooks/use_map_layers.tsx
@@ -53,9 +53,15 @@ export const useMapLayers = (): Return => {
     onDataLoad: ({ layerId, dataId }: OnDataLoadProps) => {
       setMapLayers(prevMapLayers => mergeMapLayers(prevMapLayers, dataId, layerId, true));
     },
-    onDataLoadEnd: ({ layerId, dataId, featuresCount }: OnDataLoadEndProps) => {
+    onDataLoadEnd: ({ layerId, dataId, resultMeta }: OnDataLoadEndProps) => {
       setMapLayers(prevMapLayers =>
-        mergeMapLayers(prevMapLayers, dataId, layerId, false, featuresCount)
+        mergeMapLayers(
+          prevMapLayers,
+          dataId,
+          layerId,
+          false,
+          resultMeta.featuresCount != null ? resultMeta.featuresCount : 0
+        )
       );
     },
     onDataLoadError: ({ layerId, dataId, errorMessage }: OnDataLoadErrorProps) => {

diff --git a/x-pack/legacy/plugins/siem/public/components/embeddables/types.ts b/x-pack/legacy/plugins/siem/public/components/embeddables/types.ts
@@ -75,7 +75,9 @@ export interface OnDataLoadProps {
 export interface OnDataLoadEndProps {
   layerId: string;
   dataId: string;
-  featuresCount: number;
+  resultMeta: {
+    featuresCount: number;
+  };
 }
 
 export interface OnDataLoadErrorProps {
@@ -86,7 +88,7 @@ export interface OnDataLoadErrorProps {
 
 export interface MapLayerEventHandlers {
   onDataLoad({ layerId, dataId }: OnDataLoadProps): void;
-  onDataLoadEnd({ layerId, dataId, featuresCount }: OnDataLoadEndProps): void;
+  onDataLoadEnd({ layerId, dataId, resultMeta }: OnDataLoadEndProps): void;
   onDataLoadError({ layerId, dataId, errorMessage }: OnDataLoadErrorProps): void;
 }