refactor: save baseAsset as Token entity

VenusProtocol · Dec 6, 2024 · c7677f4 · c7677f4
1 parent e9071bd
commit c7677f4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/subgraphs/protocol-reserve/schema.graphql b/subgraphs/protocol-reserve/schema.graphql
@@ -36,7 +36,7 @@ type TokenConverter @entity {
     address: Bytes!
     converterNetwork: ConverterNetwork
     destinationAddress: Bytes!
-    baseAsset: Bytes!
+    baseAsset: Token!
     configs: [TokenConverterConfig!]! @derivedFrom(field: "tokenConverter")
     paused: Boolean!
     priceOracleAddress: Bytes!

diff --git a/subgraphs/protocol-reserve/src/operations/create.ts b/subgraphs/protocol-reserve/src/operations/create.ts
@@ -37,13 +37,14 @@ export function createTokenConverter(tokenConverterAddress: Address): TokenConve
 
   if (tokenConverterAddress.equals(riskFundConverterAddress)) {
     const riskFund = RiskFund.bind(riskFundAddress);
-    tokenConverter.baseAsset = valueOrNotAvailableAddressIfReverted(
-      riskFund.try_convertibleBaseAsset(),
-    );
+    const baseAsset = valueOrNotAvailableAddressIfReverted(riskFund.try_convertibleBaseAsset());
+
+    tokenConverter.baseAsset = getOrCreateToken(baseAsset).id;
   } else {
-    tokenConverter.baseAsset = valueOrNotAvailableAddressIfReverted(
+    const baseAsset = (tokenConverter.baseAsset = valueOrNotAvailableAddressIfReverted(
       tokenConverterContract.try_baseAsset(),
-    );
+    ));
+    tokenConverter.baseAsset = getOrCreateToken(baseAsset).id;
   }
   tokenConverter.priceOracleAddress = valueOrNotAvailableAddressIfReverted(
     tokenConverterContract.try_priceOracle(),