fix: disable gap check during replay mode

hirosystems · May 14, 2024 · f1f8d44 · f1f8d44
1 parent b89802d
commit f1f8d44
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/src/env.ts b/src/env.ts
@@ -53,6 +53,8 @@ const schema = Type.Object({
     { default: 'default', replay: 'replay' },
     { default: 'default' }
   ),
+  /** If the API should automatically shut down when Ordhook ingestion mode is `replay` */
+  ORDHOOK_REPLAY_INGESTION_MODE_AUTO_SHUTDOWN: Type.Boolean({ default: true }),
 
   PGHOST: Type.String(),
   PGPORT: Type.Number({ default: 5432, minimum: 0, maximum: 65535 }),

diff --git a/src/ordhook/server.ts b/src/ordhook/server.ts
@@ -59,7 +59,11 @@ export async function startOrdhookServer(args: { db: PgStore }): Promise<Chainho
   const server = new ChainhookEventObserver(serverOpts, ordhookOpts);
   await server.start(predicates, async (uuid: string, payload: Payload) => {
     const streamed = payload.chainhook.is_streaming_blocks;
-    if (ENV.ORDHOOK_INGESTION_MODE === 'replay' && streamed) {
+    if (
+      ENV.ORDHOOK_INGESTION_MODE === 'replay' &&
+      ENV.ORDHOOK_REPLAY_INGESTION_MODE_AUTO_SHUTDOWN &&
+      streamed
+    ) {
       logger.info(`OrdhookServer finished replaying blocks, shutting down`);
       return shutdown();
     }

diff --git a/src/pg/pg-store.ts b/src/pg/pg-store.ts
@@ -82,7 +82,8 @@ export class PgStore extends BasePgStore {
    */
   async updateInscriptions(payload: BitcoinPayload): Promise<void> {
     await this.sqlWriteTransaction(async sql => {
-      const streamed = payload.chainhook.is_streaming_blocks;
+      const streamed =
+        ENV.ORDHOOK_INGESTION_MODE === 'default' && payload.chainhook.is_streaming_blocks;
       for (const event of payload.rollback) {
         logger.info(`PgStore rollback block ${event.block_identifier.index}`);
         const time = stopwatch();