crab-net 0.0.4

<TCP|UDP> + TLS packets generator
Documentation 
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
41
42

use std::time::Duration;

use byte_unit::Byte;
use kanal::{AsyncSender, bounded_async};
use log::info;
use tokio::{
    select, spawn,
    time::{Instant, interval_at},
};

pub type StatPacket = (usize, usize);

pub fn stats_task(clients: usize) -> AsyncSender<StatPacket> {
    //Define channel to send statistics update
    let (stats_tx, stats_rx) = bounded_async(clients);

    spawn(async move {
        let timer_duration = 10.;
        let duration = Duration::from_secs(timer_duration as u64);
        let mut timer = interval_at(Instant::now() + duration, duration);

        let mut bytes_sent = 0.;
        let mut packets_sent = 0;
        loop {
            select! {
                _ = timer.tick() => {
                    bytes_sent*=8.;
                    let bandwidth = Byte::from_f64(bytes_sent / timer_duration).unwrap_or_default().get_appropriate_unit(byte_unit::UnitType::Decimal).to_string();
                    let bandwidth = &bandwidth[0..bandwidth.len()-1];
                    info!("Sent {packets_sent} packets --- Bandwidth {bandwidth}bit/s");
                    bytes_sent = 0.;
                    packets_sent = 0;
                }
                stat = stats_rx.recv() => if let Ok((bytes,packets)) = stat {
                    bytes_sent += bytes as f64;
                    packets_sent += packets;
                }
            }
        }
    });
    stats_tx
}