fix: upgrade handler balances minimum

sge-network · Mar 19, 2024 · 2246096 · 2246096
1 parent ce2aca5
commit 2246096
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/app/upgrades/v7/upgrades.go b/app/upgrades/v7/upgrades.go
@@ -22,16 +22,19 @@ func CreateUpgradeHandler(
 
 		for _, sa := range allSubAccounts {
 			subAccAddr := sdk.MustAccAddressFromBech32(sa.Address)
-			accSumm, found := k.SubaccountKeeper.GetAccountSummary(ctx, subAccAddr)
+			accSummary, found := k.SubaccountKeeper.GetAccountSummary(ctx, subAccAddr)
 			if !found {
 				panic(fmt.Errorf("account summary for the subaccount not found %s", subAccAddr))
 			}
 
 			_, totalBalances := k.SubaccountKeeper.GetBalances(ctx, subAccAddr, subaccounttypes.BalanceType_BALANCE_TYPE_UNSPECIFIED)
 			bankBalance := k.BankKeeper.GetBalance(ctx, subAccAddr, params.DefaultBondDenom).Amount
 
-			totalBalanceDiff := accSumm.DepositedAmount.Sub(totalBalances)
-			missingBalance := sdkmath.MinInt(bankBalance, totalBalanceDiff)
+			totalBalanceDiff := accSummary.DepositedAmount.
+				Sub(accSummary.SpentAmount).
+				Sub(accSummary.LostAmount).
+				Sub(accSummary.WithdrawnAmount)
+			missingBalance := sdkmath.MinInt(bankBalance, totalBalanceDiff).Sub(totalBalances)
 			if missingBalance.GT(sdkmath.ZeroInt()) {
 				k.SubaccountKeeper.SetLockedBalances(ctx, subAccAddr, []subaccounttypes.LockedBalance{
 					{