agent: Make migration 14 more specific and robust

- Only effect subgraphs assigned to node = 'removed'
graphprotocol · Aug 23, 2024 · ed18006 · ed18006
1 parent 79227d8
commit ed18006
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/packages/indexer-agent/src/db/migrations/14-use-new-deployment-pause-mechanism.ts b/packages/indexer-agent/src/db/migrations/14-use-new-deployment-pause-mechanism.ts
@@ -2,6 +2,7 @@ import { Logger } from '@graphprotocol/common-ts'
 import {
   GraphNode,
   specification,
+  SubgraphDeploymentAssignment,
   SubgraphStatus,
 } from '@graphprotocol/indexer-common'
 import { QueryInterface } from 'sequelize'
@@ -38,9 +39,15 @@ export async function up({ context }: Context): Promise<void> {
       return nodeA.deployments.length - nodeB.deployments.length
     })[0]?.id || 'default'
 
-  const virtuallyPausedDeploymentAssignments =
+  const pausedDeploymentAssignments =
     await graphNode.subgraphDeploymentsAssignments(SubgraphStatus.PAUSED)
 
+  const virtuallyPausedDeploymentAssignments =
+    pausedDeploymentAssignments.filter(
+      (assignment: SubgraphDeploymentAssignment) =>
+        assignment.node === 'removed',
+    )
+
   logger.info(
     'Reassigning paused subgraphs to valid node_id (targetNode), then pausing',
     {