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: add config.summarize_last: only summarize last N seconds

Browse source
This commit is contained in:
Joakim Frostegård 2024-01-03 17:07:06 +01:00
parent e18b50227c
commit 7863782413
2 changed files with 29 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
crates/udp_load_test/src/config.rs
View file

@ -21,6 +21,10 @@ pub struct Config {
pub workers: u8,
/// Run duration (quit and generate report after this many seconds)
pub duration: usize,
/// Only report summary for the last N seconds of run
///
/// 0 = include whole run
pub summarize_last: usize,
pub network: NetworkConfig,
pub requests: RequestConfig,
#[cfg(feature = "cpu-pinning")]
@ -34,6 +38,7 @@ impl Default for Config {
log_level: LogLevel::Error,
workers: 1,
duration: 0,
summarize_last: 0,
network: NetworkConfig::default(),
requests: RequestConfig::default(),
#[cfg(feature = "cpu-pinning")]

27
crates/udp_load_test/src/lib.rs
View file

@ -33,6 +33,10 @@ pub fn run(config: Config) -> ::anyhow::Result<()> {
panic!("Error: at least one weight must be larger than zero.");
}
if config.summarize_last > config.duration {
panic!("Error: report_last_seconds can't be larger than duration");
}
println!("Starting client with config: {:#?}", config);
let mut info_hashes = Vec::with_capacity(config.requests.number_of_torrents);
@ -103,7 +107,7 @@ fn monitor_statistics(state: LoadTestState, config: &Config) {
let mut report_avg_scrape: Vec<f64> = Vec::new();
let mut report_avg_error: Vec<f64> = Vec::new();
let interval = 5;
const INTERVAL: u64 = 5;
let start_time = Instant::now();
let duration = Duration::from_secs(config.duration as u64);
@ -111,7 +115,7 @@ fn monitor_statistics(state: LoadTestState, config: &Config) {
let mut last = start_time;
let time_elapsed = loop {
thread::sleep(Duration::from_secs(interval));
thread::sleep(Duration::from_secs(INTERVAL));
let requests = fetch_and_reset(&state.statistics.requests);
let response_peers = fetch_and_reset(&state.statistics.response_peers);
@ -163,6 +167,15 @@ fn monitor_statistics(state: LoadTestState, config: &Config) {
}
};
if config.summarize_last != 0 {
let split_at = (config.duration - config.summarize_last) / INTERVAL as usize;
report_avg_connect = report_avg_connect.split_off(split_at);
report_avg_announce = report_avg_announce.split_off(split_at);
report_avg_scrape = report_avg_scrape.split_off(split_at);
report_avg_error = report_avg_error.split_off(split_at);
}
let len = report_avg_connect.len() as f64;
let avg_connect: f64 = report_avg_connect.into_iter().sum::<f64>() / len;
@ -175,7 +188,15 @@ fn monitor_statistics(state: LoadTestState, config: &Config) {
println!();
println!("# aquatic load test report");
println!();
println!("Test ran for {} seconds", time_elapsed.as_secs());
println!(
"Test ran for {} seconds {}",
time_elapsed.as_secs(),
if config.summarize_last != 0 {
format!("(only last {} included in summary)", config.summarize_last)
} else {
"".to_string()
}
);
println!("Average responses per second: {:.2}", avg_total);
println!(" - Connect responses: {:.2}", avg_connect);
println!(" - Announce responses: {:.2}", avg_announce);