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

aquatic_ws: protocol: add info hash vector parsing; refactor

Browse source
This commit is contained in:
Joakim Frostegård 2020-05-09 15:07:25 +02:00
parent 9509c9125f
commit 460a778d0d
3 changed files with 222 additions and 60 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -194,28 +194,23 @@ pub fn handle_scrape_requests(
requests: Drain<(ConnectionMeta, ScrapeRequest)>,
){
messages_out.extend(requests.map(|(meta, request)| {
let num_info_hashes = request.info_hashes
.as_ref()
.map(|v| v.len())
.unwrap_or(0);
let num_info_hashes = request.info_hashes.len();
let mut response = ScrapeResponse {
files: HashMap::with_capacity(num_info_hashes),
};
// If request.info_hashes is None, don't return scrape for all
// If request.info_hashes is empty, don't return scrape for all
// torrents, even though reference server does it. It is too expensive.
if let Some(info_hashes) = request.info_hashes {
for info_hash in info_hashes {
if let Some(torrent_data) = torrents.get(&info_hash){
let stats = ScrapeStatistics {
complete: torrent_data.num_seeders,
downloaded: 0, // No implementation planned
incomplete: torrent_data.num_leechers,
};
for info_hash in request.info_hashes {
if let Some(torrent_data) = torrents.get(&info_hash){
let stats = ScrapeStatistics {
complete: torrent_data.num_seeders,
downloaded: 0, // No implementation planned
incomplete: torrent_data.num_leechers,
};
response.files.insert(info_hash, stats);
}
response.files.insert(info_hash, stats);
}
}