From f086b11f68acbc28b709fda8c41d90918c4c1b3c Mon Sep 17 00:00:00 2001
From: Robert Leonard <leaonrd.robert.h@gmail.com>
Date: Tue, 11 Jul 2023 08:49:56 -0400
Subject: [PATCH] Handle contract call revert

---
 subgraph/src/datasources/FortaStaking.ts | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/subgraph/src/datasources/FortaStaking.ts b/subgraph/src/datasources/FortaStaking.ts
index aca73a57..5f09e717 100644
--- 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