Fix genesis vault apy before pool migration

stakewise · Jul 23, 2024 · b2b0a4e · b2b0a4e
1 parent a6ca862
commit b2b0a4e
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 8 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "v3-subgraph",
-  "version": "2.0.2",
+  "version": "2.0.3",
   "description": "Subgraph for the StakeWise Protocol",
   "repository": "https://github.com/stakewise/v3-subgraph",
   "license": "AGPL-3.0-only",

diff --git a/src/helpers/utils.ts b/src/helpers/utils.ts
@@ -116,9 +116,10 @@ export function getPoolStateUpdate(
     penaltyAssets = ethereum.decode('uint256', resultValue[2].returnData)!.toBigInt()
     principalAssets = ethereum.decode('uint256', resultValue[3].returnData)!.toBigInt()
   }
-  const newRate = BigInt.fromString(WAD)
-    .times(rewardAssets.plus(principalAssets).minus(penaltyAssets))
-    .div(principalAssets)
+  let newRate = BigInt.fromString(WAD)
+  if (principalAssets.gt(BigInt.fromI32(0))) {
+    newRate = newRate.times(rewardAssets.plus(principalAssets).minus(penaltyAssets)).div(principalAssets)
+  }
 
   return [newRate, rewardAssets, principalAssets, penaltyAssets]
 }

diff --git a/src/mappings/depositDataRegistry.ts b/src/mappings/depositDataRegistry.ts
@@ -61,7 +61,11 @@ export function handleDepositDataRootUpdated(event: DepositDataRootUpdated): voi
   const depositDataRoot = event.params.depositDataRoot
 
   // Vault must exist at the time of the event
-  const vault = Vault.load(vaultAddress) as Vault
+  const vault = Vault.load(vaultAddress)
+  if (vault === null) {
+    log.error('[DepositDataRegistry] DepositDataRootUpdated vault={} not found', [vaultAddress])
+    return
+  }
   vault.depositDataRoot = depositDataRoot
   // Update deprecated validators root
   vault.validatorsRoot = depositDataRoot

diff --git a/src/mappings/keeper.ts b/src/mappings/keeper.ts
@@ -240,7 +240,9 @@ export function updateRewards(
       slashedMevReward = slashedMevReward.plus(vault.lockedExecutionReward.minus(lockedMevReward))
     }
 
-    updateVaultApy(vault, vault.rewardsTimestamp, updateTimestamp, newRate.minus(vault.rate))
+    if (!vault.isGenesis || v2Pool.migrated) {
+      updateVaultApy(vault, vault.rewardsTimestamp, updateTimestamp, newRate.minus(vault.rate))
+    }
 
     vaultsStat.totalAssets = vaultsStat.totalAssets.minus(vault.totalAssets).plus(newTotalAssets)
     vault.totalAssets = newTotalAssets