feat: add initialised at timestamp to tranche balances

centrifuge · Nov 11, 2024 · e357a81 · e357a81
1 parent 9ecc2ae
commit e357a81
Show file tree

Hide file tree

Showing 5 changed files with 38 additions and 9 deletions.
diff --git a/schema.graphql b/schema.graphql
@@ -410,6 +410,8 @@ type TrancheBalance @entity {
   pool: Pool! @index
   tranche: Tranche! @index
 
+  initialisedAt: Date!
+
   pendingInvestCurrency: BigInt!
   claimableTrancheTokens: BigInt!
 

diff --git a/src/mappings/handlers/evmHandlers.ts b/src/mappings/handlers/evmHandlers.ts
@@ -119,7 +119,12 @@ async function _handleEvmTransfer(event: TransferLog): Promise<void> {
     const investLpCollect = InvestorTransactionService.collectLpInvestOrder({ ...orderData, address: toAccount.id })
     await investLpCollect.save()
 
-    const trancheBalance = await TrancheBalanceService.getOrInit(toAccount.id, orderData.poolId, orderData.trancheId)
+    const trancheBalance = await TrancheBalanceService.getOrInit(
+      toAccount.id,
+      orderData.poolId,
+      orderData.trancheId,
+      timestamp
+    )
     await trancheBalance.investCollect(orderData.amount)
     await trancheBalance.save()
   }

diff --git a/src/mappings/handlers/investmentsHandlers.ts b/src/mappings/handlers/investmentsHandlers.ts
@@ -67,7 +67,12 @@ async function _handleInvestOrderUpdated(event: SubstrateEvent<OrderUpdatedEvent
   await epoch.saveWithStates()
 
   // Update trancheBalance
-  const trancheBalance = await TrancheBalanceService.getOrInit(orderData.address, orderData.poolId, orderData.trancheId)
+  const trancheBalance = await TrancheBalanceService.getOrInit(
+    orderData.address,
+    orderData.poolId,
+    orderData.trancheId,
+    event.block.timestamp
+  )
   await trancheBalance.investOrder(orderData.amount)
   await trancheBalance.save()
 }
@@ -130,7 +135,12 @@ async function _handleRedeemOrderUpdated(event: SubstrateEvent<OrderUpdatedEvent
   await epoch.saveWithStates()
 
   // Update trancheBalance
-  const trancheBalance = await TrancheBalanceService.getOrInit(orderData.address, orderData.poolId, orderData.trancheId)
+  const trancheBalance = await TrancheBalanceService.getOrInit(
+    orderData.address,
+    orderData.poolId,
+    orderData.trancheId,
+    event.block.timestamp
+  )
   await trancheBalance.redeemOrder(orderData.amount)
   await trancheBalance.save()
 }
@@ -175,7 +185,12 @@ async function _handleInvestOrdersCollected(event: SubstrateEvent<InvestOrdersCo
     amount: payoutInvestmentInvest.toBigInt(),
   }
 
-  const trancheBalance = await TrancheBalanceService.getOrInit(orderData.address, orderData.poolId, orderData.trancheId)
+  const trancheBalance = await TrancheBalanceService.getOrInit(
+    orderData.address,
+    orderData.poolId,
+    orderData.trancheId,
+    event.block.timestamp
+  )
 
   if (orderData.amount > 0) {
     const it = InvestorTransactionService.collectInvestOrder(orderData)
@@ -227,7 +242,12 @@ async function _handleRedeemOrdersCollected(event: SubstrateEvent<RedeemOrdersCo
     amount: payoutInvestmentRedeem.toBigInt(),
   }
 
-  const trancheBalance = await TrancheBalanceService.getOrInit(orderData.address, orderData.poolId, orderData.trancheId)
+  const trancheBalance = await TrancheBalanceService.getOrInit(
+    orderData.address,
+    orderData.poolId,
+    orderData.trancheId,
+    event.block.timestamp
+  )
 
   if (orderData.amount > 0) {
     const it = InvestorTransactionService.collectRedeemOrder(orderData)

diff --git a/src/mappings/handlers/poolsHandlers.ts b/src/mappings/handlers/poolsHandlers.ts
@@ -231,7 +231,8 @@ async function _handleEpochExecuted(event: SubstrateEvent<EpochClosedExecutedEve
       const trancheBalance = await TrancheBalanceService.getOrInit(
         orderData.address,
         orderData.poolId,
-        orderData.trancheId
+        orderData.trancheId,
+        event.block.timestamp
       )
 
       if (oo.investAmount > BigInt(0) && epochState.investFulfillmentPercentage > BigInt(0)) {

diff --git a/src/mappings/services/trancheBalanceService.ts b/src/mappings/services/trancheBalanceService.ts
@@ -1,13 +1,14 @@
 import { TrancheBalance } from '../../types/models/TrancheBalance'
 
 export class TrancheBalanceService extends TrancheBalance {
-  static init(address: string, poolId: string, trancheId: string) {
+  static init(address: string, poolId: string, trancheId: string, timestamp: Date) {
     logger.info(`Initialising new TrancheBalance: ${address}-${poolId}-${trancheId}`)
     const trancheBalance = new this(
       `${address}-${poolId}-${trancheId}`,
       address,
       poolId,
       `${poolId}-${trancheId}`,
+      timestamp,
       BigInt(0),
       BigInt(0),
       BigInt(0),
@@ -24,10 +25,10 @@ export class TrancheBalanceService extends TrancheBalance {
     return trancheBalance as TrancheBalanceService
   }
 
-  static getOrInit = async (address: string, poolId: string, trancheId: string) => {
+  static getOrInit = async (address: string, poolId: string, trancheId: string, timestamp: Date) => {
     let trancheBalance = await this.getById(address, poolId, trancheId)
     if (trancheBalance === undefined) {
-      trancheBalance = this.init(address, poolId, trancheId)
+      trancheBalance = this.init(address, poolId, trancheId, timestamp)
       await trancheBalance.save()
     }
     return trancheBalance