From 03a344eb8e96343a95129de67b5c923781a30959 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Joakim=20Frosteg=C3=A5rd?= Date: Sun, 31 Oct 2021 19:57:30 +0100 Subject: [PATCH] aquatic_udp: mio: print num torrents and peers in statistics --- aquatic_udp/src/lib/mio/tasks.rs | 28 ++++++++++++++++++++++++---- 1 file changed, 24 insertions(+), 4 deletions(-) diff --git a/aquatic_udp/src/lib/mio/tasks.rs b/aquatic_udp/src/lib/mio/tasks.rs index 4d9fe16..c4bcac3 100644 --- a/aquatic_udp/src/lib/mio/tasks.rs +++ b/aquatic_udp/src/lib/mio/tasks.rs @@ -38,27 +38,47 @@ pub fn gather_and_print_statistics(state: &State, config: &Config) { bytes_sent_per_second * 8.0 / 1_000_000.0, ); + let mut total_num_torrents_ipv4 = 0usize; + let mut total_num_torrents_ipv6 = 0usize; + let mut total_num_peers_ipv4 = 0usize; + let mut total_num_peers_ipv6 = 0usize; + let mut peers_per_torrent = Histogram::new(); { let torrents = &mut state.torrents.lock(); for torrent in torrents.ipv4.values() { - let num_peers = (torrent.num_seeders + torrent.num_leechers) as u64; + let num_peers = torrent.num_seeders + torrent.num_leechers; - if let Err(err) = peers_per_torrent.increment(num_peers) { + if let Err(err) = peers_per_torrent.increment(num_peers as u64) { ::log::error!("error incrementing peers_per_torrent histogram: {}", err) } + + total_num_peers_ipv4 += num_peers; } for torrent in torrents.ipv6.values() { - let num_peers = (torrent.num_seeders + torrent.num_leechers) as u64; + let num_peers = torrent.num_seeders + torrent.num_leechers; - if let Err(err) = peers_per_torrent.increment(num_peers) { + if let Err(err) = peers_per_torrent.increment(num_peers as u64) { ::log::error!("error incrementing peers_per_torrent histogram: {}", err) } + + total_num_peers_ipv6 += num_peers; } + + total_num_torrents_ipv4 += torrents.ipv4.len(); + total_num_torrents_ipv6 += torrents.ipv6.len(); } + println!( + "ipv4 torrents: {}, peers: {}; ipv6 torrents: {}, peers: {}", + total_num_torrents_ipv4, + total_num_peers_ipv4, + total_num_torrents_ipv6, + total_num_peers_ipv6, + ); + if peers_per_torrent.entries() != 0 { println!( "peers per torrent: min: {}, p50: {}, p75: {}, p90: {}, p99: {}, p999: {}, max: {}",