refactor: save baseAsset as Token entity

VenusProtocol · Dec 6, 2024 · 38c3d8c · 38c3d8c
1 parent d5cc077
commit 38c3d8c
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 7 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(Address.fromBytes(baseAsset)).id;
   } else {
-    tokenConverter.baseAsset = valueOrNotAvailableAddressIfReverted(
+    const baseAsset = (tokenConverter.baseAsset = valueOrNotAvailableAddressIfReverted(
       tokenConverterContract.try_baseAsset(),
-    );
+    ));
+    tokenConverter.baseAsset = getOrCreateToken(Address.fromBytes(baseAsset)).id;
   }
   tokenConverter.priceOracleAddress = valueOrNotAvailableAddressIfReverted(
     tokenConverterContract.try_priceOracle(),

diff --git a/subgraphs/protocol-reserve/tests/ConverterNetwork/index.test.ts b/subgraphs/protocol-reserve/tests/ConverterNetwork/index.test.ts
@@ -3,7 +3,7 @@ import { assert, beforeAll, describe, test } from 'matchstick-as/assembly/index'
 
 import { handleConverterAdded, handleConverterRemoved } from '../../src/mappings/converterNetwork';
 import { getConverterNetworkId, getTokenConverterId } from '../../src/utilities/ids';
-import { createTokenConverterMock } from '../TokenConverter/mocks';
+import { createTokenConverterMock, createTokenMock } from '../TokenConverter/mocks';
 import { createConverterAddedEvent, createConverterRemovedEvent } from './events';
 
 const converterNetworkAddress = Address.fromString('0x0000000000000000000000000000000000000ccc');
@@ -27,6 +27,7 @@ beforeAll(() => {
     token1Address,
     priceOracleAddress,
   );
+  createTokenMock(token1Address, 'BNB');
 });
 
 describe('Converter Network', () => {