Update useDataPolling hook to show only one notification

trilitech · Jun 14, 2024 · 7095276 · 7095276
1 parent adc598f
commit 7095276
Showing 1 changed file with 12 additions and 25 deletions.
diff --git a/src/utils/dataPolling/useDataPolling.ts b/src/utils/dataPolling/useDataPolling.ts
@@ -14,31 +14,18 @@ import { assetsActions } from "../redux/slices/assetsSlice";
 export const useDataPolling = () => {
   const dispatch = useAppDispatch();
 
-  const { dataUpdatedAt: isMultisigsUpdatedAt, isFetching: isMultisigsFetching } =
-    usePollMultisigs();
-  const { dataUpdatedAt: isPendingOperationsUpdatedAt, isFetching: isPendingOperationsFetching } =
-    usePollPendingOperations();
-  const { dataUpdatedAt: isTezBalancesUpdatedAt, isFetching: isTezBalancesFetching } =
-    usePollTezBalances();
-  const { dataUpdatedAt: isTokenBalancesUpdatedAt, isFetching: isTokenBalancesFetching } =
-    usePollTokenBalances();
-
-  usePollConversionRate();
-  usePollBlockLevel();
-  usePollBakers();
-
-  const isFetching =
-    isMultisigsFetching ||
-    isPendingOperationsFetching ||
-    isTezBalancesFetching ||
-    isTokenBalancesFetching;
-
-  const lastUpdatedAt = Math.max(
-    isMultisigsUpdatedAt,
-    isPendingOperationsUpdatedAt,
-    isTezBalancesUpdatedAt,
-    isTokenBalancesUpdatedAt
-  );
+  const pollers = [
+    usePollMultisigs(),
+    usePollPendingOperations(),
+    usePollTezBalances(),
+    usePollTokenBalances(),
+    usePollConversionRate(),
+    usePollBlockLevel(),
+    usePollBakers(),
+  ];
+
+  const isFetching = pollers.some(poller => poller.isFetching);
+  const lastUpdatedAt = Math.max(...pollers.map(poller => poller.dataUpdatedAt));
 
   useEffect(() => {
     dispatch(assetsActions.setIsLoading(isFetching));