tokio/benches/mpsc.rs

use bencher::{black_box, Bencher};
use tokio::sync::mpsc;

type Medium = [usize; 64];
type Large = [Medium; 64];

fn create_1_medium(b: &mut Bencher) {
    b.iter(|| {
        black_box(&mpsc::channel::<Medium>(1));
    });
}

fn create_100_medium(b: &mut Bencher) {
    b.iter(|| {
        black_box(&mpsc::channel::<Medium>(100));
    });
}

fn create_100_000_medium(b: &mut Bencher) {
    b.iter(|| {
        black_box(&mpsc::channel::<Medium>(100_000));
    });
}

fn send_medium(b: &mut Bencher) {
    b.iter(|| {
        let (tx, mut rx) = mpsc::channel::<Medium>(1000);

        let _ = tx.try_send([0; 64]);

        rx.try_recv().unwrap();
    });
}

fn send_large(b: &mut Bencher) {
    b.iter(|| {
        let (tx, mut rx) = mpsc::channel::<Large>(1000);

        let _ = tx.try_send([[0; 64]; 64]);

        rx.try_recv().unwrap();
    });
}

fn contention_bounded(b: &mut Bencher) {
    let rt = tokio::runtime::Builder::new()
        .core_threads(6)
        .threaded_scheduler()
        .build()
        .unwrap();

    b.iter(|| {
        rt.block_on(async move {
            let (tx, mut rx) = mpsc::channel::<usize>(1_000_000);

            for _ in 0..5 {
                let tx = tx.clone();
                tokio::spawn(async move {
                    for i in 0..1000 {
                        tx.send(i).await.unwrap();
                    }
                });
            }

            for _ in 0..1_000 * 5 {
                let _ = rx.recv().await;
            }
        })
    });
}

fn contention_bounded_full(b: &mut Bencher) {
    let rt = tokio::runtime::Builder::new()
        .core_threads(6)
        .threaded_scheduler()
        .build()
        .unwrap();

    b.iter(|| {
        rt.block_on(async move {
            let (tx, mut rx) = mpsc::channel::<usize>(100);

            for _ in 0..5 {
                let tx = tx.clone();
                tokio::spawn(async move {
                    for i in 0..1000 {
                        tx.send(i).await.unwrap();
                    }
                });
            }

            for _ in 0..1_000 * 5 {
                let _ = rx.recv().await;
            }
        })
    });
}

fn contention_unbounded(b: &mut Bencher) {
    let rt = tokio::runtime::Builder::new()
        .core_threads(6)
        .threaded_scheduler()
        .build()
        .unwrap();

    b.iter(|| {
        rt.block_on(async move {
            let (tx, mut rx) = mpsc::unbounded_channel::<usize>();

            for _ in 0..5 {
                let tx = tx.clone();
                tokio::spawn(async move {
                    for i in 0..1000 {
                        tx.send(i).unwrap();
                    }
                });
            }

            for _ in 0..1_000 * 5 {
                let _ = rx.recv().await;
            }
        })
    });
}

fn uncontented_bounded(b: &mut Bencher) {
    let rt = tokio::runtime::Builder::new()
        .core_threads(6)
        .threaded_scheduler()
        .build()
        .unwrap();

    b.iter(|| {
        rt.block_on(async move {
            let (tx, mut rx) = mpsc::channel::<usize>(1_000_000);

            for i in 0..5000 {
                tx.send(i).await.unwrap();
            }

            for _ in 0..5_000 {
                let _ = rx.recv().await;
            }
        })
    });
}

fn uncontented_unbounded(b: &mut Bencher) {
    let rt = tokio::runtime::Builder::new()
        .core_threads(6)
        .threaded_scheduler()
        .build()
        .unwrap();

    b.iter(|| {
        rt.block_on(async move {
            let (tx, mut rx) = mpsc::unbounded_channel::<usize>();

            for i in 0..5000 {
                tx.send(i).unwrap();
            }

            for _ in 0..5_000 {
                let _ = rx.recv().await;
            }
        })
    });
}

bencher::benchmark_group!(
    create,
    create_1_medium,
    create_100_medium,
    create_100_000_medium
);

bencher::benchmark_group!(send, send_medium, send_large);

bencher::benchmark_group!(
    contention,
    contention_bounded,
    contention_bounded_full,
    contention_unbounded,
    uncontented_bounded,
    uncontented_unbounded
);

bencher::benchmark_main!(create, send, contention);