Add tripaths to routeProposer.

balancer · Aug 9, 2023 · 76363a9 · 76363a9
1 parent 6244745
commit 76363a9
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/routeProposal/index.ts b/src/routeProposal/index.ts
@@ -14,6 +14,7 @@ import {
     SorConfig,
     PoolDictionary,
 } from '../types';
+import { getTriPaths } from './triPaths';
 
 export class RouteProposer {
     cache: Record<string, { paths: NewPath[] }> = {};
@@ -79,9 +80,17 @@ export class RouteProposer {
             this.config
         );
 
+        const triPaths = getTriPaths(
+            tokenIn,
+            tokenOut,
+            poolsAllDict,
+            this.config.triPathMidPoolIds ?? []
+        );
+
         const combinedPathData = pathData
             .concat(...boostedPaths)
-            .concat(...pathsUsingStaBal);
+            .concat(...pathsUsingStaBal)
+            .concat(...triPaths);
         const [paths] = calculatePathLimits(combinedPathData, swapType);
 
         this.cache[`${tokenIn}${tokenOut}${swapType}${swapOptions.timestamp}`] =