Add index pattern info when loading embeddable

Since index pattern is not included in the visualization state anymore, need to add the index pattern information so the embeddable can be loaded correctly. Signed-off-by: abbyhu2000 <[email protected]>
opensearch-project · Sep 15, 2022 · 0eb7e63 · 0eb7e63
1 parent ab99eac
commit 0eb7e63
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/plugins/wizard/common/wizard_saved_object_attributes.ts b/src/plugins/wizard/common/wizard_saved_object_attributes.ts
@@ -13,4 +13,7 @@ export interface WizardSavedObjectAttributes extends SavedObjectAttributes {
   visualizationState?: string;
   styleState?: string;
   version: number;
+  searchSourceFields?: {
+    index?: string;
+  };
 }
diff --git a/src/plugins/wizard/public/embeddable/wizard_embeddable.tsx b/src/plugins/wizard/public/embeddable/wizard_embeddable.tsx
@@ -116,8 +116,15 @@ export class WizardEmbeddable extends Embeddable<SavedObjectEmbeddableInput, Wiz
       return;
     }
     const { visualization, style } = this.serializedState;
+
+    const vizStateWithoutIndex = JSON.parse(visualization);
+    const visualizationState = {
+      searchField: vizStateWithoutIndex.searchField,
+      activeVisualization: vizStateWithoutIndex.activeVisualization,
+      indexPattern: this.savedWizard?.searchSourceFields?.index,
+    };
     const rootState = {
-      visualization: JSON.parse(visualization),
+      visualization: visualizationState,
       style: JSON.parse(style),
     };
     const visualizationName = rootState.visualization?.activeVisualization?.name ?? '';