fix: use getLogsSafe

Author: doomsower

src/services/AddressProviderService.ts

@@ -1,12 +1,13 @@
 import type { Address, NetworkType } from "@gearbox-protocol/sdk-gov";
 import { ADDRESS_PROVIDER } from "@gearbox-protocol/sdk-gov";
 import { iAddressProviderV3Abi } from "@gearbox-protocol/types/abi";
-import { getContract, hexToString, stringToHex } from "viem";
+import { getAbiItem, hexToString, stringToHex } from "viem";
 
 import type { Config } from "../config/index.js";
 import { DI } from "../di.js";
 import { type ILogger, Logger } from "../log/index.js";
 import { TxParser } from "../utils/ethers-6-temp/txparser/index.js";
+import { getLogsSafe } from "../utils/getLogsSafe.js";
 import type Client from "./Client.js";
 
 const AP_SERVICES = [
@@ -49,19 +50,17 @@ export class AddressProviderService {
       ? ` (overrides default ${ADDRESS_PROVIDER[this.config.network]})`
       : "";
 
-    const contract = getContract({
+    const toBlock = await this.client.pub.getBlockNumber();
+
+    const logs = await getLogsSafe(this.client.pub, {
       address,
-      abi: iAddressProviderV3Abi,
-      client: this.client.pub,
+      event: getAbiItem({ abi: iAddressProviderV3Abi, name: "SetAddress" }),
+      args: { key: AP_SERVICES.map(s => stringToHex(s, { size: 32 })) },
+      fromBlock: AP_BLOCK_BY_NETWORK[this.config.network],
+      toBlock,
+      strict: true,
     });
 
-    const logs = await contract.getEvents.SetAddress(
-      {
-        key: AP_SERVICES.map(s => stringToHex(s, { size: 32 })),
-      },
-      { fromBlock: AP_BLOCK_BY_NETWORK[this.config.network], strict: true },
-    );
-
     for (const { args } of logs) {
       const { key, version, value } = args;
       const service = hexToString(key!, { size: 32 }) as AddressProviderKey;

src/services/OracleServiceV3.ts

@@ -14,6 +14,7 @@ import { DI } from "../di.js";
 import type { ILogger } from "../log/index.js";
 import { Logger } from "../log/index.js";
 import { TxParser } from "../utils/ethers-6-temp/txparser/index.js";
+import { getLogsSafe } from "../utils/getLogsSafe.js";
 import type { AddressProviderService } from "./AddressProviderService.js";
 import type Client from "./Client.js";
 
@@ -191,7 +192,7 @@ export default class OracleServiceV3 {
       return;
     }
     this.log.debug(`updating price feeds in [${this.#lastBlock}, ${toBlock}]`);
-    let logs = await this.client.pub.getLogs({
+    let logs = await getLogsSafe(this.client.pub, {
       address: this.oracle,
       events: iPriceOracleV3EventsAbi,
       fromBlock: BigInt(this.#lastBlock),

src/utils/getLogsSafe.ts

@@ -0,0 +1,120 @@
+import type { AbiEvent } from "abitype";
+import {
+  type BlockNumber,
+  type Chain,
+  type Client,
+  type GetLogsParameters,
+  type GetLogsReturnType,
+  HttpRequestError,
+  type Transport,
+} from "viem";
+import { getLogs } from "viem/actions";
+
+interface BlockRange {
+  fromBlock: bigint;
+  toBlock: bigint;
+}
+
+export async function getLogsSafe<
+  chain extends Chain | undefined,
+  const abiEvent extends AbiEvent | undefined = undefined,
+  const abiEvents extends
+    | readonly AbiEvent[]
+    | readonly unknown[]
+    | undefined = abiEvent extends AbiEvent ? [abiEvent] : undefined,
+  strict extends boolean | undefined = undefined,
+>(
+  client: Client<Transport, chain>,
+  params: GetLogsParameters<
+    abiEvent,
+    abiEvents,
+    strict,
+    BlockNumber,
+    BlockNumber
+  > = {},
+): Promise<
+  GetLogsReturnType<abiEvent, abiEvents, strict, BlockNumber, BlockNumber>
+> {
+  try {
+    const events = await getLogs<
+      chain,
+      abiEvent,
+      abiEvents,
+      strict,
+      BlockNumber,
+      BlockNumber
+    >(client, params);
+    return events;
+  } catch (e) {
+    const fromBlock = params.fromBlock as bigint;
+    const toBlock = params.toBlock as bigint;
+    const bisected = tryBisectBlockRange({ fromBlock, toBlock }, e);
+    if (!bisected) {
+      throw e;
+    }
+
+    const [left, right] = await Promise.all([
+      getLogs(client, { ...params, ...bisected[0] } as any),
+      getLogs(client, { ...params, ...bisected[1] } as any),
+    ]);
+    return [...left, ...right] as GetLogsReturnType<
+      abiEvent,
+      abiEvents,
+      strict,
+      BlockNumber,
+      BlockNumber
+    >;
+  }
+}
+
+function tryBisectBlockRange(
+  { fromBlock, toBlock }: BlockRange,
+  e: any,
+): [BlockRange, BlockRange] | undefined {
+  const alchemyMid = checkForAlchemyBlockRange(e);
+  if (alchemyMid && alchemyMid > fromBlock && alchemyMid < toBlock) {
+    return [
+      { fromBlock, toBlock: alchemyMid },
+      { fromBlock: alchemyMid + 1n, toBlock },
+    ];
+  }
+
+  const blockRangeErrors = [
+    "query exceeds max block",
+    "range is too large",
+    "eth_getLogs is limited to",
+    "eth_getLogs requests with up to",
+  ];
+
+  if (
+    e instanceof Error &&
+    blockRangeErrors.some(errorText => e.message.includes(errorText))
+  ) {
+    const middle = (fromBlock + toBlock) / 2n;
+    return [
+      { fromBlock, toBlock: middle },
+      { fromBlock: middle + 1n, toBlock },
+    ];
+  }
+  return undefined;
+}
+
+const ALCHEMY_BLOCK_RANGE_REGEX =
+  /this block range should work: \[(0x[0-9a-fA-F]+),\s*(0x[0-9a-fA-F]+)\]/;
+
+function checkForAlchemyBlockRange(e: any): bigint | undefined {
+  if (e instanceof HttpRequestError) {
+    try {
+      // exmple of alchemy error:
+      // Details: {"code":-32600,"message":"You can make eth_getLogs requests with up to a 10000 block range. Based on your parameters, this block range should work: [0x9538b4, 0x955fc3]"}
+      const err = JSON.parse(e.details);
+      if (typeof err.message === "string") {
+        const match = err.message.match(ALCHEMY_BLOCK_RANGE_REGEX);
+        if (match) {
+          return BigInt(match[2]);
+        }
+      }
+    } catch {}
+  }
+  return undefined;
+}

src/utils/index.ts

@@ -3,6 +3,7 @@ export * from "./bigint-utils.js";
 export * from "./detect-network.js";
 export * from "./etherscan.js";
 export * from "./formatters.js";
+export * from "./getLogsSafe.js";
 export * from "./retry.js";
 export * from "./simulateMulticall.js";
 export * from "./types.js";