fumble 0.6.1

an oxidized implementation of the original clumsy tool, designed to simulate adverse network conditions on Windows systems.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

use crate::network::modules::stats::util::ewma::Ewma;
use std::time::{Duration, Instant};

pub struct BandwidthStats {
    pub(crate) storage_packet_count: usize,
    pub(crate) total_byte_count: usize,
    ewma: Ewma,
    recent_byte_sent: usize,
    recent_timer: Instant,
    update_interval: Duration,
}

impl BandwidthStats {
    pub fn new(alpha: f64) -> Self {
        BandwidthStats {
            storage_packet_count: 0,
            total_byte_count: 0,
            ewma: Ewma::new(alpha),
            recent_byte_sent: 0,
            recent_timer: Instant::now(),
            update_interval: Duration::from_millis(100),
        }
    }

    pub fn record(&mut self, bytes_sent: usize) {
        self.total_byte_count += bytes_sent;
        self.recent_byte_sent += bytes_sent;
        if self.recent_timer.elapsed() >= self.update_interval {
            self.ewma.update(
                (self.recent_byte_sent as f64 / 1024f64) / self.update_interval.as_secs_f64(),
            );
            self.recent_byte_sent = 0;
            self.recent_timer = Instant::now();
        }
    }

    pub fn recent_throughput(&self) -> f64 {
        self.ewma.get().unwrap_or(0.0)
    }
}