Handle contract call revert

forta-network · Jul 11, 2023 · f086b11 · f086b11
1 parent dbeee25
commit f086b11
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/subgraph/src/datasources/FortaStaking.ts b/subgraph/src/datasources/FortaStaking.ts
@@ -159,14 +159,20 @@ function updateStake(
     sharesId.subject = _subjectId;
     sharesId.save();
   }
-  subject.activeStake = fortaStaking.activeStakeFor(
+
+  const activeStake = fortaStaking.try_activeStakeFor(
     _subjectType,
     _subject
   );
-  subject.inactiveStake = fortaStaking.inactiveStakeFor(
+
+  const inActiveStake = fortaStaking.try_inactiveStakeFor(
     _subjectType,
     _subject
   );
+
+  subject.activeStake = activeStake.reverted ? BigInt.fromI32(0) : activeStake.value;
+  subject.inactiveStake = inActiveStake.reverted ? BigInt.fromI32(0) : inActiveStake.value;
+
   subject.activeShares = fortaStaking.totalShares(
     _subjectType,
     _subject