From a7e150974c8a11915bef197e59749130458be588 Mon Sep 17 00:00:00 2001
From: popcorny <celu@infuseai.io>
Date: Wed, 25 Sep 2024 16:21:34 +0800
Subject: [PATCH] Prevent aggregate runs multiple times

Signed-off-by: popcorny <celu@infuseai.io>
---
 js/src/lib/hooks/useRun.tsx | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/js/src/lib/hooks/useRun.tsx b/js/src/lib/hooks/useRun.tsx
index be01bc6e..3d1fb22d 100644
--- a/js/src/lib/hooks/useRun.tsx
+++ b/js/src/lib/hooks/useRun.tsx
@@ -31,14 +31,16 @@ export const useRun = (runId?: string): UseRunResult => {
 
   useEffect(() => {
     if (error || run?.result || run?.error) {
-      setIsPolling(false);
-      if (run?.type === "row_count_diff") {
-        refetchRunsAggregated();
+      if (isPolling) {
+        setIsPolling(false);
+        if (run?.type === "row_count_diff") {
+          refetchRunsAggregated();
+        }
       }
     } else {
       setIsPolling(true);
     }
-  }, [run, error, refetchRunsAggregated]);
+  }, [run, error, isPolling, refetchRunsAggregated]);
 
   const onCancel = useCallback(async () => {
     setAborting(true);