logos-co/nomos-node
1
0
Fork 0
mirror of synced 2025-01-11 16:24:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
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 consensus_engine::View;
use std::collections::{HashMap, HashSet}; 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>>, cache: HashMap<View, HashSet<T>>,
threshold: usize, 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 { fn default() -> Self {
Self::new(2) 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 { pub fn new(threshold: usize) -> Self {
Self { Self {
cache: Default::default(), 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>> { pub fn tally_by(&mut self, view: View, message: T, threshold: usize) -> Option<HashSet<T>> {
let entries = self.cache.entry(view).or_default(); let entries = self.cache.entry(view).or_default();
entries.insert(message); entries.insert(message);
let entries = entries.len(); let entries_len = entries.len();
if entries == threshold { if entries_len == threshold {
Some(self.cache.remove(&view).unwrap()) Some(entries.clone())
} else { } else {
None None
} }