agent: re-fetch allocations for accurate reconciliation

graphprotocol · Mar 10, 2023 · d2decbd · d2decbd
1 parent d2b87c6
commit d2decbd
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/packages/indexer-agent/src/agent.ts b/packages/indexer-agent/src/agent.ts
@@ -755,7 +755,6 @@ class Agent {
   }
   async reconcileDeploymentAllocationAction(
     deploymentAllocationDecision: AllocationDecision,
-    activeAllocations: Allocation[],
     epoch: number,
     maxAllocationEpochs: number,
   ): Promise<void> {
@@ -764,6 +763,11 @@ class Agent {
       epoch,
     })
 
+    // Acuracy check: re-fetch allocations to ensure that we have a fresh state since the start of the reconciliation loop
+    const activeAllocations = await this.networkMonitor.allocations(
+      AllocationStatus.ACTIVE,
+    )
+
     const activeDeploymentAllocations = activeAllocations.filter(
       allocation =>
         allocation.subgraphDeployment.id.bytes32 ===
@@ -866,7 +870,6 @@ class Agent {
     await pMap(networkDeploymentAllocationDecisions, async decision => {
       await this.reconcileDeploymentAllocationAction(
         decision,
-        activeAllocations,
         epoch,
         maxAllocationEpochs,
       )