/// Script to compare the results of firefly with an identical redis script.
/// redis - default config from paru
/// cargo run --release --example push_it_redis
///
/// Performance on a 16gb - Intel i7-10510U (8 cores @ 4.9GHz)
/// ~67k ops/sec
use std::{iter::repeat_with, time::Instant};
use uuid::Uuid;

static THREADS: usize = 10;
static REQUESTS_TOTAL: usize = 1_000_000;

async fn add_records(amount: usize) {
    let client = redis::Client::open("redis://127.0.0.1/").unwrap();
    let mut con = client.get_connection().unwrap();
    let user = Uuid::new_v4().to_string();

    for _ in 0..amount {
        let key: String = repeat_with(fastrand::alphanumeric).take(64).collect();

        let _: () = redis::cmd("SET")
            .arg(&key)
            .arg(&user)
            .query(&mut con)
            .unwrap();
    }
}

#[tokio::main]
async fn main() {
    let requests_per_thread = REQUESTS_TOTAL / THREADS;
    let mut futures = Vec::with_capacity(THREADS);

    for _ in 0..THREADS {
        futures.push(add_records(requests_per_thread));
    }

    println!(
        "Starting to send {} requests per thread. ({} threads, {} requests in total)",
        requests_per_thread, THREADS, REQUESTS_TOTAL
    );
    let start = Instant::now();
    futures::future::join_all(futures).await;
    let elapsed = start.elapsed();
    println!(
        "Created {} new records by using {} connections in {:?}.",
        REQUESTS_TOTAL, THREADS, elapsed
    );
    println!(
        "This comes down to {} requests per second.",
        REQUESTS_TOTAL / elapsed.as_secs() as usize
    );
}