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

WIP: cleanup aquatic_http_load_test and aquatic_http_protocol

Browse source
This commit is contained in:
Joakim Frostegård 2020-07-20 14:48:19 +02:00
parent d1e9d24773
commit da3b2bcd8c
10 changed files with 232 additions and 410 deletions
Download patch file Download diff file Expand all files Collapse all files

29
aquatic_http_load_test/src/main.rs
View file

@ -1,22 +1,18 @@
use std::net::{SocketAddr, Ipv4Addr, Ipv6Addr};
use std::thread;
use std::sync::{Arc, atomic::Ordering};
use std::time::{Duration, Instant};
use crossbeam_channel::unbounded;
use rand::prelude::*;
use rand_distr::Pareto;
mod common;
mod handler;
mod config;
mod network;
mod utils;
use common::*;
use utils::*;
use handler::create_random_request;
use config::*;
use network::*;
use handler::run_handler_thread;
#[global_allocator]
@ -43,7 +39,7 @@ fn run(config: Config) -> ::anyhow::Result<()> {
let mut rng = SmallRng::from_entropy();
for _ in 0..config.handler.number_of_torrents {
info_hashes.push(generate_info_hash(&mut rng));
info_hashes.push(InfoHash(rng.gen()));
}
let pareto = Pareto::new(
@ -59,12 +55,7 @@ fn run(config: Config) -> ::anyhow::Result<()> {
// Start socket workers
let mut request_senders = Vec::new();
for _ in 0..config.num_socket_workers {
let (sender, receiver) = unbounded();
request_senders.push(sender);
let config = config.clone();
let state = state.clone();
@ -72,22 +63,10 @@ fn run(config: Config) -> ::anyhow::Result<()> {
thread::spawn(move || run_socket_thread(
&config,
state,
receiver,
1
));
}
// Bootstrap request cycle by adding a request to each request channel
for sender in request_senders.iter(){
let request = create_random_request(
&config,
&state,
&mut thread_rng()
);
sender.send(request.into())
.expect("bootstrap: add initial request to request queue");
}
monitor_statistics(
state,
&config