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_ws: in scrape handler, limit HashMap preallocation

Browse source
This commit is contained in:
Joakim Frostegård 2020-05-17 21:22:32 +02:00
parent 9a04173f65
commit 1a2432733e
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
aquatic_ws/src/lib/handler.rs
View file

@ -218,17 +218,17 @@ pub fn handle_scrape_requests(
requests: Drain<(ConnectionMeta, ScrapeRequest)>,
){
messages_out.extend(requests.map(|(meta, request)| {
let num_info_hashes = request.info_hashes.len();
let num_to_take = request.info_hashes.len().min(
config.network.max_scrape_torrents
);
let mut response = ScrapeResponse {
files: HashMap::with_capacity(num_info_hashes),
files: HashMap::with_capacity(num_to_take),
};
let max_torrents = config.network.max_scrape_torrents;
// If request.info_hashes is empty, don't return scrape for all
// torrents, even though reference server does it. It is too expensive.
for info_hash in request.info_hashes.into_iter().take(max_torrents){
for info_hash in request.info_hashes.into_iter().take(num_to_take){
if let Some(torrent_data) = torrents.get(&info_hash){
let stats = ScrapeStatistics {
complete: torrent_data.num_seeders,