change minor logic for SetAccountBalanceForDenom in interchainstaking

notional-labs · Oct 5, 2023 · c29277f · c29277f
1 parent a02415f
commit c29277f
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/x/interchainstaking/keeper/zones.go b/x/interchainstaking/keeper/zones.go
@@ -265,7 +265,16 @@ func (k *Keeper) SetAccountBalanceForDenom(ctx sdk.Context, zone *types.Zone, ad
 		}
 		k.Logger(ctx).Info("Matched withdrawal address", "address", address, "wg", zone.WithdrawalAddress.BalanceWaitgroup, "balance", zone.WithdrawalAddress.Balance)
 	case zone.PerformanceAddress != nil && address == zone.PerformanceAddress.Address:
-		k.Logger(ctx).Info("Matched performance address")
+		existing := zone.PerformanceAddress.Balance.AmountOf(coin.Denom)
+		err = zone.PerformanceAddress.SetBalance(zone.PerformanceAddress.Balance.Sub(sdk.NewCoins(sdk.NewCoin(coin.Denom, existing))...).Add(coin)) // reset this denom
+		if err != nil {
+			return err
+		}
+		err = zone.PerformanceAddress.DecrementBalanceWaitgroup()
+		if err != nil {
+			return err
+		}
+		k.Logger(ctx).Info("Matched performance address", "address", address, "wg", zone.PerformanceAddress.BalanceWaitgroup, "balance", zone.PerformanceAddress.Balance)
 	default:
 		panic("unexpected")
 	}