chore: preserve older height

QuokkaStake · Apr 27, 2024 · 94f1deb · 94f1deb
1 parent de1f462
commit 94f1deb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/state/generator.go b/pkg/state/generator.go
@@ -54,6 +54,7 @@ func (g *Generator) ProcessChain(
 		defer g.Mutex.Unlock()
 
 		state.SetChainProposalsError(chain, err)
+		state.SetChainProposalsHeight(chain, prevHeight)
 
 		stateChain, found := oldState.ChainInfos[chain.Name]
 		if found {
@@ -132,6 +133,9 @@ func (g *Generator) ProcessProposalAndWallet(
 
 	if err != nil {
 		proposalVote.Error = err
+		if found {
+			proposalVote.Height = oldVote.Height
+		}
 	} else {
 		proposalVote.Vote = vote
 		proposalVote.Height = voteHeight