fix: delete isRebalancingEnable for addOrDiscardWalletIfRequired

XLabs · Dec 14, 2023 · a46536b · a46536b
1 parent 406410f
commit a46536b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/src/chain-wallet-manager.ts b/src/chain-wallet-manager.ts
@@ -274,7 +274,6 @@ export class ChainWalletManager {
 
       for (const balance of balances) {
         this.walletPool.addOrDiscardWalletIfRequired(
-          isRebalancingEnabled,
           balance.address,
           balance,
           minBalanceThreshold,

diff --git a/src/wallets/wallet-pool.ts b/src/wallets/wallet-pool.ts
@@ -8,7 +8,6 @@ export interface WalletPool {
   discardWalletFromPool(wallet: string): void;
   addWalletBackToPoolIfRequired(wallet: string): void;
   addOrDiscardWalletIfRequired(
-    isRebalancingEnabled: boolean,
     address: string,
     balance: WalletBalance,
     minBalanceThreshold: number,
@@ -46,7 +45,6 @@ export class LocalWalletPool implements WalletPool {
     }
   }
 
-  // TODO: AddOrDiscard wallets when calling this method
   private async acquire(resourceId?: string): Promise<string> {
     const resource = resourceId
       ? this.resources[resourceId].resource
@@ -70,15 +68,14 @@ export class LocalWalletPool implements WalletPool {
    * @returns true if there is enough balance on the wallet, false otherwise
    */
   public addOrDiscardWalletIfRequired(
-    isRebalancingEnabled: boolean,
     address: string,
     balance: WalletBalance,
     minBalanceThreshold: number,
   ): boolean {
     const isEnoughWalletBalance =
       Number(balance.formattedBalance) >= minBalanceThreshold;
 
-    if (isRebalancingEnabled && balance.isNative) {
+    if (balance.isNative) {
       if (isEnoughWalletBalance) {
         // set's the discarded flag on the wallet to false
         this.addWalletBackToPoolIfRequired(address);