initiate balances from genesis file

Agoric · Jun 14, 2024 · 518e280 · 518e280
1 parent 31a4457
commit 518e280
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 60 deletions.
diff --git a/src/mappings/constants.ts b/src/mappings/constants.ts
@@ -70,7 +70,8 @@ export const BALANCE_FIELDS = {
   withdraw_commission: 'validator',
 };
 
-export const FETCH_ACCOUNTS_URL =
-  'https://main-a.api.agoric.net:443/cosmos/auth/v1beta1/accounts';
+export const FETCH_ACCOUNTS_URL = 'https://main-a.api.agoric.net:443/cosmos/auth/v1beta1/accounts';
 export const GET_FETCH_BALANCE_URL = (address: string) =>
-  `https://main-a.api.agoric.net:443/cosmos/bank/v1beta1/balances/${address}`;
+  `https://main-a.api.agoric.net:443/cosmos/bank/v1beta1/balances/${address}`;
+
+export const GENESIS_URL = 'https://agoric-rpc.polkachu.com/genesis';
diff --git a/src/mappings/mappingHandlers.ts b/src/mappings/mappingHandlers.ts
@@ -29,6 +29,7 @@ import {
   BALANCE_FIELDS,
   FETCH_ACCOUNTS_URL,
   GET_FETCH_BALANCE_URL,
+  GENESIS_URL,
 } from './constants';
 import { psmEventKit } from './events/psm';
 import { boardAuxEventKit } from './events/boardAux';
@@ -354,65 +355,28 @@ const findAddress = (obj: any) => {
 
   return null;
 };
-const extractAddresses = (accounts: (BaseAccount | ModuleAccount | VestingAccount)[]): Array<string> => {
-  return accounts
-    .map((account) => {
-      return findAddress(account);
-    })
-    .filter((address) => address !== null);
-};
 
-const fetchBalance = async (address: string): Promise<BalancesResponse | void> => {
+export const initiateBalancesTable = async (block: CosmosBlock): Promise<void> => {
   try {
-    const response = await crossFetch(GET_FETCH_BALANCE_URL(address));
-    const balance: BalancesResponse = await response.json();
-    return balance;
-  } catch (error) {
-    logger.error(`Error fetching balance for address ${address}: ${error}`);
-  }
-};
-
-const saveAccountBalances = async (address: string, balances: Balance[]) => {
-  for (let { denom, amount } of balances) {
-    const newBalance = new Balances(address);
-    newBalance.address = address;
-    newBalance.balance = BigInt(amount);
-    newBalance.denom = denom;
-
-    await newBalance.save();
-  }
-};
-
-const fetchAndSaveBalances = async (accountAddresses: string[]) => {
-  const fetchPromises = accountAddresses.map(async (address) => {
-    const response = await fetchBalance(address);
-    if (response) {
-      return saveAccountBalances(address, response.balances);
+    logger.info(`Initiating Balances Table`);
+    const response = await crossFetch(GENESIS_URL);
+    const parsedResponse = await response.json();
+    const data = parsedResponse.genesis.app_state.bank;
+
+    for (let element of data.balances) {
+      let newBalance;
+      for (const coin of element.coins) {
+        newBalance = new Balances(`${element.address}-${coin.denom}`);
+        newBalance.address = element.address;
+        newBalance.balance = BigInt(coin.amount);
+        newBalance.denom = coin.denom;
+
+        await newBalance.save();
+      }
     }
-  });
-
-  await Promise.all(fetchPromises);
-};
 
-export const initiateBalancesTable = async (block: CosmosBlock): Promise<void> => {
-  let nextKey: string | null = null;
-  let offset: number = 0;
-  const limit: number = 1000;
-
-  logger.info('Initiate Balances Table');
-  do {
-    logger.info('Fetching Accounts');
-    const [accounts, nextPaginationKey] = await fetchAccounts(String(offset), String(limit));
-    logger.info(`Accounts Fetched: ${accounts.length}`);
-    logger.info(`Pagination Key ${nextKey}`);
-    nextKey = nextPaginationKey;
-    if (accounts) {
-      logger.info('Storing Balances');
-      const accountAddresses = extractAddresses(accounts).filter(Boolean);
-      logger.info(`Accounts Extracted: ${accountAddresses.length}`);
-      await fetchAndSaveBalances(accountAddresses);
-      logger.info('Balances Stored Successfully');
-      offset += accounts.length; 
-    }
-  } while (nextKey);
+    logger.info(`Balances Table Initiated`);
+  } catch (error) {
+    logger.error(`Error initiating balances table: ${error}`);
+  }
 };