feat: optimize getAddressesBalances to minimize RPC calls

RealToken-Community · Jan 1, 2025 · ea9d1da · ea9d1da
1 parent b56b8e2
commit ea9d1da
Show file tree

Hide file tree

Showing 4 changed files with 45 additions and 33 deletions.
diff --git a/src/hooks/useREG.ts b/src/hooks/useREG.ts
@@ -64,12 +64,20 @@ const getREG = async (
     ERC20ABI,
     GnosisRpcProvider,
   )
-  const availableBalance = await getAddressesBalances(
+  let availableBalance = await getAddressesBalances(
     REG_ContractAddress,
     addressList,
-    providers,
+    GnosisRpcProvider,
   )
 
+  if (includeETH) {
+    availableBalance += await getAddressesBalances(
+      REG_ContractAddress,
+      addressList,
+      EthereumRpcProvider,
+    )
+  }
+
   const regVaultAbiGetUserGlobalStateOnly =
     getRegVaultAbiGetUserGlobalStateOnly()
 

diff --git a/src/hooks/useREGVotingPower.ts b/src/hooks/useREGVotingPower.ts
@@ -20,7 +20,6 @@ const getRegVotingPower = async (
   addressList: string[],
 ): Promise<REGVotingPowertoken> => {
   const { GnosisRpcProvider } = await initializeProviders()
-  const providers = [GnosisRpcProvider]
   const RegVotingPowerContract = new Contract(
     RegVotingPower_Gnosis_ContractAddress,
     ERC20ABI,
@@ -29,7 +28,7 @@ const getRegVotingPower = async (
   const totalAmount = await getAddressesBalances(
     RegVotingPower_Gnosis_ContractAddress,
     addressList,
-    providers,
+    GnosisRpcProvider,
   )
 
   const contractRegVotePowerTotalSupply =

diff --git a/src/hooks/useRWA.ts b/src/hooks/useRWA.ts
@@ -42,7 +42,6 @@ const getRWA = async (
 ): Promise<RWARealtoken> => {
   const { GnosisRpcProvider /* , EthereumRpcProvider */ } =
     await initializeProviders()
-  const providers = [GnosisRpcProvider]
 
   const contractRwa_Gnosis = new Contract(
     RWA_ContractAddress,
@@ -52,7 +51,7 @@ const getRWA = async (
   const totalAmount = await getAddressesBalances(
     RWA_ContractAddress,
     addressList,
-    providers,
+    GnosisRpcProvider,
   )
   const RwaContractTotalSupply = await contractRwa_Gnosis.totalSupply()
   const totalTokens = Number(RwaContractTotalSupply) / 10 ** RWAtokenDecimals

diff --git a/src/utils/blockchain/erc20Infos.ts b/src/utils/blockchain/erc20Infos.ts
@@ -1,13 +1,15 @@
+import { AaveV3Ethereum, AaveV3Gnosis } from '@bgd-labs/aave-address-book'
+
 import { Contract, JsonRpcProvider } from 'ethers'
 
 import { getErc20AbiBalanceOfOnly } from 'src/utils/blockchain/ERC20'
 
-import { batchCallOneContractOneFunctionMultipleParams } from './contract'
+import { WalletBalanceProviderABI } from './abi/WalletBalanceProviderABI'
 
 const getAddressesBalances = async (
   contractAddress: string,
   addressList: string[],
-  providers: JsonRpcProvider[],
+  provider: JsonRpcProvider,
   consoleWarnOnError = false,
 ) => {
   let totalAmount = 0
@@ -20,38 +22,42 @@ const getAddressesBalances = async (
       consoleWarnOnError && console.error('Invalid address list')
       return totalAmount
     }
-    if (!providers?.length) {
+    if (!provider) {
       consoleWarnOnError && console.error('Invalid providers')
       return totalAmount
     }
     const erc20AbiBalanceOfOnly = getErc20AbiBalanceOfOnly()
     if (!erc20AbiBalanceOfOnly) {
       throw new Error('balanceOf ABI not found')
     }
-    const balancesPromises = providers.map((provider: JsonRpcProvider) => {
-      const Erc20BalanceContract = new Contract(
-        contractAddress,
-        erc20AbiBalanceOfOnly,
-        provider,
-      )
-      const balances = batchCallOneContractOneFunctionMultipleParams(
-        Erc20BalanceContract,
-        'balanceOf',
-        addressList.map((address: string) => [address as unknown as object]),
-      )
-      return balances
-    })
-
-    const balancesArray = await Promise.all(balancesPromises.flat())
-    const balances = balancesArray.flat()
-    // Sum all valid balances
-    balances.forEach((balance: object | null | undefined) => {
-      try {
-        if (balance) {
-          totalAmount += Number(balance)
-        }
-      } catch (error) {}
-    })
+
+    let walletBalanceProviderAddress
+    switch (provider._network.chainId) {
+      case 100n:
+        walletBalanceProviderAddress = AaveV3Gnosis.WALLET_BALANCE_PROVIDER
+        break
+      case 1n:
+        walletBalanceProviderAddress = AaveV3Ethereum.WALLET_BALANCE_PROVIDER
+        break
+      default:
+        consoleWarnOnError && console.error('Invalid provider')
+        return totalAmount
+    }
+
+    const balanceWalletProviderContract = new Contract(
+      walletBalanceProviderAddress,
+      WalletBalanceProviderABI,
+      provider,
+    )
+
+    const balances = await balanceWalletProviderContract.batchBalanceOf(
+      addressList,
+      [contractAddress],
+    )
+    totalAmount = balances.reduce(
+      (acc: number, b: bigint) => acc + Number(b),
+      0,
+    )
     return totalAmount
   } catch (error) {
     console.warn('Failed to get balances', error)