Merge pull request #402 from balancer/4.1.1-beta.14

4.1.1 beta.14
balancer · Jul 20, 2023 · da0f9f7 · da0f9f7
2 parents 87224f8 + bb7db7a
commit da0f9f7
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 7 deletions.
diff --git a/.gitignore b/.gitignore
@@ -4,4 +4,5 @@ yarn-error.log
 .DS_Store
 .env
 dist/
-cache/
+cache/
+.idea/
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
     "name": "@balancer-labs/sor",
-    "version": "4.1.1-beta.13",
+    "version": "4.1.1-beta.14",
     "license": "GPL-3.0-only",
     "main": "dist/index.js",
     "module": "dist/index.esm.js",

diff --git a/src/poolCacher.ts b/src/poolCacher.ts
@@ -46,9 +46,15 @@ export class PoolCacher {
      * @param {GraphQLArgs} queryArgs - Optional query arguments to pass to pool data service.
      * @returns {boolean} True if pools fetched successfully, False if not.
      */
-    public async fetchPools(queryArgs?: GraphQLArgs): Promise<boolean> {
+    public async fetchPools(
+        queryArgs?: GraphQLArgs,
+        chunkSize?: number
+    ): Promise<boolean> {
         try {
-            this.pools = await this.poolDataService.getPools(queryArgs);
+            this.pools = await this.poolDataService.getPools(
+                queryArgs,
+                chunkSize
+            );
             this._finishedFetching = true;
             return true;
         } catch (err) {

diff --git a/src/types.ts b/src/types.ts
@@ -275,7 +275,10 @@ export interface TokenPriceService {
 }
 
 export interface PoolDataService {
-    getPools(query?: GraphQLArgs): Promise<SubgraphPoolBase[]>;
+    getPools(
+        query?: GraphQLArgs,
+        chunkSize?: number
+    ): Promise<SubgraphPoolBase[]>;
 }
 
 export type FundManagement = {

diff --git a/src/wrapper.ts b/src/wrapper.ts
@@ -70,8 +70,11 @@ export class SOR {
      * fetchPools Retrieves pools information and saves to internal pools cache.
      * @returns {boolean} True if pools fetched successfully, False if not.
      */
-    async fetchPools(queryArgs?: GraphQLArgs): Promise<boolean> {
-        return this.poolCacher.fetchPools(queryArgs);
+    async fetchPools(
+        queryArgs?: GraphQLArgs,
+        chunkSize?: number
+    ): Promise<boolean> {
+        return this.poolCacher.fetchPools(queryArgs, chunkSize);
     }
 
     /**