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

Merge pull request #119 from greatest-ape/ws-load-test-fixes

Browse source 
ws load test: enable measuring once all connections are open
This commit is contained in:
Joakim Frostegård 2023-01-26 21:12:50 +01:00 • committed by GitHub
commit 04a93bac59
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 46 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

2
aquatic_ws_load_test/src/config.rs
View file

@ -15,6 +15,7 @@ pub struct Config {
pub num_connections_per_worker: usize,
pub connection_creation_interval_ms: u64,
pub duration: usize,
pub measure_after_max_connections_reached: bool,
pub torrents: TorrentConfig,
pub cpu_pinning: CpuPinningConfigDesc,
}
@ -34,6 +35,7 @@ impl Default for Config {
num_connections_per_worker: 16,
connection_creation_interval_ms: 10,
duration: 0,
measure_after_max_connections_reached: true,
torrents: TorrentConfig::default(),
cpu_pinning: Default::default(),
}

62
aquatic_ws_load_test/src/main.rs
View file

@ -118,6 +118,7 @@ fn create_tls_config() -> anyhow::Result<Arc<rustls::ClientConfig>> {
fn monitor_statistics(state: LoadTestState, config: &Config) {
let start_time = Instant::now();
let mut time_max_connections_reached = None;
let mut report_avg_response_vec: Vec<f64> = Vec::new();
let interval = 5;
@ -155,7 +156,15 @@ fn monitor_statistics(state: LoadTestState, config: &Config) {
+ responses_scrape_per_second
+ responses_error_per_second;
report_avg_response_vec.push(responses_per_second);
if !config.measure_after_max_connections_reached || time_max_connections_reached.is_some() {
report_avg_response_vec.push(responses_per_second);
} else if connections >= config.num_workers * config.num_connections_per_worker {
time_max_connections_reached = Some(Instant::now());
println!();
println!("Max connections reached");
println!();
}
println!();
println!("Requests out: {:.2}/second", requests_per_second);
@ -170,26 +179,43 @@ fn monitor_statistics(state: LoadTestState, config: &Config) {
println!(" - Error responses: {:.2}", responses_error_per_second);
println!("Active connections: {}", connections);
let time_elapsed = start_time.elapsed();
let duration = Duration::from_secs(config.duration as u64);
if config.measure_after_max_connections_reached {
if let Some(start) = time_max_connections_reached {
let time_elapsed = start.elapsed();
if config.duration != 0 && time_elapsed >= duration {
let report_len = report_avg_response_vec.len() as f64;
let report_sum: f64 = report_avg_response_vec.into_iter().sum();
let report_avg: f64 = report_sum / report_len;
if config.duration != 0
&& time_elapsed >= Duration::from_secs(config.duration as u64)
{
report(config, report_avg_response_vec, time_elapsed);
println!(
concat!(
"\n# aquatic load test report\n\n",
"Test ran for {} seconds.\n",
"Average responses per second: {:.2}\n\nConfig: {:#?}\n"
),
time_elapsed.as_secs(),
report_avg,
config
);
break;
}
}
} else {
let time_elapsed = start_time.elapsed();
break;
if config.duration != 0 && time_elapsed >= Duration::from_secs(config.duration as u64) {
report(config, report_avg_response_vec, time_elapsed);
break;
}
}
}
}
fn report(config: &Config, report_avg_response_vec: Vec<f64>, time_elapsed: Duration) {
let report_len = report_avg_response_vec.len() as f64;
let report_sum: f64 = report_avg_response_vec.into_iter().sum();
let report_avg: f64 = report_sum / report_len;
println!(
concat!(
"\n# aquatic load test report\n\n",
"Test ran for {} seconds.\n",
"Average responses per second: {:.2}\n\nConfig: {:#?}\n"
),
time_elapsed.as_secs(),
report_avg,
config
);
}