indexer: add reloadIfNeeded helper

Author: jaipaljadeja

packages/indexer/src/index.ts

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

packages/indexer/src/indexer.ts

@@ -33,7 +33,7 @@ import {
 import { createIndexerMetrics, createTracer } from "./otel";
 import type { IndexerPlugin } from "./plugins";
 import { useInternalContext } from "./plugins/context";
-import { ReloadIndexerRequest } from "./utils";
+import { reloadIfNeeded } from "./utils";
 
 export type UseMiddlewareFunction = (
   fn: MiddlewareFunction<IndexerContext>,
@@ -318,10 +318,7 @@ export async function run<TFilter, TBlock>(
               },
             );
 
-            if (context._reload) {
-              context._reload = false;
-              throw new ReloadIndexerRequest();
-            }
+            reloadIfNeeded();
 
             await middleware(context, async () => {
               let block: TBlock | null;
@@ -401,10 +398,7 @@ export async function run<TFilter, TBlock>(
               }
             });
 
-            if (context._reload) {
-              context._reload = false;
-              throw new ReloadIndexerRequest();
-            }
+            reloadIfNeeded();
 
             span.end();
           });
@@ -444,7 +438,11 @@ export async function run<TFilter, TBlock>(
         }
         case "heartbeat": {
           await tracer.startActiveSpan("message heartbeat", async (span) => {
+            reloadIfNeeded();
+
             await indexer.hooks.callHook("message:heartbeat");
+            reloadIfNeeded();
+
             span.end();
           });
           break;

packages/indexer/src/utils/index.ts

@@ -5,6 +5,14 @@ export function reloadIndexer() {
   context._reload = true;
 }
 
+export function reloadIfNeeded() {
+  const context = useIndexerContext();
+  if (context._reload) {
+    context._reload = false;
+    throw new ReloadIndexerRequest();
+  }
+}
+
 export class ReloadIndexerRequest extends Error {
   constructor(message?: string) {
     super(message);