fix: compressor batch size config

Gearbox-protocol · Dec 25, 2024 · 2e5c451 · 2e5c451
1 parent 7eca058
commit 2e5c451
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/src/attachSDK.ts b/src/attachSDK.ts
@@ -63,7 +63,7 @@ export default async function attachSDK(): Promise<CreditAccountsService> {
   //   },
   // );
   const service = new CreditAccountsService(sdk, {
-    batchSize: config.batchSize, // TODO: technically, this is different batch size, needs to be changed
+    batchSize: config.compressorBatchSize,
   });
 
   return service;

diff --git a/src/config/env.ts b/src/config/env.ts
@@ -8,6 +8,7 @@ const envConfigMapping: Record<keyof ConfigSchema, string | string[]> = {
   debugAccount: "DEBUG_ACCOUNT",
   debugManager: "DEBUG_MANAGER",
   batchSize: "BATCH_SIZE",
+  compressorBatchSize: "COMPRESSOR_BATCH_SIZE",
   castBin: "CAST_BIN",
   deployPartialLiquidatorContracts: "DEPLOY_PARTIAL_LIQUIDATOR",
   deployBatchLiquidatorContracts: "DEPLOY_BATCH_LIQUIDATOR",

diff --git a/src/config/schema.ts b/src/config/schema.ts
@@ -109,6 +109,10 @@ export const ConfigSchema = z.object({
    * Number of accounts to liquidate at once using batch liquidator
    */
   batchSize: z.coerce.number().nonnegative().default(10),
+  /**
+   * Limit number of accounts to load from compressor. 0 = unlimited, let compressor decide
+   */
+  compressorBatchSize: z.coerce.number().nonnegative().default(0),
   /**
    * Slippage value for pathfined
    */