YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-04-02 18:55:32 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

udp: statistics: show number of torrents and access list len

Browse source
This commit is contained in:
Joakim Frostegård 2021-11-19 12:04:16 +01:00
parent 31e44db469
commit 59e95894b9
7 changed files with 84 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

17
aquatic_udp/src/lib/handlers.rs
View file

@ -3,6 +3,7 @@ use std::net::IpAddr;
use std::net::Ipv4Addr;
use std::net::Ipv6Addr;
use std::net::SocketAddr;
use std::sync::atomic::Ordering;
use std::time::Duration;
use std::time::Instant;
@ -84,6 +85,7 @@ pub fn run_request_worker(
state: State,
request_receiver: Receiver<(SocketWorkerIndex, ConnectedRequest, SocketAddr)>,
response_sender: ConnectedResponseSender,
worker_index: RequestWorkerIndex,
) {
let mut torrents = TorrentMaps::default();
let mut small_rng = SmallRng::from_entropy();
@ -92,9 +94,12 @@ pub fn run_request_worker(
let mut peer_valid_until = ValidUntil::new(config.cleaning.max_peer_age);
let cleaning_interval = Duration::from_secs(config.cleaning.torrent_cleaning_interval);
let statistics_update_interval = Duration::from_secs(config.statistics.interval);
let mut last_cleaning = Instant::now();
let mut last_statistics_update = Instant::now();
let mut iter_counter = 0usize;
let mut last_cleaning = Instant::now();
loop {
if let Ok((sender_index, request, src)) = request_receiver.recv_timeout(timeout) {
@ -125,6 +130,16 @@ pub fn run_request_worker(
last_cleaning = now;
}
if !statistics_update_interval.is_zero()
&& now > last_statistics_update + statistics_update_interval
{
state.statistics.torrents_ipv4[worker_index.0]
.store(torrents.ipv4.len(), Ordering::SeqCst);
state.statistics.torrents_ipv6[worker_index.0]
.store(torrents.ipv6.len(), Ordering::SeqCst);
last_statistics_update = now;
}
}
iter_counter = iter_counter.wrapping_add(1);