wip

superfluid-finance · Jan 8, 2024 · 75c8c55 · 75c8c55
1 parent c28853d
commit 75c8c55
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/packages/subgraph/schema.graphql b/packages/subgraph/schema.graphql
@@ -1809,6 +1809,8 @@ type PoolMember @entity {
     units: BigInt!
     isConnected: Boolean!
     totalAmountClaimed: BigInt!
+    poolTotalAmountDistributedUntilUpdatedAt: BigInt!
+    totalAmountReceivedUntilUpdatedAt: BigInt!
 
     account: Account!
     pool: Pool!

diff --git a/packages/subgraph/src/mappings/superfluidPool.ts b/packages/subgraph/src/mappings/superfluidPool.ts
@@ -73,6 +73,13 @@ export function handleMemberUnitsUpdated(event: MemberUnitsUpdated): void {
     pool.totalUnits = pool.totalUnits.plus(unitsDelta);
     pool.save();
 
+    const newReceivedAmount = pool.totalAmountDistributedUntilUpdatedAt
+        .minus(poolMember.totalAmountDistributedUntilUpdatedAt)
+        .div(pool.totalUnits)
+        .mul(poolMember.units);
+    poolMember.totalAmountReceivedUntilUpdatedAt = poolMember.totalAmountReceivedUntilUpdatedAt.plus(newReceivedAmount);
+    poolMember.totalAmountDistributedUntilUpdatedAt = pool.totalAmountDistributedUntilUpdatedAt;
+
     // 0 units to > 0 units
     if (previousUnits.equals(BIG_INT_ZERO) && event.params.newUnits.gt(BIG_INT_ZERO)) {
         pool.totalMembers = pool.totalMembers + 1;