divide by zero error

superfluid-finance · Jan 10, 2024 · 77555f8 · 77555f8
1 parent 7cd6bb3
commit 77555f8
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/packages/subgraph/src/mappingHelpers.ts b/packages/subgraph/src/mappingHelpers.ts
@@ -1620,11 +1620,14 @@ export function updateAggregateEntitiesTransferData(
  * @returns the updated pool member entity to be saved
  */
 export function updatePoolMemberTotalAmountUntilUpdatedAtFields(pool: Pool, poolMember: PoolMember): PoolMember {
-    const amountReceivedDelta = pool.totalAmountDistributedUntilUpdatedAt
-        .minus(poolMember.poolTotalAmountDistributedUntilUpdatedAt)
-        .div(pool.totalUnits)
-        .times(poolMember.units);
-    poolMember.totalAmountReceivedUntilUpdatedAt = poolMember.totalAmountReceivedUntilUpdatedAt.plus(amountReceivedDelta);
+    const amountReceivedDelta = pool.totalUnits.equals(BIG_INT_ZERO)
+        ? BIG_INT_ZERO
+        : pool.totalAmountDistributedUntilUpdatedAt
+              .minus(poolMember.poolTotalAmountDistributedUntilUpdatedAt)
+              .div(pool.totalUnits)
+              .times(poolMember.units);
+    poolMember.totalAmountReceivedUntilUpdatedAt =
+        poolMember.totalAmountReceivedUntilUpdatedAt.plus(amountReceivedDelta);
     poolMember.poolTotalAmountDistributedUntilUpdatedAt = pool.totalAmountDistributedUntilUpdatedAt;
 
     return poolMember;