add null guard for useSubAssetBalance, #5180

opensquare-network · Dec 11, 2024 · 3d46e06 · 3d46e06
1 parent 43e3c9b
commit 3d46e06
Showing 1 changed file with 19 additions and 5 deletions.
diff --git a/packages/next-common/components/assets/useSubAssetBalance.js b/packages/next-common/components/assets/useSubAssetBalance.js
@@ -64,15 +64,29 @@ export function useQueryAddressAssets(address) {
 }
 
 export default function useSubAssetBalance(assetId, address) {
+  const [transferrable, setTransferrable] = useState(null);
+  const [isFrozen, setIsFrozen] = useState(null);
+  const [balance, setBalance] = useState(null);
+
   const { result, loading } = useSubStorage("assets", "account", [
-    assetId,
+    String(assetId),
     address,
   ]);
 
-  const data = result?.unwrap();
-  const balance = data?.balance?.toString() || 0;
-  const isFrozen = data?.status?.isFrozen;
-  const transferrable = isFrozen ? 0 : balance;
+  useEffect(() => {
+    if (loading || result?.isNone || isNil(assetId) || isNil(address)) {
+      return;
+    }
+
+    const data = result?.unwrap();
+    const balance = data?.balance?.toString() || 0;
+    const isFrozen = data?.status?.isFrozen;
+    const transferrable = isFrozen ? 0 : balance;
+
+    setBalance(balance);
+    setTransferrable(transferrable);
+    setIsFrozen(isFrozen);
+  }, [loading, result, assetId, address]);
 
   return {
     result: {