acceleration create button update

Signed-off-by: Shenoy Pratik <sgguruda@amazon.com>
ps48 · Mar 25, 2024 · 7c303c2 · 7c303c2
1 parent c03c676
commit 7c303c2
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/...omponents/manage/accelerations/create_accelerations_flyout/create/create_acceleration.tsx b/...omponents/manage/accelerations/create_accelerations_flyout/create/create_acceleration.tsx
@@ -44,6 +44,7 @@ export interface CreateAccelerationProps {
   resetFlyout: () => void;
   databaseName?: string;
   tableName?: string;
+  refreshHandler?: () => void;
 }
 
 export const CreateAcceleration = ({

diff --git a/...ts/manage/accelerations/create_accelerations_flyout/create/create_acceleration_button.tsx b/...ts/manage/accelerations/create_accelerations_flyout/create/create_acceleration_button.tsx
@@ -62,9 +62,11 @@ export const CreateAccelerationButton = ({
     }
   }, [directqueryLoadStatus]);
 
-  return (
+  const createAccelerationBtn = (
     <EuiButton onClick={createAcceleration} fill isLoading={isLoading}>
       {isLoading ? 'Creating acceleration' : 'Create acceleration'}
     </EuiButton>
   );
+
+  return createAccelerationBtn;
 };
diff --git a/public/framework/datasources/direct_query_hook.tsx b/public/framework/datasources/direct_query_hook.tsx
@@ -74,6 +74,7 @@ export const useDirectQuery = () => {
       setLoadStatus(status);
       stopLoading();
     } else if (status === DirectQueryLoadingStatus.FAILED) {
+      setLoadStatus(status);
       stopLoading();
       const formattedError = formatError(
         '',