logos-co
/
nomos-node
mirror of
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Do not remove entry cache when the threshold is reached (#266) 

Co-authored-by: Gusto <bacvinka@gmail.com>
This commit is contained in:
Daniel Sanchez 2023-07-19 18:04:08 +02:00 committed by GitHub
parent 9f71dbb24c
commit adf935830e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
simulations/src/node/carnot/tally.rs
View File

@ -1,18 +1,18 @@
use consensus_engine::View;
use std::collections::{HashMap, HashSet};
pub(crate) struct Tally<T: core::hash::Hash + Eq> {
pub(crate) struct Tally<T: core::hash::Hash + Eq + Clone> {
cache: HashMap<View, HashSet<T>>,
threshold: usize,
}
impl<T: core::hash::Hash + Eq> Default for Tally<T> {
impl<T: core::hash::Hash + Eq + Clone> Default for Tally<T> {
fn default() -> Self {
Self::new(2)
}
}
impl<T: core::hash::Hash + Eq> Tally<T> {
impl<T: core::hash::Hash + Eq + Clone> Tally<T> {
pub fn new(threshold: usize) -> Self {
Self {
cache: Default::default(),
@ -27,9 +27,9 @@ impl<T: core::hash::Hash + Eq> Tally<T> {
pub fn tally_by(&mut self, view: View, message: T, threshold: usize) -> Option<HashSet<T>> {
let entries = self.cache.entry(view).or_default();
entries.insert(message);
let entries = entries.len();
if entries == threshold {
Some(self.cache.remove(&view).unwrap())
let entries_len = entries.len();
if entries_len == threshold {
Some(entries.clone())
} else {
None
}