enh: improve types in ingestion.ts (#4)

* enh: improve types in ingestion.ts * Align import
BemiHQ · Jun 2, 2024 · 055b3dc · 055b3dc
1 parent 34d0423
commit 055b3dc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/core/src/ingestion.ts b/core/src/ingestion.ts
@@ -1,4 +1,5 @@
 import { MikroORM } from '@mikro-orm/postgresql';
+import { Consumer } from 'nats';
 
 import { logger } from './logger'
 import { NatsMessage } from './nats'
@@ -34,7 +35,7 @@ const persistFetchedRecords = (
 
 const fetchNatsMessages = async (
   { consumer, fetchBatchSize, lastStreamSequence }:
-  { consumer: any, fetchBatchSize: number, lastStreamSequence: number | null }
+  { consumer: Consumer, fetchBatchSize: number, lastStreamSequence: number | null }
 ) => {
   const natsMessageBySequence: { [sequence: number]: NatsMessage } = {}
   let pendingMessageCount = 0
@@ -65,7 +66,7 @@ export const runIngestionLoop = async (
     useBuffer = false,
   }: {
     orm: MikroORM,
-    consumer: any,
+    consumer: Consumer,
     fetchBatchSize?: number,
     insertBatchSize?: number,
     useBuffer?: boolean,