bugfix: fix logic error in dpos methods

planetarium · Apr 9, 2024 · 014ef32 · 014ef32
1 parent 2f876d0
commit 014ef32
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 15 deletions.
diff --git a/Lib9c/Action/DPoS/Control/Bond.cs b/Lib9c/Action/DPoS/Control/Bond.cs
@@ -81,8 +81,6 @@ internal static (IWorld, FungibleAssetValue) Cancel(
             Address delegationAddress,
             IImmutableSet<Currency> nativeTokens)
         {
-            long blockHeight = ctx.BlockIndex;
-
             // Currency check
             if (!share.Currency.Equals(Asset.Share))
             {

diff --git a/Lib9c/Action/DPoS/Control/DelegateCtrl.cs b/Lib9c/Action/DPoS/Control/DelegateCtrl.cs
@@ -9,6 +9,7 @@
 using Nekoyume.Action.DPoS.Model;
 using Nekoyume.Action.DPoS.Util;
 using Nekoyume.Module;
+using Serilog;
 
 namespace Nekoyume.Action.DPoS.Control
 {
@@ -122,6 +123,12 @@ internal static IWorld Distribute(
                     delegation.LatestDistributeHeight,
                     blockHeight);
 
+                // Skip if there is no reward to distribute.
+                if (delegationRewardSum.RawValue == 0)
+                {
+                    continue;
+                }
+
                 if (!(ValidatorCtrl.TokenPortionByShare(
                     states,
                     delegation.ValidatorAddress,

diff --git a/Lib9c/Action/DPoS/Control/ValidatorCtrl.cs b/Lib9c/Action/DPoS/Control/ValidatorCtrl.cs
@@ -214,12 +214,17 @@ internal static IWorld Unbond(
             validator.UnbondingCompletionBlockHeight = blockHeight + UnbondingSet.Period;
             if (validator.Status == BondingStatus.Bonded)
             {
-                states = states.TransferAsset(
-                    ctx,
-                    ReservedAddress.BondedPool,
-                    ReservedAddress.UnbondedPool,
-                    Asset.GovernanceFromConsensus(
-                        states.GetBalance(validator.Address, Asset.ConsensusToken)));
+                var consensusToken = states.GetBalance(validator.Address, Asset.ConsensusToken);
+                if (consensusToken.RawValue > 0)
+                {
+                    // Transfer consensus token to unbonded pool if remaining.
+                    states = states.TransferAsset(
+                        ctx,
+                        ReservedAddress.BondedPool,
+                        ReservedAddress.UnbondedPool,
+                        Asset.GovernanceFromConsensus(
+                            states.GetBalance(validator.Address, Asset.ConsensusToken)));
+                }
             }
 
             validator.Status = BondingStatus.Unbonding;

diff --git a/Lib9c/Action/DPoS/Sys/UpdateValidators.cs b/Lib9c/Action/DPoS/Sys/UpdateValidators.cs
@@ -1,3 +1,4 @@
+using System.Linq;
 using Bencodex.Types;
 using Libplanet.Action;
 using Libplanet.Action.State;
@@ -36,13 +37,13 @@ public override IWorld Execute(IActionContext context)
             states = ValidatorSetCtrl.Update(states, context);
             ValidatorSet bondedSet;
             (states, bondedSet) = ValidatorSetCtrl.FetchBondedValidatorSet(states);
-            foreach (var validator in bondedSet.Set)
-            {
-                states = states.SetValidator(
-                    new Libplanet.Types.Consensus.Validator(
-                        validator.OperatorPublicKey,
-                        validator.ConsensusToken.RawValue));
-            }
+            var validatorSet = new Libplanet.Types.Consensus.ValidatorSet(
+                bondedSet.Set.Select(
+                        v => new Libplanet.Types.Consensus.Validator(
+                            v.OperatorPublicKey,
+                            v.ConsensusToken.RawValue))
+                    .ToList());
+            states = states.UpdateValidatorSet(validatorSet);
 
             return states;
         }