diff --git a/src/main.rs b/src/main.rs index 5005a28..b9c1eb9 100644 --- a/src/main.rs +++ b/src/main.rs @@ -47,6 +47,9 @@ fn main() { .unwrap_or_else(|| "20".to_owned()) .parse() .expect("second parameter should interger"); + + let fmt = std::env::args().nth(3); + let mut stats = IncrementalStats::new(); for i in 0..iter_num { let start = Instant::now(); @@ -54,12 +57,17 @@ fn main() { Ok(response) => { if response.status().is_success() { let duration = start.elapsed(); - let r = response.bytes().unwrap(); - println!( - "第{i}次测试, 访问 {} 花费了 {:?} 毫秒", - url, - duration.as_millis() - ); + let _r = response.bytes().unwrap(); + if let Some(_name) = fmt.as_ref() { + // println!("{name}, {i},{}", duration.as_millis()) + } else { + println!( + "第{i}次测试, 访问 {} 花费了 {:?} 毫秒", + url, + duration.as_millis() + ); + } + // time_rec.push(duration); stats.add(duration.as_millis() as f64); } else { @@ -75,9 +83,14 @@ fn main() { } } } - println!( - "平均: {:?}ms, 方差: {:?}", - stats.average(), - stats.variance() - ); + + if let Some(name) = fmt.as_ref() { + println!("{name}, {},{}", stats.average(), stats.variance()) + } else { + println!( + "平均: {:?}ms, 方差: {:?}", + stats.average(), + stats.variance() + ); + } }