From 6432e708d6303e76386b07cfaa73458ee8d760a4 Mon Sep 17 00:00:00 2001
From: Jedrzej Stuczynski <jedrzej.stuczynski@gmail.com>
Date: Fri, 31 Jan 2020 16:10:00 +0000
Subject: [PATCH] Extra argument to cover traffic stream

---
 nym-client/src/client/cover_traffic_stream.rs | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/nym-client/src/client/cover_traffic_stream.rs b/nym-client/src/client/cover_traffic_stream.rs
index c48c0c4801..c2aa3853da 100644
--- a/nym-client/src/client/cover_traffic_stream.rs
+++ b/nym-client/src/client/cover_traffic_stream.rs
@@ -1,22 +1,21 @@
 use crate::client::mix_traffic::MixMessage;
 use crate::client::topology_control::TopologyInnerRef;
-use crate::client::LOOP_COVER_AVERAGE_DELAY;
 use futures::channel::mpsc;
 use log::{error, info, trace, warn};
 use sphinx::route::Destination;
-use std::time::Duration;
+use std::time;
 use topology::NymTopology;
 
 pub(crate) async fn start_loop_cover_traffic_stream<T: NymTopology>(
     tx: mpsc::UnboundedSender<MixMessage>,
     our_info: Destination,
     topology_ctrl_ref: TopologyInnerRef<T>,
+    average_delay_duration: time::Duration,
 ) {
     info!("Starting loop cover traffic stream");
     loop {
         trace!("next cover message!");
-        let delay = mix_client::poisson::sample(LOOP_COVER_AVERAGE_DELAY);
-        let delay_duration = Duration::from_secs_f64(delay);
+        let delay_duration = mix_client::poisson::sample_from_duration(average_delay_duration);
         tokio::time::delay_for(delay_duration).await;
 
         let read_lock = topology_ctrl_ref.read().await;