YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-03-31 17:55:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

udp: mio: don't recalculate PeerUntil every announce request

Browse source
This commit is contained in:
Joakim Frostegård 2024-02-10 22:22:19 +01:00
parent 19533b3f8e
commit 94247b8e35
1 changed files with 18 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

24
crates/udp/src/workers/socket/mio.rs
View file

@ -32,6 +32,7 @@ pub struct SocketWorker {
socket: UdpSocket,
buffer: [u8; BUFFER_SIZE],
rng: SmallRng,
peer_valid_until: ValidUntil,
}
impl SocketWorker {
@ -45,6 +46,10 @@ impl SocketWorker {
) -> anyhow::Result<()> {
let socket = UdpSocket::from_std(create_socket(&config, priv_dropper)?);
let access_list_cache = create_access_list_cache(&shared_state.access_list);
let peer_valid_until = ValidUntil::new(
shared_state.server_start_instant,
config.cleaning.max_peer_age,
);
let mut worker = Self {
config,
@ -56,6 +61,7 @@ impl SocketWorker {
socket,
buffer: [0; BUFFER_SIZE],
rng: SmallRng::from_entropy(),
peer_valid_until,
};
worker.run_inner()
@ -73,6 +79,8 @@ impl SocketWorker {
let poll_timeout = Duration::from_millis(self.config.network.poll_timeout_ms);
let mut iter_counter = 0u64;
loop {
poll.poll(&mut events, Some(poll_timeout)).context("poll")?;
@ -88,6 +96,15 @@ impl SocketWorker {
self.send_response(&mut None, addr, response);
}
}
if iter_counter % 256 == 0 {
self.peer_valid_until = ValidUntil::new(
self.shared_state.server_start_instant,
self.config.cleaning.max_peer_age,
);
}
iter_counter = iter_counter.wrapping_add(1);
}
}
@ -194,18 +211,13 @@ impl SocketWorker {
.load()
.allows(access_list_mode, &request.info_hash.0)
{
let peer_valid_until = ValidUntil::new(
self.shared_state.server_start_instant,
self.config.cleaning.max_peer_age,
);
let response = self.shared_state.torrent_maps.announce(
&self.config,
&self.statistics_sender,
&mut self.rng,
&request,
src,
peer_valid_until,
self.peer_valid_until,
);
return Some(response);