updating the details on screen after edit dbt

DalgoT4D · Jul 14, 2023 · 8551f7b · 8551f7b
1 parent ec30354
commit 8551f7b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/components/DBT/DBTSetup.tsx b/src/components/DBT/DBTSetup.tsx
@@ -17,6 +17,7 @@ interface DBTSetupProps {
   gitrepoUrl: string;
   schema: string;
   mode: string;
+  fetchUpdatedWorkspace: (...args: any) => any;
 }
 
 interface DBTCreateWorkspaceParams {
@@ -34,6 +35,7 @@ export const DBTSetup = ({
   gitrepoUrl,
   schema,
   mode,
+  fetchUpdatedWorkspace,
 }: DBTSetupProps) => {
   const { register, handleSubmit, reset } = useForm<DBTCreateWorkspaceParams>();
   const { data: session }: any = useSession();
@@ -138,6 +140,7 @@ export const DBTSetup = ({
       };
       try {
         await httpPut(session, 'dbt/schema/', updateSchemaPayload);
+        fetchUpdatedWorkspace();
       } catch (err: any) {
         console.error(err);
         errorToast(err.message, [], toastContext);
@@ -159,6 +162,7 @@ export const DBTSetup = ({
         setTimeout(() => {
           checkProgress(message.task_id);
         }, 1000);
+        fetchUpdatedWorkspace();
       } catch (err: any) {
         console.error(err);
         errorToast(err.message, [], toastContext);

diff --git a/src/pages/pipeline/transform.tsx b/src/pages/pipeline/transform.tsx
@@ -280,6 +280,7 @@ const Transform = () => {
               gitrepoUrl=""
               schema=""
               mode="create"
+              fetchUpdatedWorkspace={() => setRerender(!rerender)}
             />
           ) : dbtSetupStage === 'complete' && workspace ? (
             <DBTSetup
@@ -293,6 +294,7 @@ const Transform = () => {
               gitrepoUrl={workspace?.gitrepo_url}
               schema={workspace?.default_schema}
               mode="edit"
+              fetchUpdatedWorkspace={() => setRerender(!rerender)}
             />
           ) : (
             ''