fix mem leak in proposerVotes

onflow · Oct 22, 2021 · 5df7142 · 5df7142
1 parent 36efab8
commit 5df7142
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/consensus/hotstuff/voteaggregator/vote_aggregator.go b/consensus/hotstuff/voteaggregator/vote_aggregator.go
@@ -153,6 +153,15 @@ func (va *VoteAggregator) StoreProposerVote(vote *model.Vote) bool {
 		return false
 	}
 	va.proposerVotes[vote.BlockID] = vote
+	// update viewToBlockIDSet
+	blockIDSet, exists := va.viewToBlockIDSet[vote.View]
+	if exists {
+		blockIDSet[vote.BlockID] = struct{}{}
+	} else {
+		blockIDSet = make(map[flow.Identifier]struct{})
+		blockIDSet[vote.BlockID] = struct{}{}
+		va.viewToBlockIDSet[vote.View] = blockIDSet
+	}
 	return true
 }