feat: add totalXvsDistributedWei to market

VenusProtocol · Sep 4, 2023 · 3602e27 · 3602e27
1 parent 88f57e0
commit 3602e27
Show file tree

Hide file tree

Showing 5 changed files with 26 additions and 1 deletion.
diff --git a/subgraphs/venus/schema.graphql b/subgraphs/venus/schema.graphql
@@ -65,6 +65,8 @@ type Market @entity {
     underlyingPriceUSD: BigDecimal!
     "Underlying token decimal length"
     underlyingDecimals: Int!
+    "Total XVS Distributed for this market"
+    totalXvsDistributedMantissa: BigInt!
 }
 
 """

diff --git a/subgraphs/venus/src/constants/index.ts b/subgraphs/venus/src/constants/index.ts
@@ -0,0 +1,3 @@
+import { BigInt } from '@graphprotocol/graph-ts';
+
+export const zeroBigInt32 = BigInt.fromString('0');
diff --git a/subgraphs/venus/src/mappings/comptroller.ts b/subgraphs/venus/src/mappings/comptroller.ts
@@ -1,8 +1,9 @@
 /* eslint-disable prefer-const */
 // to satisfy AS compiler
-import { log } from '@graphprotocol/graph-ts';
+import { BigInt, log } from '@graphprotocol/graph-ts';
 
 import {
+  DistributedSupplierVenus,
   MarketEntered,
   MarketExited,
   MarketListed,
@@ -128,3 +129,16 @@ export const handleNewPriceOracle = (event: NewPriceOracle): void => {
   comptroller.priceOracle = event.params.newPriceOracle;
   comptroller.save();
 };
+
+// Also handles DistributedBorrowerVenus with same signature
+export function handleXvsDistributed(event: DistributedSupplierVenus): void {
+  let vTokenAddress = event.params.vToken.toHex();
+  let venusDelta = event.params.venusDelta.toHex();
+  let market = Market.load(vTokenAddress);
+  if (market == null) {
+    market = createMarket(vTokenAddress);
+  }
+  market.totalXvsDistributedMantissa = market.totalXvsDistributedMantissa.plus(
+    BigInt.fromString(venusDelta),
+  );
+}
diff --git a/subgraphs/venus/src/operations/create.ts b/subgraphs/venus/src/operations/create.ts
@@ -3,6 +3,7 @@ import { Address, BigDecimal, BigInt, log } from '@graphprotocol/graph-ts';
 import { Account, AccountVToken, Market } from '../../generated/schema';
 import { BEP20 } from '../../generated/templates/VToken/BEP20';
 import { VToken } from '../../generated/templates/VToken/VToken';
+import { zeroBigInt32 } from '../constants';
 import { nullAddress, vBnbAddress } from '../constants/addresses';
 import { zeroBD } from '../utilities/exponentToBigDecimal';
 import { getUnderlyingPrice } from '../utilities/getUnderlyingPrice';
@@ -96,6 +97,7 @@ export const createMarket = (marketAddress: string): Market => {
   market.blockTimestamp = 0;
   market.borrowIndex = zeroBD;
   market.reserveFactor = reserveFactor.reverted ? BigInt.fromI32(0) : reserveFactor.value;
+  market.totalXvsDistributedMantissa = zeroBigInt32;
 
   return market;
 };
diff --git a/subgraphs/venus/template.yaml b/subgraphs/venus/template.yaml
@@ -43,6 +43,10 @@ dataSources:
           handler: handleNewLiquidationIncentive
         - event: NewPriceOracle(address,address)
           handler: handleNewPriceOracle
+        - event: DistributedBorrowerVenus(indexed address,indexed address,uint256,uint256)
+          handler: handleXvsDistributed
+        - event: DistributedSupplierVenus(indexed address,indexed address,uint256,uint256)
+          handler: handleXvsDistributed
 templates:
   - name: VToken
     kind: ethereum/contract