fix: track total delegated tokens changes when rewards or fees are co…

…llected
graphprotocol · Jan 29, 2024 · 5f8f785 · 5f8f785
1 parent 6f18ba6
commit 5f8f785
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/mappings/rewardsManager.ts b/src/mappings/rewardsManager.ts
@@ -85,6 +85,7 @@ export function handleRewardsAssigned(event: RewardsAssigned): void {
   graphNetwork.totalIndexingDelegatorRewards = graphNetwork.totalIndexingDelegatorRewards.plus(
     delegatorIndexingRewards,
   )
+  graphNetwork.totalDelegatedTokens = graphNetwork.totalDelegatedTokens.plus(delegatorIndexingRewards)
   graphNetwork.save()
 }
 

diff --git a/src/mappings/staking.ts b/src/mappings/staking.ts
@@ -591,6 +591,7 @@ export function handleRebateClaimed(event: RebateClaimed): void {
   graphNetwork.totalUnclaimedQueryFeeRebates = graphNetwork.totalUnclaimedQueryFeeRebates.minus(
     event.params.delegationFees.plus(event.params.tokens),
   )
+  graphNetwork.totalDelegatedTokens = graphNetwork.totalDelegatedTokens.plus(event.params.delegationFees)
   graphNetwork.save()
 }