fewat(rwa): update the investorbalance by graphql subscription (#2777)

kadena-community · Jan 8, 2025 · e465bf3 · e465bf3
1 parent 3d7d20a
commit e465bf3
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/packages/apps/rwa-demo/src/hooks/getInvestorBalance.ts b/packages/apps/rwa-demo/src/hooks/getInvestorBalance.ts
@@ -41,8 +41,15 @@ export const useGetInvestorBalance = ({
 
     subscriptionData.events?.map((event) => {
       const params = JSON.parse(event.parameters ?? '[]');
-      if (params[0] === investorAccount && params.length === 2) {
-        setData(parseInt(params[1]));
+
+      const senderAccount = params.length >= 2 && params[1];
+      const receiverAccount = params.length >= 3 && params[2];
+
+      if (senderAccount && senderAccount.account === investorAccount) {
+        setData(senderAccount.current);
+      }
+      if (receiverAccount && receiverAccount.account === investorAccount) {
+        setData(receiverAccount.current);
       }
     });
   }, [subscriptionData]);