Merge branch 'neitdung/msg-server-bugs' into msg-server-test

realiotech · Jan 19, 2024 · 0043dd9 · 0043dd9
2 parents b019f7b + e49cf92
commit 0043dd9
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/x/multi-staking/keeper/msg_server.go b/x/multi-staking/keeper/msg_server.go
@@ -215,6 +215,14 @@ func (k msgServer) CancelUnbondingDelegation(goCtx context.Context, msg *staking
 	cancelBondAmount := cancelMSCoin.BondValue()
 	cancelUnbondingCoin := sdk.NewCoin(k.keeper.stakingKeeper.BondDenom(ctx), cancelBondAmount)
 
+	lockID := types.MultiStakingLockID(msg.DelegatorAddress, msg.ValidatorAddress)
+	lock := k.keeper.GetOrCreateMultiStakingLock(ctx, lockID)
+	err = lock.AddCoinToMultiStakingLock(cancelMSCoin)
+	if err != nil {
+		return nil, err
+	}
+	k.keeper.SetMultiStakingLock(ctx, lock)
+
 	sdkMsg := &stakingtypes.MsgCancelUnbondingDelegation{
 		DelegatorAddress: msg.DelegatorAddress,
 		ValidatorAddress: msg.ValidatorAddress,