feat(bpos): remove from NodeOwnerKeys when producer canceled

elastos · Oct 26, 2023 · 2463368 · 2463368
1 parent 159d8ab
commit 2463368
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/dpos/state/state.go b/dpos/state/state.go
@@ -1726,8 +1726,11 @@ func (s *State) processTransactions(txs []interfaces.Transaction, height uint32)
 	cancelDposV2AndDposV1V2Producer := func(key string, producer *Producer) {
 		oriState := producer.state
 		oriDepositAmount := producer.depositAmount
+		nodePublicKey := common.BytesToHexString(producer.info.NodePublicKey)
+		oriOwnerKey := s.NodeOwnerKeys[nodePublicKey]
 		s.History.Append(height, func() {
 			producer.state = Canceled
+			delete(s.NodeOwnerKeys, nodePublicKey)
 			producer.depositAmount -= state.MinDPoSV2DepositAmount
 			s.CanceledProducers[key] = producer
 			switch oriState {
@@ -1750,6 +1753,7 @@ func (s *State) processTransactions(txs []interfaces.Transaction, height uint32)
 			case Illegal:
 				s.IllegalProducers[key] = producer
 			}
+			s.NodeOwnerKeys[common.BytesToHexString(producer.info.NodePublicKey)] = oriOwnerKey
 			delete(s.CanceledProducers, key)
 			s.Nicknames[producer.info.NickName] = struct{}{}
 		})