CJ_Clippy
24fd2bd5b6
subrepo: subdir: "apps/aquatic" merged: "b139e39f" upstream: origin: "https://github.com/greatest-ape/aquatic" branch: "master" commit: "b139e39f" git-subrepo: version: "0.4.9" origin: "???" commit: "???"
82 lines
2.2 KiB
Rust
82 lines
2.2 KiB
Rust
use std::sync::Arc;
|
|
|
|
use rand::distributions::WeightedIndex;
|
|
use rand::prelude::*;
|
|
use rand_distr::Gamma;
|
|
|
|
use crate::common::*;
|
|
use crate::config::*;
|
|
|
|
pub fn create_random_request(
|
|
config: &Config,
|
|
state: &LoadTestState,
|
|
rng: &mut SmallRng,
|
|
) -> Request {
|
|
let weights = [
|
|
config.torrents.weight_announce as u32,
|
|
config.torrents.weight_scrape as u32,
|
|
];
|
|
|
|
let items = [RequestType::Announce, RequestType::Scrape];
|
|
|
|
let dist = WeightedIndex::new(weights).expect("random request weighted index");
|
|
|
|
match items[dist.sample(rng)] {
|
|
RequestType::Announce => create_announce_request(config, state, rng),
|
|
RequestType::Scrape => create_scrape_request(config, state, rng),
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
fn create_announce_request(config: &Config, state: &LoadTestState, rng: &mut impl Rng) -> Request {
|
|
let (event, bytes_left) = {
|
|
if rng.gen_bool(config.torrents.peer_seeder_probability) {
|
|
(AnnounceEvent::Completed, 0)
|
|
} else {
|
|
(AnnounceEvent::Started, 50)
|
|
}
|
|
};
|
|
|
|
let info_hash_index = select_info_hash_index(config, state, rng);
|
|
|
|
Request::Announce(AnnounceRequest {
|
|
info_hash: state.info_hashes[info_hash_index],
|
|
peer_id: PeerId(rng.gen()),
|
|
bytes_left,
|
|
event,
|
|
key: None,
|
|
numwant: None,
|
|
port: rng.gen(),
|
|
bytes_uploaded: 0,
|
|
bytes_downloaded: 0,
|
|
})
|
|
}
|
|
|
|
#[inline]
|
|
fn create_scrape_request(config: &Config, state: &LoadTestState, rng: &mut impl Rng) -> Request {
|
|
let mut scrape_hashes = Vec::with_capacity(5);
|
|
|
|
for _ in 0..5 {
|
|
let info_hash_index = select_info_hash_index(config, state, rng);
|
|
|
|
scrape_hashes.push(state.info_hashes[info_hash_index]);
|
|
}
|
|
|
|
Request::Scrape(ScrapeRequest {
|
|
info_hashes: scrape_hashes,
|
|
})
|
|
}
|
|
|
|
#[inline]
|
|
fn select_info_hash_index(config: &Config, state: &LoadTestState, rng: &mut impl Rng) -> usize {
|
|
gamma_usize(rng, &state.gamma, config.torrents.number_of_torrents - 1)
|
|
}
|
|
|
|
#[inline]
|
|
fn gamma_usize(rng: &mut impl Rng, gamma: &Arc<Gamma<f64>>, max: usize) -> usize {
|
|
let p: f64 = gamma.sample(rng);
|
|
let p = (p.min(101.0f64) - 1.0) / 100.0;
|
|
|
|
(p * max as f64) as usize
|
|
}
|