Fix Gate-io, update helper (DefiLlama#15191)

Author: 0xpeluche

projects/gate-io/index.js

@@ -1671,7 +1671,5 @@ unsupportedChains.forEach(chain => delete config[chain]);
 
 module.exports = mergeExports([
   cexExports(config),
-  { ethereum: { tvl: getStakedEthTVL({ withdrawalAddress: '0x287a66c7d9cba7504e90fa638911d74c4dc6a147' }) } },
-  { ethereum: { tvl: getStakedEthTVL({ withdrawalAddress: '0xbcf03ce48091e6b820a7c33e166e5d0109d8e712' }) } },
-  { ethereum: { tvl: getStakedEthTVL({ withdrawalAddress: '0x7a3f9b7120386249528c93e5eb373b78e54d5ba9' }) } },
+  { ethereum: { tvl: getStakedEthTVL({ withdrawalAddresses: ['0x287a66c7d9cba7504e90fa638911d74c4dc6a147', '0xbcf03ce48091e6b820a7c33e166e5d0109d8e712', '0x7a3f9b7120386249528c93e5eb373b78e54d5ba9'], sleepTime: 10000, size: 200 }) } },
 ]);

projects/helper/utils.js

@@ -320,26 +320,30 @@ function once(func) {
   return wrapped
 }
 
+function getStakedEthTVL({ withdrawalAddress, withdrawalAddresses, skipValidators = 0, size = 200, sleepTime = 10000 }) {
+return async (api) => {
+    const addresses = withdrawalAddresses ?? [withdrawalAddress];
+
+    for (const addr of addresses) {
+      let fetchedValidators = skipValidators;
+
+      api.sumTokens({ owner: addr, tokens: [nullAddress] });
+
+      do {
+        const validators = (
+          await http.get(
+            `https://beaconcha.in/api/v1/validator/withdrawalCredentials/${addr}?limit=${size}&offset=${fetchedValidators}`
+          )
+        ).data.map((i) => i.publickey);
+
+        fetchedValidators += validators.length;
+        api.log(`Fetching balances for validators (${addr})`, validators.length);
+        await addValidatorBalance(validators);
+        if (sleepTime > 0) await sleep(sleepTime);
+      } while (fetchedValidators % size === 0);
+    }
 
-function getStakedEthTVL({ withdrawalAddress, skipValidators = 0 }) {
-  return async (api) => {
-    let fetchedValidators = skipValidators;
-    let size = 200;
-    api.sumTokens({ owner: withdrawalAddress, tokens: [nullAddress] })
-    do {
-      const validators = (
-        await http.get(
-          `https://beaconcha.in/api/v1/validator/withdrawalCredentials/${withdrawalAddress}?limit=${size}&offset=${fetchedValidators}`
-        )
-      ).data.map((i) => i.publickey);
-      fetchedValidators += validators.length;
-      api.log("Fetching balances for validators", validators.length);
-      await addValidatorBalance(validators);
-      await sleep(10000);
-    } while (fetchedValidators % size === 0);
-
-    return api.getBalances()
-
+    return api.getBalances();
 
     async function addValidatorBalance(validators) {
       if (validators.length > 100) {
@@ -352,10 +356,9 @@ function getStakedEthTVL({ withdrawalAddress, skipValidators = 0 }) {
         indicesOrPubkey: validators.join(","),
       });
 
-
       data.forEach((i) => api.addGasToken(i.balance * 1e9));
     }
-  }
+  };
 }