From 6295ae71b5d3a2810767968780be7c7eaccc60d7 Mon Sep 17 00:00:00 2001
From: Joe Reuter <johannes.reuter@elastic.co>
Date: Wed, 24 Mar 2021 14:04:19 +0100
Subject: [PATCH] [Lens] Cache avilable operations meta data (#95034)

---
 .../dimension_panel/operation_support.ts                    | 4 ++--
 .../plugins/lens/public/indexpattern_datasource/loader.ts   | 6 ++++++
 .../indexpattern_datasource/operations/__mocks__/index.ts   | 1 +
 .../public/indexpattern_datasource/operations/operations.ts | 4 +++-
 4 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/x-pack/plugins/lens/public/indexpattern_datasource/dimension_panel/operation_support.ts b/x-pack/plugins/lens/public/indexpattern_datasource/dimension_panel/operation_support.ts
index d462d73740aaa..801b1b17a1831 100644
--- a/x-pack/plugins/lens/public/indexpattern_datasource/dimension_panel/operation_support.ts
+++ b/x-pack/plugins/lens/public/indexpattern_datasource/dimension_panel/operation_support.ts
@@ -8,7 +8,7 @@
 import _ from 'lodash';
 import { DatasourceDimensionDropProps } from '../../types';
 import { OperationType } from '../indexpattern';
-import { getAvailableOperationsByMetadata } from '../operations';
+import { memoizedGetAvailableOperationsByMetadata } from '../operations';
 import { IndexPatternPrivateState } from '../types';
 
 export interface OperationSupportMatrix {
@@ -30,7 +30,7 @@ export const getOperationSupportMatrix = (props: Props): OperationSupportMatrix
   const layerId = props.layerId;
   const currentIndexPattern = props.state.indexPatterns[props.state.layers[layerId].indexPatternId];
 
-  const filteredOperationsByMetadata = getAvailableOperationsByMetadata(
+  const filteredOperationsByMetadata = memoizedGetAvailableOperationsByMetadata(
     currentIndexPattern
   ).filter((operation) => props.filterOperations(operation.operationMetaData));
 
diff --git a/x-pack/plugins/lens/public/indexpattern_datasource/loader.ts b/x-pack/plugins/lens/public/indexpattern_datasource/loader.ts
index 7a50b1e60d13f..7052a69ee6fb7 100644
--- a/x-pack/plugins/lens/public/indexpattern_datasource/loader.ts
+++ b/x-pack/plugins/lens/public/indexpattern_datasource/loader.ts
@@ -29,6 +29,7 @@ import { VisualizeFieldContext } from '../../../../../src/plugins/ui_actions/pub
 import { documentField } from './document_field';
 import { readFromStorage, writeToStorage } from '../settings_storage';
 import { getFieldByNameFactory } from './pure_helpers';
+import { memoizedGetAvailableOperationsByMetadata } from './operations';
 
 type SetState = StateSetter<IndexPatternPrivateState>;
 type IndexPatternsService = Pick<IndexPatternsContract, 'get' | 'getIdsWithTitle'>;
@@ -49,6 +50,11 @@ export async function loadIndexPatterns({
     return cache;
   }
 
+  if (memoizedGetAvailableOperationsByMetadata.cache.clear) {
+    // clear operations meta data cache because index pattern reference may change
+    memoizedGetAvailableOperationsByMetadata.cache.clear();
+  }
+
   const allIndexPatterns = await Promise.allSettled(
     missingIds.map((id) => indexPatternsService.get(id))
   );
diff --git a/x-pack/plugins/lens/public/indexpattern_datasource/operations/__mocks__/index.ts b/x-pack/plugins/lens/public/indexpattern_datasource/operations/__mocks__/index.ts
index dfb86ec16da30..6ac208913af2e 100644
--- a/x-pack/plugins/lens/public/indexpattern_datasource/operations/__mocks__/index.ts
+++ b/x-pack/plugins/lens/public/indexpattern_datasource/operations/__mocks__/index.ts
@@ -19,6 +19,7 @@ jest.spyOn(actualHelpers, 'getErrorMessages');
 
 export const {
   getAvailableOperationsByMetadata,
+  memoizedGetAvailableOperationsByMetadata,
   getOperations,
   getOperationDisplay,
   getOperationTypesForField,
diff --git a/x-pack/plugins/lens/public/indexpattern_datasource/operations/operations.ts b/x-pack/plugins/lens/public/indexpattern_datasource/operations/operations.ts
index 63671fe35e99e..140ebc813f6c1 100644
--- a/x-pack/plugins/lens/public/indexpattern_datasource/operations/operations.ts
+++ b/x-pack/plugins/lens/public/indexpattern_datasource/operations/operations.ts
@@ -5,7 +5,7 @@
  * 2.0.
  */
 
-import _ from 'lodash';
+import { memoize } from 'lodash';
 import { OperationMetadata } from '../../types';
 import {
   operationDefinitionMap,
@@ -187,3 +187,5 @@ export function getAvailableOperationsByMetadata(indexPattern: IndexPattern) {
 
   return Object.values(operationByMetadata);
 }
+
+export const memoizedGetAvailableOperationsByMetadata = memoize(getAvailableOperationsByMetadata);