refactor: add price oracle address

VenusProtocol · Nov 26, 2024 · 9c1bc29 · 9c1bc29
1 parent e42f4ae
commit 9c1bc29
Show file tree

Hide file tree

Showing 4 changed files with 34 additions and 1 deletion.
diff --git a/subgraphs/protocol-reserve/src/mappings/tokenConverter.ts b/subgraphs/protocol-reserve/src/mappings/tokenConverter.ts
@@ -7,6 +7,7 @@ import {
   ConversionResumed,
   ConverterNetworkAddressUpdated,
   DestinationAddressUpdated,
+  PriceOracleUpdated,
 } from '../../generated/BTCBPrimeConverter/TokenConverter';
 import {
   btcbPrimeConverterAddress,
@@ -18,6 +19,7 @@ import {
   wethPrimeConverterAddress,
   xvsVaultConverterAddress,
 } from '../constants/addresses';
+import { getTokenConverter } from '../operations/get';
 import { getOrCreateTokenConverter } from '../operations/getOrCreate';
 import { updateOrCreateTokenConverterConfig } from '../operations/updateOrCreate';
 import { getConverterNetworkId } from '../utilities/ids';
@@ -91,3 +93,9 @@ export function handleBaseAssetUpdated(event: BaseAssetUpdated): void {
   tokenConverter.baseAsset = event.params.newBaseAsset;
   tokenConverter.save();
 }
+
+export function handlePriceOracleUpdated(event: PriceOracleUpdated): void {
+  const tokenConverter = getTokenConverter(event.address)!;
+  tokenConverter.priceOracleAddress = event.params.priceOracle;
+  tokenConverter.save();
+}
diff --git a/subgraphs/protocol-reserve/src/operations/create.ts b/subgraphs/protocol-reserve/src/operations/create.ts
@@ -26,6 +26,9 @@ export function createTokenConverter(tokenConverterAddress: Address): TokenConve
       tokenConverterContract.try_baseAsset(),
     );
   }
+  tokenConverter.priceOracleAddress = valueOrNotAvailableAddressIfReverted(
+    tokenConverterContract.try_priceOracle(),
+  );
   tokenConverter.paused = false;
   tokenConverter.save();
   return tokenConverter;

diff --git a/subgraphs/protocol-reserve/template-eth.yaml b/subgraphs/protocol-reserve/template-eth.yaml
@@ -63,6 +63,8 @@ dataSources:
           handler: handleDestinationAddressUpdated
         - event: BaseAssetUpdated(indexed address,indexed address)
           handler: handleDestinationAddressUpdated
+        - event: PriceOracleUpdated(indexed address,indexed address)
+          handler: handlePriceOracleUpdated
       blockHandlers:
         - handler: handleInitializationWbtcPrimeConverter
           filter:
@@ -101,6 +103,8 @@ dataSources:
           handler: handleDestinationAddressUpdated
         - event: BaseAssetUpdated(indexed address,indexed address)
           handler: handleDestinationAddressUpdated
+        - event: PriceOracleUpdated(indexed address,indexed address)
+          handler: handlePriceOracleUpdated
       blockHandlers:
         - handler: handleInitializationWethPrimeConverter
           filter:
@@ -139,6 +143,8 @@ dataSources:
           handler: handleDestinationAddressUpdated
         - event: BaseAssetUpdated(indexed address,indexed address)
           handler: handleDestinationAddressUpdated
+        - event: PriceOracleUpdated(indexed address,indexed address)
+          handler: handlePriceOracleUpdated
       blockHandlers:
         - handler: handleInitializationUsdcPrimeConverter
           filter:
@@ -177,6 +183,8 @@ dataSources:
           handler: handleDestinationAddressUpdated
         - event: BaseAssetUpdated(indexed address,indexed address)
           handler: handleDestinationAddressUpdated
+        - event: PriceOracleUpdated(indexed address,indexed address)
+          handler: handlePriceOracleUpdated
       blockHandlers:
         - handler: handleInitializationUsdtPrimeConverter
           filter:
@@ -219,6 +227,8 @@ dataSources:
         - handler: handleInitializationXvsVaultConverter
           filter:
             kind: once
+        - event: PriceOracleUpdated(indexed address,indexed address)
+          handler: handlePriceOracleUpdated
 templates:
   - name: ERC20
     kind: ethereum/contract
@@ -242,4 +252,4 @@ templates:
           topic1: ["{{ wbtcPrimeConverterAddress }}", "{{ wethPrimeConverterAddress }}", "{{ usdcPrimeConverterAddress }}", "{{ usdtPrimeConverterAddress }}", "{{ xvsVaultConverterAddress }}"]
         - event: Transfer(indexed address,indexed address,uint256)
           handler: handleTransferIn
-          topic2: ["{{ wbtcPrimeConverterAddress }}", "{{ wethPrimeConverterAddress }}", "{{ usdcPrimeConverterAddress }}", "{{ usdtPrimeConverterAddress }}", "{{ xvsVaultConverterAddress }}"] 
+          topic2: ["{{ wbtcPrimeConverterAddress }}", "{{ wethPrimeConverterAddress }}", "{{ usdcPrimeConverterAddress }}", "{{ usdtPrimeConverterAddress }}", "{{ xvsVaultConverterAddress }}"] 
diff --git a/subgraphs/protocol-reserve/template.yaml b/subgraphs/protocol-reserve/template.yaml
@@ -62,6 +62,8 @@ dataSources:
           handler: handleDestinationAddressUpdated
         - event: BaseAssetUpdated(indexed address,indexed address)
           handler: handleDestinationAddressUpdated
+        - event: PriceOracleUpdated(indexed address,indexed address)
+          handler: handlePriceOracleUpdated
       blockHandlers:
         - handler: handleInitializationBtcbPrimeConverter
           filter:
@@ -100,6 +102,8 @@ dataSources:
           handler: handleDestinationAddressUpdated
         - event: BaseAssetUpdated(indexed address,indexed address)
           handler: handleDestinationAddressUpdated
+        - event: PriceOracleUpdated(indexed address,indexed address)
+          handler: handlePriceOracleUpdated
       blockHandlers:
         - handler: handleInitializationEthPrimeConverter
           filter:
@@ -136,6 +140,8 @@ dataSources:
           handler: handleConverterNetworkAddressUpdated
         - event: DestinationAddressUpdated(indexed address,indexed address)
           handler: handleDestinationAddressUpdated
+        - event: PriceOracleUpdated(indexed address,indexed address)
+          handler: handlePriceOracleUpdated
       blockHandlers:
         - handler: handleInitializationRiskFundConverter
           filter:
@@ -174,6 +180,8 @@ dataSources:
           handler: handleDestinationAddressUpdated
         - event: BaseAssetUpdated(indexed address,indexed address)
           handler: handleDestinationAddressUpdated
+        - event: PriceOracleUpdated(indexed address,indexed address)
+          handler: handlePriceOracleUpdated
       blockHandlers:
         - handler: handleInitializationUsdcPrimeConverter
           filter:
@@ -212,6 +220,8 @@ dataSources:
           handler: handleDestinationAddressUpdated
         - event: BaseAssetUpdated(indexed address,indexed address)
           handler: handleDestinationAddressUpdated
+        - event: PriceOracleUpdated(indexed address,indexed address)
+          handler: handlePriceOracleUpdated
       blockHandlers:
         - handler: handleInitializationUsdtPrimeConverter
           filter:
@@ -250,6 +260,8 @@ dataSources:
           handler: handleDestinationAddressUpdated
         - event: BaseAssetUpdated(indexed address,indexed address)
           handler: handleBaseAssetUpdated
+        - event: PriceOracleUpdated(indexed address,indexed address)
+          handler: handlePriceOracleUpdated
       blockHandlers:
         - handler: handleInitializationXvsVaultConverter
           filter: