indexer: update reloadIndexer() logic with context

Author: jaipaljadeja

change/@apibara-indexer-d23b48af-f33a-40c0-afc3-44ae8d4459d6.json

@@ -0,0 +1,7 @@
+{
+  "type": "prerelease",
+  "comment": "indexer: update reloadIndexer() logic with context",
+  "packageName": "@apibara/indexer",
+  "email": "jadejajaipal5@gmail.com",
+  "dependentChangeType": "patch"
+}

examples/cli-drizzle/indexers/2-starknet.indexer.ts

@@ -1,9 +1,8 @@
+import { starknetUsdcTransfers } from "@/lib/schema";
 import { drizzleStorage, useDrizzleStorage } from "@apibara/plugin-drizzle";
 import { drizzle } from "@apibara/plugin-drizzle";
 import { StarknetStream } from "@apibara/starknet";
 import { defineIndexer } from "apibara/indexer";
-
-import { starknetUsdcTransfers } from "@/lib/schema";
 import { useLogger } from "apibara/plugins";
 import type { ApibaraRuntimeConfig } from "apibara/types";
 import { hash } from "starknet";
@@ -16,8 +15,8 @@ export default async function (runtimeConfig: ApibaraRuntimeConfig) {
 
   console.log("[2-starknet] Starknet indexer waiting.....");
 
-  // Simulating a API call that takes 5 seconds
-  await new Promise((resolve) => setTimeout(resolve, 5000));
+  // Simulating a API call that takes 500 ms
+  await new Promise((resolve) => setTimeout(resolve, 500));
 
   console.log("[2-starknet] Starknet indexer done waiting.....");
 

packages/cli/src/core/index.ts

@@ -2,4 +2,3 @@ export { createApibara } from "./apibara";
 export { build } from "./build/build";
 export { prepare } from "./build/prepare";
 export { writeTypes } from "./build/types";
-export { ReloadIndexerRequest, reloadIndexer } from "./utils";

packages/cli/src/runtime/dev.ts

@@ -1,7 +1,6 @@
-import { runWithReconnect } from "@apibara/indexer";
+import { ReloadIndexerRequest, runWithReconnect } from "@apibara/indexer";
 import { createAuthenticatedClient } from "@apibara/protocol";
 import { getRuntimeDataFromEnv } from "apibara/common";
-import { ReloadIndexerRequest } from "apibara/core";
 import { defineCommand, runMain } from "citty";
 import type { ConsolaInstance } from "consola";
 import { blueBright } from "picocolors";

packages/cli/src/runtime/start.ts

@@ -1,11 +1,10 @@
-import { runWithReconnect } from "@apibara/indexer";
+import { ReloadIndexerRequest, runWithReconnect } from "@apibara/indexer";
 import { createAuthenticatedClient } from "@apibara/protocol";
 import {
   checkForUnknownArgs,
   getProcessedRuntimeConfig,
   getRuntimeDataFromEnv,
 } from "apibara/common";
-import { ReloadIndexerRequest } from "apibara/core";
 import { defineCommand, runMain } from "citty";
 import consola, { type ConsolaInstance } from "consola";
 import { blueBright } from "picocolors";

packages/indexer/package.json

@@ -82,4 +82,4 @@
   "peerDependencies": {
     "vitest": "^1.6.0"
   }
-}
+}

packages/indexer/src/index.ts

@@ -1,2 +1,3 @@
 export * from "./indexer";
 export { useIndexerContext } from "./context";
+export * from "./utils";

packages/indexer/src/indexer.ts

@@ -33,6 +33,7 @@ import {
 import { createIndexerMetrics, createTracer } from "./otel";
 import type { IndexerPlugin } from "./plugins";
 import { useInternalContext } from "./plugins/context";
+import { ReloadIndexerRequest } from "./utils";
 
 export type UseMiddlewareFunction = (
   fn: MiddlewareFunction<IndexerContext>,
@@ -317,6 +318,11 @@ export async function run<TFilter, TBlock>(
               },
             );
 
+            if (context._reload) {
+              context._reload = false;
+              throw new ReloadIndexerRequest();
+            }
+
             await middleware(context, async () => {
               let block: TBlock | null;
 
@@ -395,6 +401,11 @@ export async function run<TFilter, TBlock>(
               }
             });
 
+            if (context._reload) {
+              context._reload = false;
+              throw new ReloadIndexerRequest();
+            }
+
             span.end();
           });
 

packages/cli/src/core/utils.ts packages/indexer/src/utils/index.tspackages/cli/src/core/utils.ts renamed to packages/indexer/src/utils/index.ts

@@ -1,5 +1,8 @@
+import { useIndexerContext } from "../context";
+
 export function reloadIndexer() {
-  throw new ReloadIndexerRequest();
+  const context = useIndexerContext();
+  context._reload = true;
 }
 
 export class ReloadIndexerRequest extends Error {