fix(agent): defend against graph-node bug #5550 (null paused)

graphprotocol · Aug 8, 2024 · b2188e6 · b2188e6
1 parent 3efefd9
commit b2188e6
Showing 1 changed file with 32 additions and 3 deletions.
diff --git a/packages/indexer-common/src/graph-node.ts b/packages/indexer-common/src/graph-node.ts
@@ -153,18 +153,47 @@ export class GraphNode {
   ): Promise<SubgraphDeploymentAssignment[]> {
     try {
       this.logger.debug('Fetch subgraph deployment assignments')
-      const result = await this.status
+
+      // FIXME: remove this initial check for just node when graph-node releases
+      // https://github.com/graphprotocol/graph-node/pull/5551
+      const nodeOnlyResult = await this.status
         .query(gql`
           {
             indexingStatuses {
               subgraphDeployment: subgraph
               node
-              paused
             }
           }
         `)
         .toPromise()
 
+      if (nodeOnlyResult.error) {
+        throw nodeOnlyResult.error
+      }
+
+      const withAssignments: string[] = nodeOnlyResult.data.indexingStatuses
+        .filter((result: QueryResult) => {
+          return result.node !== undefined
+        })
+        .map((result: QueryResult) => {
+          return result.subgraphDeployment
+        })
+
+      const result = await this.status
+        .query(
+          gql`
+          {
+            indexingStatuses($subgraphs) {
+              subgraphDeployment: subgraph
+              node
+              paused
+            }
+          }
+        `,
+          { subgraphs: withAssignments },
+        )
+        .toPromise()
+
       if (result.error) {
         throw result.error
       }
@@ -178,7 +207,7 @@ export class GraphNode {
 
       type QueryResult = {
         subgraphDeployment: string
-        node: string
+        node: string | undefined
         paused: boolean | undefined
       }