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

Add winning_probability param to replace num_hops in CoverTrafficSettings

This commit is contained in:
Youngjoon Lee 2024-11-08 19:46:43 +07:00
parent 360a9aa558
commit e188b6109c
No known key found for this signature in database
GPG Key ID: 25CA11F37F095E5D
1 changed files with 2 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
nomos-mix/core/src/cover_traffic.rs
View File

@ -12,13 +12,12 @@ use std::task::{Context, Poll};
#[derive(Copy, Clone, Deserialize)]
pub struct CoverTrafficSettings {
pub node_id: [u8; 32],
pub number_of_hops: usize,
pub winning_probability: f64,
pub slots_per_epoch: usize,
pub network_size: usize,
}
pub struct CoverTraffic<EpochStream, SlotStream, Message> {
winning_probability: f64,
settings: CoverTrafficSettings,
epoch_stream: EpochStream,
slot_stream: SlotStream,
@ -36,9 +35,7 @@ where
epoch_stream: EpochStream,
slot_stream: SlotStream,
) -> Self {
let winning_probability = winning_probability(settings.number_of_hops);
CoverTraffic {
winning_probability,
settings,
epoch_stream,
slot_stream,
@ -58,7 +55,6 @@ where
fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
let Self {
winning_probability,
settings,
epoch_stream,
slot_stream,
@ -71,7 +67,7 @@ where
epoch,
settings.network_size,
settings.slots_per_epoch,
*winning_probability,
settings.winning_probability,
);
}
if let Poll::Ready(Some(slot)) = slot_stream.poll_next_unpin(cx) {
@ -107,7 +103,3 @@ fn select_slot<Id: Hash + Eq + AsRef<[u8]> + Copy>(
}
w
}
fn winning_probability(number_of_hops: usize) -> f64 {
1.0 / number_of_hops as f64
}