gil 0.8.0

use gil::spmc::channel;
use std::{hint::black_box, num::NonZeroUsize, thread::spawn, time::SystemTime};

fn main() {
    let (mut tx, rx) = channel(NonZeroUsize::new(4096).unwrap());

    const RECEIVERS: usize = 4;
    const MESSAGES: usize = 5_000_000;

    let start = SystemTime::now();

    let mut handles = Vec::with_capacity(RECEIVERS);
    for _ in 0..RECEIVERS {
        let mut rx = rx.clone();
        handles.push(spawn(move || {
            for _ in 0..MESSAGES {
                loop {
                    let Some(x) = rx.try_recv() else {
                        continue;
                    };
                    black_box(x);
                    break;
                }
            }
        }));
    }

    // Drop the original receiver since we've cloned it for all threads
    drop(rx);

    for i in 0..(RECEIVERS * MESSAGES) {
        while tx.try_send(black_box(i)).is_err() {}
    }

    for handle in handles {
        handle.join().unwrap();
    }

    let time = start.elapsed().unwrap();
    println!(
        "time={time:?}\nmil/s={throughput:.2}",
        throughput = MESSAGES as f64 / 1_000_000.0 / time.as_secs_f64()
    );
}