feat(deployment): auto top up managed deployment with available amount

akash-network · Dec 5, 2024 · 699c6f1 · 699c6f1
1 parent 62aefe3
commit 699c6f1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/...deployment/services/top-up-managed-deployments/top-up-managed-deployments.service.spec.ts b/...deployment/services/top-up-managed-deployments/top-up-managed-deployments.service.spec.ts
@@ -83,7 +83,8 @@ describe(TopUpManagedDeploymentsService.name, () => {
 
   const data = [
     seedFor({}),
-    seedFor({ balance: "5000000", expectedDeploymentsTopUpCount: 1 }),
+    seedFor({ balance: "50000000", expectedDeploymentsTopUpCount: 2 }),
+    seedFor({ balance: "1408022", expectedDeploymentsTopUpCount: 1 }),
     seedFor({ hasDeployments: false }),
     seedFor({ balance: "0", expectedDeploymentsTopUpCount: 0 })
   ];

diff --git a/.../src/deployment/services/top-up-managed-deployments/top-up-managed-deployments.service.ts b/.../src/deployment/services/top-up-managed-deployments/top-up-managed-deployments.service.ts
@@ -52,6 +52,7 @@ export class TopUpManagedDeploymentsService implements DeploymentsRefiller {
   }
 
   private async topUpForWallet(wallet: UserWalletOutput, options: TopUpDeploymentsOptions, summary: TopUpSummarizer) {
+    const depositor = await this.managedMasterWalletService.getFirstAddress();
     summary.inc("walletsCount");
     const owner = wallet.address;
     const denom = this.billingConfig.DEPLOYMENT_GRANT_DENOM;
@@ -63,14 +64,15 @@ export class TopUpManagedDeploymentsService implements DeploymentsRefiller {
     }
 
     const signer = await this.txSignerService.getClientForAddressIndex(wallet.id);
-    const depositor = await this.managedMasterWalletService.getFirstAddress();
 
     let balance = await this.balancesService.retrieveAndCalcDeploymentLimit(wallet);
     let hasTopUp = false;
 
     for (const deployment of drainingDeployments) {
-      const amount = await this.drainingDeploymentService.calculateTopUpAmount(deployment);
-      if (amount > balance) {
+      let amount = await this.drainingDeploymentService.calculateTopUpAmount(deployment);
+      amount = Math.min(amount, balance);
+
+      if (amount <= 0) {
         this.logger.info({ event: "INSUFFICIENT_BALANCE", owner, balance, amount });
         summary.inc("insufficientBalanceCount");
         break;