fix(dpos): reocrd last dpos rewards by history

elastos · Jan 1, 2025 · f300d17 · f300d17
1 parent 9954c12
commit f300d17
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/core/transaction/dposv2claimrewardtransaction.go b/core/transaction/dposv2claimrewardtransaction.go
@@ -9,6 +9,8 @@ import (
 	"bytes"
 	"errors"
 	"fmt"
+
+	"github.com/elastos/Elastos.ELA/common"
 	common2 "github.com/elastos/Elastos.ELA/core/types/common"
 
 	"github.com/elastos/Elastos.ELA/blockchain"

diff --git a/dpos/state/arbitrators.go b/dpos/state/arbitrators.go
@@ -832,7 +832,13 @@ func (a *Arbiters) accumulateReward(block *types.Block, confirm *payload.Confirm
 				rewards = a.LastDPoSRewards[recordedSponsor]
 			}
 
-			swapReardMapContent(&a.LastDPoSRewards, &possibleRewards)
+			originRewardsMap := copyDPoSRewardMap(a.LastDPoSRewards)
+			a.History.Append(block.Height, func() {
+				a.LastDPoSRewards = possibleRewards
+			}, func() {
+				a.LastDPoSRewards = originRewardsMap
+			})
+
 		} else {
 			if confirm != nil {
 				// get sponsor from cache first
@@ -880,12 +886,6 @@ func (a *Arbiters) accumulateReward(block *types.Block, confirm *payload.Confirm
 	}
 }
 
-func swapReardMapContent(map1, map2 *map[string]map[string]common.Fixed64) {
-	temp := *map1
-	*map1 = *map2
-	*map2 = temp
-}
-
 func (a *Arbiters) clearingDPOSReward(block *types.Block, historyHeight uint32,
 	smoothClearing bool) (err error) {
 	if block.Height < a.ChainParams.PublicDPOSHeight ||