fix: routerGraph cache

curvefi · Nov 21, 2024 · d8335f0 · d8335f0
1 parent 5a90bc9
commit d8335f0
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/router.ts b/src/router.ts
@@ -3,7 +3,7 @@ import memoize from "memoizee";
 import BigNumber from "bignumber.js";
 import {ethers} from "ethers";
 import { curve, OLD_CHAINS } from "./curve.js";
-import {IDict, IRoute, IRouteOutputAndCost, IRouteStep} from "./interfaces";
+import {IDict, IRoute, IRouteOutputAndCost, IRouteStep, IChainId} from "./interfaces";
 import {
     _cutZeros,
     _get_price_impact,
@@ -54,9 +54,8 @@ function mapDict<T, U>(dict: IDict<T>, mapper: (key: string, value: T) => U): ID
     return result;
 }
 
-const _buildRouteGraph = memoize(async (): Promise<IDict<IDict<IRouteStep[]>>> => {
+const _buildRouteGraph = memoize(async (chainId: IChainId): Promise<IDict<IDict<IRouteStep[]>>> => {
     const constants = curve.constants;
-    const chainId = curve.chainId;
     const allPools = Object.entries(curve.getPoolsData()).filter(([id]) => !["crveth", "y", "busd", "pax"].includes(id));
     const amplificationCoefficientDict = await _getAmplificationCoefficientsFromApi();
     const poolTvlDict: IDict<number> = await entriesToDictAsync(allPools, _getTVL);
@@ -69,7 +68,7 @@ const _buildRouteGraph = memoize(async (): Promise<IDict<IDict<IRouteStep[]>>> =
 });
 
 const _findRoutes = async (inputCoinAddress: string, outputCoinAddress: string): Promise<IRoute[]>  => {
-    const routerGraph = await _buildRouteGraph();
+    const routerGraph = await _buildRouteGraph(curve.chainId); // It's important to pass chainId to not use cache from another network
     // extract only the fields we need for the worker
     const poolData = mapDict(
         curve.getPoolsData(),