Chunk size

.gitignore

@@ -4,4 +4,5 @@ yarn-error.log
 .DS_Store
 .env
 dist/
-cache/
+cache/
+.idea/

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) {

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 = {

src/wrapper.ts

@@ -23,6 +23,7 @@ import {
     TokenPriceService,
     PoolDataService,
     SorConfig,
+    GraphQLArgs,
 } from './types';
 import { Zero } from '@ethersproject/constants';
 
@@ -69,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(): Promise<boolean> {
-        return this.poolCacher.fetchPools();
+    async fetchPools(
+        queryArgs?: GraphQLArgs,
+        chunkSize?: number
+    ): Promise<boolean> {
+        return this.poolCacher.fetchPools(queryArgs, chunkSize);
     }
 
     /**