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

aquatic_http: improve mio and common request handling code

Browse source
This commit is contained in:
Joakim Frostegård 2021-10-26 23:40:11 +02:00
parent ea2366c808
commit 4fc1509a79
2 changed files with 13 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

3
aquatic_http/src/lib/common/handlers.rs
View file

@ -171,7 +171,8 @@ pub fn upsert_peer_and_get_response_peers<I: Ip>(
pub fn handle_scrape_request(
config: &Config,
torrent_maps: &mut TorrentMaps,
(meta, request): (ConnectionMeta, ScrapeRequest),
meta: ConnectionMeta,
request: ScrapeRequest,
) -> ScrapeResponse {
let num_to_take = request
.info_hashes

17
aquatic_http/src/lib/mio/handler.rs
View file

@ -30,7 +30,7 @@ pub fn run_request_worker(
let timeout = Duration::from_micros(config.handlers.channel_recv_timeout_microseconds);
loop {
let mut opt_torrent_map_guard: Option<MutexGuard<TorrentMaps>> = None;
let mut opt_torrent_maps: Option<MutexGuard<TorrentMaps>> = None;
// If torrent state mutex is locked, just keep collecting requests
// and process them later. This can happen with either multiple
@ -51,7 +51,7 @@ pub fn run_request_worker(
}
None => {
if let Some(torrent_guard) = state.torrent_maps.try_lock() {
opt_torrent_map_guard = Some(torrent_guard);
opt_torrent_maps = Some(torrent_guard);
break;
}
@ -59,8 +59,8 @@ pub fn run_request_worker(
}
}
let mut torrent_map_guard =
opt_torrent_map_guard.unwrap_or_else(|| state.torrent_maps.lock());
let mut torrent_maps =
opt_torrent_maps.unwrap_or_else(|| state.torrent_maps.lock());
let valid_until = ValidUntil::new(config.cleaning.max_peer_age);
@ -68,7 +68,7 @@ pub fn run_request_worker(
let response = handle_announce_request(
&config,
&mut rng,
&mut torrent_map_guard,
&mut torrent_maps,
valid_until,
meta,
request
@ -79,7 +79,12 @@ pub fn run_request_worker(
}
for (meta, request) in scrape_requests.drain(..) {
let response = handle_scrape_request(&config, &mut torrent_map_guard, (meta, request));
let response = handle_scrape_request(
&config,
&mut torrent_maps,
meta,
request
);
response_channel_sender.send(meta, Response::Scrape(response));
wake_socket_workers[meta.worker_index] = true;