fix: add form data for empty state control to save dataset (#21944)

apache · Oct 28, 2022 · d3f930a · d3f930a
1 parent 7f563cf
commit d3f930a
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/superset-frontend/src/SqlLab/components/SaveDatasetModal/index.tsx b/superset-frontend/src/SqlLab/components/SaveDatasetModal/index.tsx
@@ -66,13 +66,20 @@ export interface ISimpleColumn {
   is_dttm?: boolean | null;
 }
 
+export type Database = {
+  backend: string;
+  id: number;
+  parameter: object;
+};
+
 export interface ISaveableDatasource {
   columns: ISimpleColumn[];
   name: string;
   dbId: number;
   sql: string;
   templateParams?: string | object | null;
   schema?: string | null;
+  database?: Database;
 }
 
 interface SaveDatasetModalProps {
@@ -283,7 +290,7 @@ export const SaveDatasetModal = ({
       createDatasource({
         schema: datasource.schema,
         sql: datasource.sql,
-        dbId: datasource.dbId,
+        dbId: datasource.dbId || datasource?.database?.id,
         templateParams,
         datasourceName: datasetName,
         columns: selectedColumns,