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

udp bench: use gamma distribution for torrent selection

Browse source
This commit is contained in:
Joakim Frostegård 2022-10-26 20:03:59 +02:00
parent db561a1101
commit 84f420e2c5
3 changed files with 12 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

6
aquatic_udp_bench/src/scrape.rs
View file

@ -5,7 +5,7 @@ use aquatic_common::CanonicalSocketAddr;
use crossbeam_channel::{Receiver, Sender};
use indicatif::ProgressIterator;
use rand::Rng;
use rand_distr::Pareto;
use rand_distr::Gamma;
use aquatic_udp::common::*;
use aquatic_udp_protocol::*;
@ -93,7 +93,7 @@ pub fn create_requests(
number: usize,
hashes_per_request: usize,
) -> Vec<(ScrapeRequest, CanonicalSocketAddr)> {
let pareto = Pareto::new(1., PARETO_SHAPE).unwrap();
let gamma = Gamma::new(GAMMA_SHAPE, GAMMA_SCALE).unwrap();
let max_index = info_hashes.len() - 1;
@ -103,7 +103,7 @@ pub fn create_requests(
let mut request_info_hashes = Vec::new();
for _ in 0..hashes_per_request {
let info_hash_index = pareto_usize(rng, pareto, max_index);
let info_hash_index = gamma_usize(rng, gamma, max_index);
request_info_hashes.push(info_hashes[info_hash_index])
}