fix update balance

airalab · Oct 9, 2024 · 4a37677 · 4a37677
1 parent 5c26bf3
commit 4a37677
Showing 1 changed file with 16 additions and 12 deletions.
diff --git a/src/hooks/useBalance.js b/src/hooks/useBalance.js
@@ -5,17 +5,21 @@ export const useBalance = (account) => {
   const balance = ref(null);
   const { isReady, getInstance } = useRobonomics();
   let unsubscribe;
-  watch([account, isReady], async () => {
-    if (unsubscribe) {
-      unsubscribe();
-    }
-    if (!isReady.value) {
-      return;
-    }
-    const robonomics = getInstance();
-    unsubscribe = await robonomics.account.getBalance(account.value, (r) => {
-      balance.value = r.free.sub(r.frozen).toNumber();
-    });
-  });
+  watch(
+    [account, isReady],
+    async () => {
+      if (unsubscribe) {
+        unsubscribe();
+      }
+      if (!isReady.value || !account.value) {
+        return;
+      }
+      const robonomics = getInstance();
+      unsubscribe = await robonomics.account.getBalance(account.value, (r) => {
+        balance.value = r.free.sub(r.frozen).toNumber();
+      });
+    },
+    { immediate: true }
+  );
   return { balance, unsubscribe };
 };