chore: refactor

centrifuge · Jun 11, 2024 · fdd53ab · fdd53ab
1 parent 5f27952
commit fdd53ab
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 19 deletions.
diff --git a/src/mappings/handlers/blockHandlers.ts b/src/mappings/handlers/blockHandlers.ts
@@ -19,7 +19,6 @@ import {
   TrancheSnapshot,
 } from '../../types/models'
 import { AssetPositionService } from '../services/assetPositionService'
-import { bnToBn } from '@polkadot/util'
 
 const timekeeper = TimekeeperService.init()
 
@@ -66,24 +65,11 @@ async function _handleBlock(block: SubstrateBlock): Promise<void> {
       pool.resetUnrealizedProfit()
       for (const loanId in activeLoanData) {
         const asset = await AssetService.getById(pool.id, loanId)
-        const previousUnrealizedProfitByPeriod = bnToBn(asset.unrealizedProfitByPeriod)
         await asset.updateActiveAssetData(activeLoanData[loanId])
 
-        const unrealizedProfitAtMarketPrice = await AssetPositionService.computeUnrealizedProfitAtPrice(
-          asset.id,
-          asset.currentPrice
-        )
-        const unrealizedProfitAtNotional = await AssetPositionService.computeUnrealizedProfitAtPrice(
-          asset.id,
-          asset.notional
-        )
-        logger.info(`byPeriod for ${asset.id} = ${unrealizedProfitAtMarketPrice} - ${previousUnrealizedProfitByPeriod}`)
-        const unrealizedProfitByPeriod = unrealizedProfitAtMarketPrice - previousUnrealizedProfitByPeriod
-        logger.info(`byPeriod for ${asset.id} = ${unrealizedProfitByPeriod}`)
         await asset.updateUnrealizedProfit(
-          unrealizedProfitAtMarketPrice,
-          unrealizedProfitAtNotional,
-          unrealizedProfitByPeriod
+          await AssetPositionService.computeUnrealizedProfitAtPrice(asset.id, asset.currentPrice),
+          await AssetPositionService.computeUnrealizedProfitAtPrice(asset.id, asset.notional)
         )
         await asset.save()
         await pool.increaseInterestAccrued(asset.interestAccruedByPeriod)

diff --git a/src/mappings/services/assetService.ts b/src/mappings/services/assetService.ts
@@ -225,11 +225,15 @@ export class AssetService extends Asset {
     )
   }
 
-  public updateUnrealizedProfit(atMarketPrice: bigint, atNotional: bigint, byPeriod: bigint) {
-    logger.info(`Updating unrealizedProfit for asset ${this.id}: ${atMarketPrice}, ${atNotional}, ${byPeriod}`)
+  public updateUnrealizedProfit(atMarketPrice: bigint, atNotional: bigint) {
+    logger.info(`Updating unrealizedProfit for asset ${this.id}: ${atMarketPrice}, ${atNotional}`)
     this.unrealizedProfitAtMarketPrice = atMarketPrice
     this.unrealizedProfitAtNotional = atNotional
-    this.unrealizedProfitByPeriod = byPeriod
+
+    logger.info(
+      `byPeriod: ${atMarketPrice} - ${this.unrealizedProfitByPeriod} = ${atMarketPrice - this.unrealizedProfitByPeriod}`
+    )
+    this.unrealizedProfitByPeriod = atMarketPrice - this.unrealizedProfitByPeriod
   }
 }