http_stress_test.rs

use std::error::Error;
use clap::Parser;
use futures::StreamExt;
use reqwest::Client;

#[derive(clap::Parser)]
struct Cli {
    #[arg(short, long, default_value_t = 1)]
    count: u32,
}

#[tokio::main(flavor = "current_thread")]
async fn main() -> Result<(), Box<dyn Error>> {
    const URL: &str = "http://127.0.0.1:8983/solr/radio/select?q=utt:hello";
    let cli = Cli::parse();
    let requests = Vec::from_iter(0..cli.count);

    let time_start = std::time::Instant::now();
    let tasks = futures::stream::iter(requests.chunks(200))
        .flat_map(|chunk| {
            let client = Client::new();
            futures::stream::iter(chunk)
                .map(move |i| client.get(format!("{URL} OR utt:{i}")).send())
        })
        .buffer_unordered(800)
        .collect::<Vec<_>>()
        .await;
    let time_elapsed = time_start.elapsed();

    let successful = tasks.iter().filter(|&x| x.is_ok()).count();
    //println!("{}", tasks.pop().unwrap()?.text().await?);
    println!("Received count: {}", successful);
    println!("Time elapsed (ms): {}", time_elapsed.as_millis());
    Ok(())
}