*: Merge Epoch, Network and Tap subgraph client classes

- For user, this change effectively adds support for local querying on
epoch and tap subgraphs (previously only supported on network subgraph).
- For contributors, this simplifies the codebase to have 1
SubgraphClient class instead of 3 separate implementations:
EpochSubgraph, NetworkSubgraph and TapSubgraph.

Author: fordN

packages/indexer-agent/src/syncing-server.ts

@@ -7,13 +7,13 @@ import { Logger } from '@graphprotocol/common-ts'
 import { parse } from 'graphql'
 import {
   NetworkMapped,
-  NetworkSubgraph,
+  SubgraphClient,
   resolveChainId,
 } from '@graphprotocol/indexer-common'
 
 export interface CreateSyncingServerOptions {
   logger: Logger
-  networkSubgraphs: NetworkMapped<NetworkSubgraph>
+  networkSubgraphs: NetworkMapped<SubgraphClient>
   port: number
 }
 

packages/indexer-common/src/__tests__/subgraph.test.ts

@@ -5,7 +5,7 @@ import {
   ProviderInterface,
   SubgraphQueryInterface,
 } from '../subgraphs'
-import { QueryResult } from '../network-subgraph'
+import { QueryResult } from '../subgraph-client'
 import gql from 'graphql-tag'
 import { mergeSelectionSets } from '../utils'
 

packages/indexer-common/src/allocations/__tests__/tap-pagination.test.ts

@@ -2,7 +2,7 @@ import { Address, Eventual, createLogger, createMetrics } from '@graphprotocol/c
 import {
   Allocation,
   AllocationsResponse,
-  NetworkSubgraph,
+  SubgraphClient,
   QueryFeeModels,
   QueryResult,
   TapCollector,
@@ -11,7 +11,6 @@ import {
   TransactionManager,
 } from '@graphprotocol/indexer-common'
 import { NetworkContracts as TapContracts } from '@semiotic-labs/tap-contracts-bindings'
-import { TAPSubgraph } from '../../tap-subgraph'
 import { NetworkSpecification } from 'indexer-common/src/network-specification'
 import { createMockAllocation } from '../../indexer-management/__tests__/helpers.test'
 import { getContractAddress } from 'ethers/lib/utils'
@@ -155,10 +154,10 @@ const setup = () => {
 
     const tapSubgraph = {
       query: mockQueryTapSubgraph,
-    } as unknown as TAPSubgraph
+    } as unknown as SubgraphClient
     const networkSubgraph = {
       query: mockQueryNetworkSubgraph,
-    } as unknown as NetworkSubgraph
+    } as unknown as SubgraphClient
 
     tapCollector = TapCollector.create({
       logger,

packages/indexer-common/src/allocations/escrow-accounts.ts

@@ -1,5 +1,5 @@
 import { Address, toAddress } from '@graphprotocol/common-ts'
-import { TAPSubgraph } from '../tap-subgraph'
+import { SubgraphClient } from '../subgraph-client'
 import gql from 'graphql-tag'
 
 type U256 = bigint
@@ -44,7 +44,7 @@ export class EscrowAccounts {
 }
 
 export const getEscrowAccounts = async (
-  tapSubgraph: TAPSubgraph,
+  tapSubgraph: SubgraphClient,
   indexer: Address,
 ): Promise<EscrowAccounts> => {
   const result = await tapSubgraph.query<EscrowAccountResponse>(

packages/indexer-common/src/allocations/query-fees.ts

@@ -20,12 +20,12 @@ import {
   TransactionManager,
   specification as spec,
   sequentialTimerMap,
+  SubgraphClient,
 } from '..'
 import { DHeap } from '@thi.ng/heaps'
 import { BigNumber, BigNumberish, Contract } from 'ethers'
 import { Op } from 'sequelize'
 import pReduce from 'p-reduce'
-import { NetworkSubgraph } from '../network-subgraph'
 
 // Receipts are collected with a delay of 20 minutes after
 // the corresponding allocation was closed
@@ -71,7 +71,7 @@ export interface AllocationReceiptCollectorOptions {
   allocations: Eventual<Allocation[]>
   models: QueryFeeModels
   networkSpecification: spec.NetworkSpecification
-  networkSubgraph: NetworkSubgraph
+  networkSubgraph: SubgraphClient
 }
 
 export interface ReceiptCollector {
@@ -94,7 +94,7 @@ export class AllocationReceiptCollector implements ReceiptCollector {
   declare voucherRedemptionBatchThreshold: BigNumber
   declare voucherRedemptionMaxBatchSize: number
   declare protocolNetwork: string
-  declare networkSubgraph: NetworkSubgraph
+  declare networkSubgraph: SubgraphClient
 
   // eslint-disable-next-line @typescript-eslint/no-empty-function -- Private constructor to prevent direct instantiation
   private constructor() {}

packages/indexer-common/src/allocations/tap-collector.ts

@@ -25,8 +25,7 @@ import {
 } from '..'
 import { BigNumber } from 'ethers'
 import pReduce from 'p-reduce'
-import { TAPSubgraph } from '../tap-subgraph'
-import { NetworkSubgraph, QueryResult } from '../network-subgraph'
+import { SubgraphClient, QueryResult } from '../subgraph-client'
 import gql from 'graphql-tag'
 import { getEscrowAccounts } from './escrow-accounts'
 
@@ -52,8 +51,8 @@ interface TapCollectorOptions {
   allocations: Eventual<Allocation[]>
   models: QueryFeeModels
   networkSpecification: spec.NetworkSpecification
-  tapSubgraph: TAPSubgraph
-  networkSubgraph: NetworkSubgraph
+  tapSubgraph: SubgraphClient
+  networkSubgraph: SubgraphClient
 }
 
 interface ValidRavs {
@@ -107,8 +106,8 @@ export class TapCollector {
   declare allocations: Eventual<Allocation[]>
   declare ravRedemptionThreshold: BigNumber
   declare protocolNetwork: string
-  declare tapSubgraph: TAPSubgraph
-  declare networkSubgraph: NetworkSubgraph
+  declare tapSubgraph: SubgraphClient
+  declare networkSubgraph: SubgraphClient
   declare finalityTime: number
   declare indexerAddress: Address
 

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

@@ -1,5 +1,5 @@
 import { BigNumber } from 'ethers'
-import { NetworkSubgraph, SubgraphDeployment } from '@graphprotocol/indexer-common'
+import { SubgraphClient, SubgraphDeployment } from '@graphprotocol/indexer-common'
 
 import { Logger, Address } from '@graphprotocol/common-ts'
 
@@ -31,7 +31,7 @@ export enum AllocationStatus {
 export interface MonitorEligibleAllocationsOptions {
   indexer: Address
   logger: Logger
-  networkSubgraph: NetworkSubgraph
+  networkSubgraph: SubgraphClient
   interval: number
   protocolNetwork: string
 }

packages/indexer-common/src/epoch-subgraph.ts

@@ -1,16 +1,15 @@
 export * from './actions'
 export * from './allocations'
 export * from './async-cache'
-export * from './epoch-subgraph'
 export * from './errors'
 export * from './indexer-management'
 export * from './graph-node'
 export * from './operator'
 export * from './network'
-export * from './network-subgraph'
 export * from './query-fees'
 export * from './rules'
 export * from './subgraphs'
+export * from './subgraph-client'
 export * from './transactions'
 export * from './types'
 export * from './utils'

packages/indexer-common/src/index.ts

@@ -23,12 +23,11 @@ import { literal, Op, Sequelize } from 'sequelize'
 import {
   Allocation,
   AllocationStatus,
-  EpochSubgraph,
   indexerError,
   IndexerErrorCode,
   GraphNode,
   NetworkMonitor,
-  NetworkSubgraph,
+  SubgraphClient,
   resolveChainAlias,
   resolveChainId,
   SubgraphDeployment,
@@ -48,8 +47,8 @@ let models: IndexerManagementModels
 let ethereum: ethers.providers.BaseProvider
 let contracts: NetworkContracts
 let graphNode: GraphNode
-let networkSubgraph: NetworkSubgraph
-let epochSubgraph: EpochSubgraph
+let networkSubgraph: SubgraphClient
+let epochSubgraph: SubgraphClient
 let networkMonitor: NetworkMonitor
 let logger: Logger
 
@@ -84,18 +83,22 @@ const setupMonitor = async () => {
   )
 
   const INDEXER_TEST_API_KEY: string = process.env['INDEXER_TEST_API_KEY'] || ''
-  networkSubgraph = await NetworkSubgraph.create({
+
+  networkSubgraph = await SubgraphClient.create({
+    name: 'NetworkSubgraph',
     logger,
     endpoint: `https://gateway-arbitrum.network.thegraph.com/api/${INDEXER_TEST_API_KEY}/subgraphs/id/3xQHhMudr1oh69ut36G2mbzpYmYxwqCeU6wwqyCDCnqV`,
     deployment: undefined,
     subgraphFreshnessChecker,
   })
 
-  epochSubgraph = new EpochSubgraph(
-    `https://gateway-arbitrum.network.thegraph.com/api/${INDEXER_TEST_API_KEY}/subgraphs/id/BhnsdeZihU4SuokxZMLF4FQBVJ3jgtZf6v51gHvz3bSS`,
-    subgraphFreshnessChecker,
+  epochSubgraph = await SubgraphClient.create({
+    name: 'EpochSubgraph',
     logger,
-  )
+    endpoint: `https://gateway-arbitrum.network.thegraph.com/api/${INDEXER_TEST_API_KEY}/subgraphs/id/BhnsdeZihU4SuokxZMLF4FQBVJ3jgtZf6v51gHvz3bSS`,
+    subgraphFreshnessChecker,
+  })
+
   graphNode = new GraphNode(
     logger,
     'http://test-admin-endpoint.xyz',