use std::time::{Duration, Instant};
use corsa::{Result, fast::SmallVec};
use crate::stats::Stats;
pub async fn measure<F, Fut>(iterations: usize, mut f: F) -> Result<Stats>
where
F: FnMut() -> Fut,
Fut: std::future::Future<Output = Result<()>>,
{
let mut samples = SmallVec::<[Duration; 32]>::new();
let mut index = 0_usize;
while index < iterations {
let started = Instant::now();
f().await?;
samples.push(started.elapsed());
index += 1;
}
Ok(Stats::from_samples(samples))
}
pub async fn measure_warm<F, Fut>(iterations: usize, mut f: F) -> Result<Stats>
where
F: FnMut() -> Fut,
Fut: std::future::Future<Output = Result<()>>,
{
f().await?;
measure(iterations, f).await
}