all: ensure we don't pause an allocation if auto-grafting - bump patch to 0.24.1

Author: dwerner

CLAUDE.md

@@ -131,7 +131,7 @@ Tests require:
    ```
 
 ## Current State
-- Version: v0.24.0
+- Version: v0.24.1
 - Active branch: release-v0.24
 - Main branch: main
 - Uncommitted changes in docs/ and config/

docs/auto-graft.md

@@ -170,6 +170,10 @@ After each dependency reaches its target block, it's automatically paused to:
 - Save computational resources
 - Ensure data consistency
 
+Auto-graft will check for active allocations before pausing a dependency. If a subgraph has an active allocation (meaning it's still serving queries), it will NOT be automatically paused to prevent service disruption. This ensures that customers can continue querying subgraphs that are actively allocated.
+
+*Note: This allocation check was introduced in v0.24.1*
+
 ## Monitoring Auto-Graft
 
 ### Key Log Messages

docs/networks/arbitrum-sepolia.md

@@ -7,8 +7,8 @@ The Graph Network's testnet is on Arbitrum Sepolia (eip155:421614). Sepolia netw
 | Component          | Release                                                                              |
 | ------------------ | ------------------------------------------------------------------------------------ |
 | contracts          | [5.3.3](https://github.com/graphprotocol/contracts/releases/tag/v5.3.3)              |
-| indexer-agent      | [0.24.0](https://github.com/graphprotocol/indexer/releases/tag/v0.24.0)              |
-| indexer-cli        | [0.24.0](https://github.com/graphprotocol/indexer/releases/tag/v0.24.0)              |
+| indexer-agent      | [0.24.1](https://github.com/graphprotocol/indexer/releases/tag/v0.24.1)              |
+| indexer-cli        | [0.24.1](https://github.com/graphprotocol/indexer/releases/tag/v0.24.1)              |
 | indexer-service-rs | [1.0.0](https://github.com/graphprotocol/indexer-rs/releases/tag/v1.0.0)             |
 | tap-agent          | [1.0.0](https://github.com/graphprotocol/indexer-rs/releases/tag/v1.0.0)             |
 | graph-node         | [0.35.1](https://github.com/graphprotocol/graph-node/releases/tag/v0.35.1)           |

lerna.json

@@ -4,5 +4,5 @@
   ],
   "npmClient": "yarn",
   "useWorkspaces": true,
-  "version": "0.24.0"
+  "version": "0.24.1"
 }

packages/indexer-agent/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@graphprotocol/indexer-agent",
-  "version": "0.24.0",
+  "version": "0.24.1",
   "description": "Indexer agent",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -30,7 +30,7 @@
   },
   "dependencies": {
     "@graphprotocol/common-ts": "2.0.11",
-    "@graphprotocol/indexer-common": "^0.24.0",
+    "@graphprotocol/indexer-common": "^0.24.1",
     "@thi.ng/heaps": "^1.3.1",
     "axios": "0.26.1",
     "bs58": "5.0.0",

packages/indexer-agent/src/agent.ts

@@ -6,6 +6,7 @@ import {
   Metrics,
   SubgraphDeploymentID,
   timer,
+  NetworkContracts,
 } from '@graphprotocol/common-ts'
 import {
   ActivationCriteria,
@@ -1256,13 +1257,19 @@ export class Agent {
   }
 
   // TODO: After indexer-service deprecation: Move to be an initialization check inside Network.create()
-  async ensureSubgraphIndexing(deployment: string, networkIdentifier: string) {
+  async ensureSubgraphIndexing(
+    deployment: string,
+    networkIdentifier: string,
+    contracts?: NetworkContracts,
+  ) {
     try {
       // TODO: Check both the local deployment and the external subgraph endpoint
       // Make sure the subgraph is being indexed
       await this.graphNode.ensure(
         `indexer-agent/${deployment.slice(-10)}`,
         new SubgraphDeploymentID(deployment),
+        undefined, // currentAssignments
+        contracts,
       )
 
       // Validate if the Network Subgraph belongs to the current provider's network.
@@ -1288,6 +1295,7 @@ export class Agent {
       await this.ensureSubgraphIndexing(
         network.specification.subgraphs.networkSubgraph.deployment,
         network.specification.networkIdentifier,
+        network.contracts,
       )
     }
     // Epoch subgraph
@@ -1297,13 +1305,15 @@ export class Agent {
       await this.ensureSubgraphIndexing(
         network.specification.subgraphs.epochSubgraph.deployment,
         network.specification.networkIdentifier,
+        network.contracts,
       )
     }
     // TAP subgraph
     if (network.specification.subgraphs.tapSubgraph?.deployment !== undefined) {
       await this.ensureSubgraphIndexing(
         network.specification.subgraphs.tapSubgraph.deployment,
         network.specification.networkIdentifier,
+        network.contracts,
       )
     }
   }

packages/indexer-cli/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@graphprotocol/indexer-cli",
-  "version": "0.24.0",
+  "version": "0.24.1",
   "description": "Indexer CLI for The Graph Network",
   "main": "./dist/cli.js",
   "files": [
@@ -27,7 +27,7 @@
   },
   "dependencies": {
     "@graphprotocol/common-ts": "2.0.11",
-    "@graphprotocol/indexer-common": "^0.24.0",
+    "@graphprotocol/indexer-common": "^0.24.1",
     "@iarna/toml": "2.2.5",
     "@thi.ng/iterators": "5.1.74",
     "@urql/core": "3.1.0",

packages/indexer-common/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@graphprotocol/indexer-common",
-  "version": "0.24.0",
+  "version": "0.24.1",
   "description": "Common library for Graph Protocol indexer components",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",

packages/indexer-common/src/graph-node.ts

@@ -1,6 +1,6 @@
 import gql from 'graphql-tag'
 import jayson, { Client as RpcClient } from 'jayson/promise'
-import { Logger, SubgraphDeploymentID } from '@graphprotocol/common-ts'
+import { Logger, SubgraphDeploymentID, NetworkContracts } from '@graphprotocol/common-ts'
 import { Client, createClient } from '@urql/core'
 import {
   INDEXER_ERROR_MESSAGES,
@@ -665,6 +665,7 @@ export class GraphNode {
     name: string,
     deployment: SubgraphDeploymentID,
     currentAssignments?: SubgraphDeploymentAssignment[],
+    contracts?: NetworkContracts,
   ): Promise<void> {
     this.logger.debug('Ensure subgraph deployment is syncing', {
       name,
@@ -693,7 +694,12 @@ export class GraphNode {
         await this.resume(deployment)
       } else {
         // Subgraph deployment not found
-        await this.autoGraftDeployDependencies(deployment, deploymentAssignments, name)
+        await this.autoGraftDeployDependencies(
+          deployment,
+          deploymentAssignments,
+          name,
+          contracts,
+        )
 
         // Create and deploy the subgraph
         this.logger.debug(
@@ -732,6 +738,7 @@ export class GraphNode {
     deployment: SubgraphDeploymentID,
     deploymentAssignments: SubgraphDeploymentAssignment[],
     name: string,
+    contracts?: NetworkContracts,
   ) {
     this.logger.debug('Auto graft deploy subgraph dependencies')
     const { network: subgraphChainName } = await this.subgraphFeatures(deployment)
@@ -805,6 +812,7 @@ export class GraphNode {
           dependency.block,
           dependency.base,
           subgraphChainName,
+          contracts,
         )
       }
     }
@@ -820,6 +828,7 @@ export class GraphNode {
     blockHeight: number,
     subgraphDeployment: SubgraphDeploymentID,
     chainName: string | null,
+    contracts?: NetworkContracts,
   ): Promise<void> {
     async function waitForMs(ms: number) {
       return new Promise((resolve) => setTimeout(resolve, ms))
@@ -912,12 +921,71 @@ export class GraphNode {
       // Is the graftBaseBlock within the range of the earliest and head of the chain?
       if (chain.latestBlock && chain.latestBlock.number >= blockHeight) {
         if (!deployed[0].paused) {
-          this.logger.debug(`Subgraph synced to block! Pausing as requirement is met.`, {
-            subgraph: subgraphDeployment.ipfsHash,
-            indexingStatus,
-          })
-          // pause the subgraph to prevent further indexing
-          await this.pause(subgraphDeployment)
+          // Check if there's an active allocation before pausing
+          let hasActiveAllocation = false
+          let checkFailed = false
+
+          // If contracts are provided, check for active allocations
+          if (contracts) {
+            try {
+              // Check if any allocations exist for this subgraph deployment
+              const filter = contracts.staking.filters.AllocationCreated(
+                null, // indexer (any)
+                subgraphDeployment.bytes32, // subgraphDeploymentID
+              )
+              const events = await contracts.staking.queryFilter(filter)
+
+              // Check each allocation to see if it's still active
+              for (const event of events) {
+                const allocationId = event.args?.allocationID
+                if (allocationId) {
+                  const state = await contracts.staking.getAllocationState(allocationId)
+                  // State 1 means Active allocation
+                  if (state === 1) {
+                    hasActiveAllocation = true
+                    break
+                  }
+                }
+              }
+
+              if (hasActiveAllocation) {
+                this.logger.warn(`Subgraph has active allocation, not going to pause`, {
+                  subgraph: subgraphDeployment.ipfsHash,
+                  blockHeight,
+                  indexingStatus,
+                })
+              }
+            } catch (error) {
+              checkFailed = true
+              this.logger.error(`Failed to check allocation state, not going to pause`, {
+                subgraph: subgraphDeployment.ipfsHash,
+                error,
+              })
+            }
+          } else {
+            // No contracts provided, skip pause for safety
+            this.logger.warn(
+              `No contracts provided to check allocation state, not going to pause`,
+              {
+                subgraph: subgraphDeployment.ipfsHash,
+                blockHeight,
+              },
+            )
+            checkFailed = true
+          }
+
+          // Only pause if we successfully checked and found no active allocations
+          if (!hasActiveAllocation && !checkFailed) {
+            this.logger.debug(
+              `Subgraph synced to block! Pausing as requirement is met.`,
+              {
+                subgraph: subgraphDeployment.ipfsHash,
+                indexingStatus,
+              },
+            )
+            // pause the subgraph to prevent further indexing
+            await this.pause(subgraphDeployment)
+          }
         } else {
           this.logger.debug(`Subgraph already paused and synced to block.`, {
             subgraph: subgraphDeployment.ipfsHash,

packages/indexer-common/src/indexer-management/allocations.ts

@@ -433,6 +433,7 @@ export class AllocationManager {
           `indexer-agent/${action.deploymentID!.slice(-10)}`,
           new SubgraphDeploymentID(action.deploymentID!),
           currentAssignments,
+          this.network.contracts,
         ),
       {
         stopOnError: false,