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

udp load test: store peer scrape_hash_indices as boxed slice

Browse source
This commit is contained in:
Joakim Frostegård 2024-01-04 17:28:48 +01:00
parent 99791c7154
commit 55516956ca
2 changed files with 9 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

2
crates/udp_load_test/src/common.rs
View file

@ -7,7 +7,7 @@ use aquatic_udp_protocol::*;
#[derive(PartialEq, Eq, Clone)]
pub struct TorrentPeer {
pub info_hash: InfoHash,
pub scrape_hash_indeces: Vec<usize>,
pub scrape_hash_indices: Box<[usize]>,
pub connection_id: ConnectionId,
pub peer_id: PeerId,
pub port: Port,

17
crates/udp_load_test/src/worker/request_gen.rs
View file

@ -172,11 +172,11 @@ fn create_scrape_request(
torrent_peer: &TorrentPeer,
transaction_id: TransactionId,
) -> Request {
let indeces = &torrent_peer.scrape_hash_indeces;
let indeces = &torrent_peer.scrape_hash_indices;
let mut scape_hashes = Vec::with_capacity(indeces.len());
for i in indeces {
for i in indeces.iter() {
scape_hashes.push(info_hashes[*i].to_owned())
}
@ -195,19 +195,18 @@ fn create_torrent_peer(
info_hashes: &Arc<[InfoHash]>,
connection_id: ConnectionId,
) -> TorrentPeer {
let num_scape_hashes = rng.gen_range(1..config.requests.scrape_max_torrents);
let num_scrape_hashes = rng.gen_range(1..config.requests.scrape_max_torrents);
let mut scrape_hash_indeces = Vec::new();
for _ in 0..num_scape_hashes {
scrape_hash_indeces.push(select_info_hash_index(config, rng, gamma))
}
let scrape_hash_indices = (0..num_scrape_hashes)
.map(|_| select_info_hash_index(config, rng, gamma))
.collect::<Vec<_>>()
.into_boxed_slice();
let info_hash_index = select_info_hash_index(config, rng, gamma);
TorrentPeer {
info_hash: info_hashes[info_hash_index],
scrape_hash_indeces,
scrape_hash_indices,
connection_id,
peer_id: generate_peer_id(),
port: Port::new(rng.gen()),