Merge pull request #3102 from LeoHChen/merge_to_to_master_0525

Merge t3 to master 0525

internal/chain/engine.go

@@ -335,10 +335,14 @@ func payoutUndelegations(
 				"[Finalize] failed to get validator from state to finalize",
 			)
 		}
+		lockPeriod := staking.LockPeriodInEpoch
+		if chain.Config().IsQuickUnlock(header.Epoch()) {
+			lockPeriod = staking.LockPeriodInEpochV2
+		}
 		for i := range wrapper.Delegations {
 			delegation := &wrapper.Delegations[i]
 			totalWithdraw := delegation.RemoveUnlockedUndelegations(
-				header.Epoch(), wrapper.LastEpochInCommittee,
+				header.Epoch(), wrapper.LastEpochInCommittee, lockPeriod,
 			)
 			state.AddBalance(delegation.DelegatorAddress, totalWithdraw)
 		}

internal/params/config.go

@@ -30,6 +30,7 @@ var (
 		CrossLinkEpoch:   big.NewInt(186),
 		StakingEpoch:     big.NewInt(186),
 		PreStakingEpoch:  big.NewInt(185),
+		QuickUnlockEpoch: big.NewInt(191),
 		EIP155Epoch:      big.NewInt(28),
 		S3Epoch:          big.NewInt(28),
 		ReceiptLogEpoch:  big.NewInt(101),
@@ -42,6 +43,7 @@ var (
 		CrossLinkEpoch:   big.NewInt(2),
 		StakingEpoch:     big.NewInt(2),
 		PreStakingEpoch:  big.NewInt(1),
+		QuickUnlockEpoch: big.NewInt(0),
 		EIP155Epoch:      big.NewInt(0),
 		S3Epoch:          big.NewInt(0),
 		ReceiptLogEpoch:  big.NewInt(0),
@@ -55,6 +57,7 @@ var (
 		CrossLinkEpoch:   big.NewInt(2),
 		StakingEpoch:     big.NewInt(2),
 		PreStakingEpoch:  big.NewInt(1),
+		QuickUnlockEpoch: big.NewInt(0),
 		EIP155Epoch:      big.NewInt(0),
 		S3Epoch:          big.NewInt(0),
 		ReceiptLogEpoch:  big.NewInt(0),
@@ -68,6 +71,7 @@ var (
 		CrossLinkEpoch:   big.NewInt(2),
 		StakingEpoch:     big.NewInt(2),
 		PreStakingEpoch:  big.NewInt(1),
+		QuickUnlockEpoch: big.NewInt(0),
 		EIP155Epoch:      big.NewInt(0),
 		S3Epoch:          big.NewInt(0),
 		ReceiptLogEpoch:  big.NewInt(0),
@@ -81,6 +85,7 @@ var (
 		CrossLinkEpoch:   big.NewInt(2),
 		StakingEpoch:     big.NewInt(2),
 		PreStakingEpoch:  big.NewInt(1),
+		QuickUnlockEpoch: big.NewInt(0),
 		EIP155Epoch:      big.NewInt(0),
 		S3Epoch:          big.NewInt(0),
 		ReceiptLogEpoch:  big.NewInt(0),
@@ -93,6 +98,7 @@ var (
 		CrossLinkEpoch:   big.NewInt(2),
 		StakingEpoch:     big.NewInt(2),
 		PreStakingEpoch:  big.NewInt(0),
+		QuickUnlockEpoch: big.NewInt(0),
 		EIP155Epoch:      big.NewInt(0),
 		S3Epoch:          big.NewInt(0),
 		ReceiptLogEpoch:  big.NewInt(0),
@@ -107,6 +113,7 @@ var (
 		big.NewInt(0),             // CrossLinkEpoch
 		big.NewInt(0),             // StakingEpoch
 		big.NewInt(0),             // PreStakingEpoch
+		big.NewInt(0),             // QuickUnlockEpoch
 		big.NewInt(0),             // EIP155Epoch
 		big.NewInt(0),             // S3Epoch
 		big.NewInt(0),             // ReceiptLogEpoch
@@ -121,6 +128,7 @@ var (
 		big.NewInt(0), // CrossLinkEpoch
 		big.NewInt(0), // StakingEpoch
 		big.NewInt(0), // PreStakingEpoch
+		big.NewInt(0), // QuickUnlockEpoch
 		big.NewInt(0), // EIP155Epoch
 		big.NewInt(0), // S3Epoch
 		big.NewInt(0), // ReceiptLogEpoch
@@ -165,6 +173,9 @@ type ChainConfig struct {
 	// PreStakingEpoch is the epoch we allow staking transactions
 	PreStakingEpoch *big.Int `json:"prestaking-epoch,omitempty"`
 
+	// QuickUnlockEpoch is the epoch when undelegation will be unlocked at the current epoch
+	QuickUnlockEpoch *big.Int `json:"quick-unlock-epoch,omitempty"`
+
 	// EIP155 hard fork epoch (include EIP158 too)
 	EIP155Epoch *big.Int `json:"eip155-epoch,omitempty"`
 
@@ -222,6 +233,11 @@ func (c *ChainConfig) IsPreStaking(epoch *big.Int) bool {
 	return isForked(c.PreStakingEpoch, epoch)
 }
 
+// IsQuickUnlock determines whether it's the epoch when the undelegation should be unlocked at end of current epoch
+func (c *ChainConfig) IsQuickUnlock(epoch *big.Int) bool {
+	return isForked(c.QuickUnlockEpoch, epoch)
+}
+
 // IsCrossLink returns whether epoch is either equal to the CrossLink fork epoch or greater.
 func (c *ChainConfig) IsCrossLink(epoch *big.Int) bool {
 	return isForked(c.CrossLinkEpoch, epoch)

node/node_handler.go

@@ -281,10 +281,8 @@ func (node *Node) BroadcastCrossLink() {
 		batchSize := crossLinkBatchSize
 		diff := curBlock.Number().Uint64() - latestBlockNum
 
-		if diff > 100 {
 		// Increase batch size by 1 for every 100 blocks beyond
-			batchSize += int(diff-100) / 100
+		batchSize += int(diff) / 100
-		}
 
 		// Cap at a sane size to avoid overload network
 		if batchSize > crossLinkBatchSize*2 {

staking/types/delegation.go

@@ -19,6 +19,8 @@ var (
 const (
 	// LockPeriodInEpoch is the number of epochs a undelegated token needs to be before it's released to the delegator's balance
 	LockPeriodInEpoch = 7
+	// LockPeriodInEpochV2 there is no extended locking time besides the current epoch time.
+	LockPeriodInEpochV2 = 0
 )
 
 // Delegation represents the bond with tokens held by an account. It is
@@ -176,13 +178,13 @@ func (d *Delegation) DeleteEntry(epoch *big.Int) {
 // RemoveUnlockedUndelegations removes all fully unlocked
 // undelegations and returns the total sum
 func (d *Delegation) RemoveUnlockedUndelegations(
-	curEpoch, lastEpochInCommittee *big.Int,
+	curEpoch, lastEpochInCommittee *big.Int, lockPeriod int,
 ) *big.Int {
 	totalWithdraw := big.NewInt(0)
 	count := 0
 	for j := range d.Undelegations {
-		if big.NewInt(0).Sub(curEpoch, d.Undelegations[j].Epoch).Int64() >= LockPeriodInEpoch ||
+		if big.NewInt(0).Sub(curEpoch, d.Undelegations[j].Epoch).Int64() >= int64(lockPeriod) ||
-			big.NewInt(0).Sub(curEpoch, lastEpochInCommittee).Int64() >= LockPeriodInEpoch {
+			big.NewInt(0).Sub(curEpoch, lastEpochInCommittee).Int64() >= int64(lockPeriod) {
 			// need to wait at least 7 epochs to withdraw; or the validator has been out of committee for 7 epochs
 			totalWithdraw.Add(totalWithdraw, d.Undelegations[j].Amount)
 			count++

staking/types/delegation_test.go

@@ -75,7 +75,7 @@ func TestUnlockedLastEpochInCommittee(t *testing.T) {
 	amount4 := big.NewInt(4000)
 	delegation.Undelegate(epoch4, amount4)
 
-	result := delegation.RemoveUnlockedUndelegations(curEpoch, lastEpochInCommittee)
+	result := delegation.RemoveUnlockedUndelegations(curEpoch, lastEpochInCommittee, 7)
 	if result.Cmp(big.NewInt(8000)) != 0 {
 		t.Errorf("removing an unlocked undelegation fails")
 	}
@@ -90,7 +90,7 @@ func TestUnlockedLastEpochInCommitteeFail(t *testing.T) {
 	amount4 := big.NewInt(4000)
 	delegation.Undelegate(epoch4, amount4)
 
-	result := delegation.RemoveUnlockedUndelegations(curEpoch, lastEpochInCommittee)
+	result := delegation.RemoveUnlockedUndelegations(curEpoch, lastEpochInCommittee, 7)
 	if result.Cmp(big.NewInt(0)) != 0 {
 		t.Errorf("premature delegation shouldn't be unlocked")
 	}
@@ -104,7 +104,21 @@ func TestUnlockedFullPeriod(t *testing.T) {
 	amount5 := big.NewInt(4000)
 	delegation.Undelegate(epoch5, amount5)
 
-	result := delegation.RemoveUnlockedUndelegations(curEpoch, lastEpochInCommittee)
+	result := delegation.RemoveUnlockedUndelegations(curEpoch, lastEpochInCommittee, 7)
+	if result.Cmp(big.NewInt(4000)) != 0 {
+		t.Errorf("removing an unlocked undelegation fails")
+	}
+}
+
+func TestQuickUnlock(t *testing.T) {
+	lastEpochInCommittee := big.NewInt(44)
+	curEpoch := big.NewInt(44)
+
+	epoch7 := big.NewInt(44)
+	amount7 := big.NewInt(4000)
+	delegation.Undelegate(epoch7, amount7)
+
+	result := delegation.RemoveUnlockedUndelegations(curEpoch, lastEpochInCommittee, 0)
 	if result.Cmp(big.NewInt(4000)) != 0 {
 		t.Errorf("removing an unlocked undelegation fails")
 	}
@@ -119,7 +133,7 @@ func TestUnlockedFullPeriodFail(t *testing.T) {
 	amount5 := big.NewInt(4000)
 	delegation.Undelegate(epoch5, amount5)
 
-	result := delegation.RemoveUnlockedUndelegations(curEpoch, lastEpochInCommittee)
+	result := delegation.RemoveUnlockedUndelegations(curEpoch, lastEpochInCommittee, 7)
 	if result.Cmp(big.NewInt(0)) != 0 {
 		t.Errorf("premature delegation shouldn't be unlocked")
 	}
@@ -133,7 +147,7 @@ func TestUnlockedPremature(t *testing.T) {
 	amount6 := big.NewInt(4000)
 	delegation.Undelegate(epoch6, amount6)
 
-	result := delegation.RemoveUnlockedUndelegations(curEpoch, lastEpochInCommittee)
+	result := delegation.RemoveUnlockedUndelegations(curEpoch, lastEpochInCommittee, 7)
 	if result.Cmp(big.NewInt(0)) != 0 {
 		t.Errorf("premature delegation shouldn't be unlocked")
 	}