fix: disable dryrun autofetch (#36941)

set useQuery to be enabled after action button is clicked instead of automatically closes issue #36119
apache · Jan 22, 2024 · e0f8c6e · e0f8c6e
1 parent 10c04a4
commit e0f8c6e
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 2 deletions.
diff --git a/airflow/www/static/js/api/useClearTaskDryRun.ts b/airflow/www/static/js/api/useClearTaskDryRun.ts
@@ -39,6 +39,7 @@ const useClearTaskDryRun = ({
   recursive,
   failed,
   mapIndexes = [],
+  enabled = false,
 }: {
   dagId: string;
   runId: string;
@@ -52,6 +53,7 @@ const useClearTaskDryRun = ({
   recursive: boolean;
   failed: boolean;
   mapIndexes?: number[];
+  enabled?: boolean;
 }) =>
   useQuery(
     [
@@ -101,7 +103,8 @@ const useClearTaskDryRun = ({
           },
         }
       );
-    }
+    },
+    { enabled }
   );
 
 export default useClearTaskDryRun;
diff --git a/airflow/www/static/js/api/useMarkTaskDryRun.ts b/airflow/www/static/js/api/useMarkTaskDryRun.ts
@@ -36,6 +36,7 @@ const useMarkTaskDryRun = ({
   upstream,
   downstream,
   mapIndexes = [],
+  enabled = false,
 }: {
   dagId: string;
   runId: string;
@@ -47,6 +48,7 @@ const useMarkTaskDryRun = ({
   upstream: boolean;
   downstream: boolean;
   mapIndexes?: number[];
+  enabled?: boolean;
 }) =>
   useQuery(
     [
@@ -84,7 +86,8 @@ const useMarkTaskDryRun = ({
       return axios.get<AxiosResponse, MinimalTaskInstance[]>(confirmUrl, {
         params,
       });
-    }
+    },
+    { enabled }
   );
 
 export default useMarkTaskDryRun;
diff --git a/airflow/www/static/js/dag/details/taskInstance/taskActions/ClearInstance.tsx b/airflow/www/static/js/dag/details/taskInstance/taskActions/ClearInstance.tsx
@@ -102,6 +102,7 @@ const ClearModal = ({
       recursive,
       failed,
       mapIndexes,
+      enabled: isOpen,
     });
 
   const { mutateAsync: clearTask, isLoading } = useClearTask({

diff --git a/airflow/www/static/js/dag/details/taskInstance/taskActions/MarkInstanceAs.tsx b/airflow/www/static/js/dag/details/taskInstance/taskActions/MarkInstanceAs.tsx
@@ -107,6 +107,7 @@ const MarkAsModal = ({
       upstream,
       downstream,
       mapIndexes,
+      enabled: isOpen,
     }
   );