Add index pattern info when loading embeddable (opensearch-project#2363)

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]> Signed-off-by: abbyhu2000 <[email protected]>
ananzh · Sep 15, 2022 · dc0d7d5 · dc0d7d5
1 parent 77b6068
commit dc0d7d5
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 ?? '';