fix: experiment

graphprotocol · Jul 28, 2023 · 5a38a4f · 5a38a4f
1 parent f2882c5
commit 5a38a4f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/packages/indexer-common/src/allocations/query-fees.ts b/packages/indexer-common/src/allocations/query-fees.ts
@@ -89,6 +89,7 @@ export class AllocationReceiptCollector implements ReceiptCollector {
   private voucherRedemptionBatchThreshold: BigNumber
   private voucherRedemptionMaxBatchSize: number
   private protocolNetwork: string
+  private stop: boolean
 
   constructor({
     logger,
@@ -98,6 +99,7 @@ export class AllocationReceiptCollector implements ReceiptCollector {
     allocationExchange,
     networkSpecification,
   }: AllocationReceiptCollectorOptions) {
+    this.stop = false
     this.logger = logger.child({ component: 'AllocationReceiptCollector' })
     this.metrics = registerReceiptMetrics(metrics, networkSpecification.networkIdentifier)
     this.transactionManager = transactionManager
@@ -231,6 +233,10 @@ export class AllocationReceiptCollector implements ReceiptCollector {
     }
   }
 
+  async stopCollectingReceipts(): Promise<void> {
+    this.stop = true
+  }
+
   private startReceiptCollecting() {
     this.receiptsToCollect = new DHeap<AllocationReceiptsBatch>(null, {
       compare: (t1, t2) => t1.timeout - t2.timeout,
@@ -261,7 +267,11 @@ export class AllocationReceiptCollector implements ReceiptCollector {
   }
 
   private startVoucherProcessing() {
+    const self = this
     timer(30_000).pipe(async () => {
+      if (self.stop) {
+        return
+      }
       const pendingVouchers = await this.pendingVouchers() // Ordered by value
 
       const logger = this.logger.child({})

diff --git a/packages/indexer-common/src/indexer-management/__tests__/allocations.test.ts b/packages/indexer-common/src/indexer-management/__tests__/allocations.test.ts
@@ -36,6 +36,7 @@ let managementModels: IndexerManagementModels
 let metrics: Metrics
 let queryFeeModels: QueryFeeModels
 let sequelize: Sequelize
+let network: Network
 
 const setup = async () => {
   logger = createLogger({
@@ -59,7 +60,7 @@ const setup = async () => {
     [],
   )
 
-  const network = await Network.create(
+  network = await Network.create(
     logger,
     testNetworkSpecification,
     queryFeeModels,
@@ -95,6 +96,7 @@ const teardownEach = async () => {
 }
 
 const teardownAll = async () => {
+  await network.receiptCollector.stopCollectingReceipts()
   await sequelize.drop({})
 }