feat: add isLegacy boolean to indexer entity

Signed-off-by: Tomás Migone <[email protected]>

Author: tmigone

schema.graphql

@@ -698,6 +698,9 @@ type Indexer @entity {
   "Default display name is the current default name. Used for filtered queries"
   defaultDisplayName: String
 
+  "Whether the indexer is a legacy indexer. This is true for all indexers registered before the Horizon upgrade"
+  isLegacy: Boolean!
+
   # Staking data
   "CURRENT tokens staked in the protocol. Decreases on withdraw, not on lock"
   stakedTokens: BigInt!

src/mappings/helpers/helpers.ts

@@ -129,14 +129,17 @@ export function createOrLoadSubgraphDeployment(
   return deployment as SubgraphDeployment
 }
 
-export function createOrLoadIndexer(indexerAddress: Bytes, timestamp: BigInt): Indexer {
+export function createOrLoadIndexer(indexerAddress: Bytes, timestamp: BigInt ): Indexer {
   let id = indexerAddress.toHexString()
   let indexer = Indexer.load(id)
   if (indexer == null) {
     indexer = new Indexer(id)
     indexer.createdAt = timestamp.toI32()
     indexer.account = id
 
+    // By default we assume indexers are not legacy. Legacy codepaths should flip this to true
+    indexer.isLegacy = false
+
     indexer.stakedTokens = BigInt.fromI32(0)
     indexer.transferredToL2 = false
     indexer.stakedTokensTransferredToL2 = BigInt.fromI32(0)
@@ -197,6 +200,13 @@ export function createOrLoadIndexer(indexerAddress: Bytes, timestamp: BigInt): I
   return indexer as Indexer
 }
 
+export function createOrLoadLegacyIndexer(indexerAddress: Bytes, timestamp: BigInt): Indexer {
+  let indexer = createOrLoadIndexer(indexerAddress, timestamp)
+  indexer.isLegacy = true
+  indexer.save()
+  return indexer
+}
+
 export function createOrLoadProvision(indexerAddress: Bytes, verifierAddress: Bytes, timestamp: BigInt): Provision {
   let id = joinID([indexerAddress.toHexString(), verifierAddress.toHexString()])
   let provision = Provision.load(id)

src/mappings/serviceRegistry.ts

@@ -1,7 +1,7 @@
 import { ServiceRegistered, ServiceUnregistered } from '../types/ServiceRegistry/ServiceRegistry'
 import { Indexer } from '../types/schema'
 
-import { createOrLoadIndexer, createOrLoadGraphAccount } from './helpers/helpers'
+import { createOrLoadLegacyIndexer, createOrLoadGraphAccount } from './helpers/helpers'
 
 /**
  * @dev handleServiceRegistered
@@ -11,7 +11,7 @@ export function handleServiceRegistered(event: ServiceRegistered): void {
   // Creates Graph Account, if needed
   createOrLoadGraphAccount(event.params.indexer, event.block.timestamp)
 
-  let indexer = createOrLoadIndexer(event.params.indexer, event.block.timestamp)
+  let indexer = createOrLoadLegacyIndexer(event.params.indexer, event.block.timestamp)
   indexer.url = event.params.url
   indexer.geoHash = event.params.geohash
   indexer.save()

src/mappings/staking.ts

@@ -33,7 +33,7 @@ import {
 
 import {
   createOrLoadSubgraphDeployment,
-  createOrLoadIndexer,
+  createOrLoadLegacyIndexer,
   createOrLoadPool,
   createOrLoadEpoch,
   joinID,
@@ -53,7 +53,7 @@ import { addresses } from '../../config/addresses'
 
 export function handleDelegationParametersUpdated(event: DelegationParametersUpdated): void {
   let graphNetwork = createOrLoadGraphNetwork(event.block.number, event.address)
-  let indexer = createOrLoadIndexer(event.params.indexer, event.block.timestamp)
+  let indexer = createOrLoadLegacyIndexer(event.params.indexer, event.block.timestamp)
   indexer.indexingRewardCut = event.params.indexingRewardCut.toI32()
   indexer.queryFeeCut = event.params.queryFeeCut.toI32()
   indexer.delegatorParameterCooldown = event.params.cooldownBlocks.toI32()
@@ -73,7 +73,7 @@ export function handleDelegationParametersUpdated(event: DelegationParametersUpd
 export function handleStakeDeposited(event: StakeDeposited): void {
   let graphNetwork = createOrLoadGraphNetwork(event.block.number, event.address)
   // update indexer
-  let indexer = createOrLoadIndexer(event.params.indexer, event.block.timestamp)
+  let indexer = createOrLoadLegacyIndexer(event.params.indexer, event.block.timestamp)
   let previousStake = indexer.stakedTokens
   indexer.stakedTokens = indexer.stakedTokens.plus(event.params.tokens)
   indexer = updateAdvancedIndexerMetrics(indexer as Indexer)
@@ -177,7 +177,7 @@ export function handleStakeDelegated(event: StakeDelegated): void {
   let zeroShares = event.params.shares.equals(BigInt.fromI32(0))
 
   // update indexer
-  let indexer = createOrLoadIndexer(event.params.indexer, event.block.timestamp)
+  let indexer = createOrLoadLegacyIndexer(event.params.indexer, event.block.timestamp)
   indexer.delegatedTokens = indexer.delegatedTokens.plus(event.params.tokens)
   indexer.delegatorShares = indexer.delegatorShares.plus(event.params.shares)
 

src/mappings/subgraphService.ts

@@ -26,6 +26,9 @@ export function handleServiceProviderRegistered(event: ServiceProviderRegistered
         indexer.url = url
         indexer.geoHash = geoHash
         indexer.rewardsDestination = rewardsDestination
+
+        // Change legacy status in case the indexer was created before the Horizon upgrade
+        indexer.isLegacy = false
         indexer.save()
     } else {
         log.warning("ServiceProviderRegistered failed to decode: {}", [event.params.data.toHexString()])