nats 0.9.16

use criterion::{
    criterion_group, criterion_main, BenchmarkId, Criterion, Throughput,
};
use std::time::{Duration, Instant};

use nats;

pub fn pub_benchmark(c: &mut Criterion) {
    let mut group = c.benchmark_group("publish");
    group.warm_up_time(Duration::from_secs(1));
    // TODO(dlc) - Can both be done at some time or requires custom?
    //group.throughput(Throughput::Bytes(*size as u64));
    group.throughput(Throughput::Elements(1));

    let bmsg: Vec<u8> = (0..32768).map(|_| 22).collect();
    for size in [32, 128, 256, 1024, 4096, 8192].iter() {
        let nc = nats::connect("127.0.0.1").unwrap();
        let msg = &bmsg[0..*size];
        group.bench_with_input(
            BenchmarkId::from_parameter(size),
            size,
            |b, _| {
                b.iter_custom(|n| {
                    let start = Instant::now();
                    for _i in 0..n {
                        nc.publish("bench", &msg).unwrap();
                    }
                    nc.flush().unwrap();
                    start.elapsed()
                });
            },
        );
    }
    group.finish();
}

criterion_group!(benches, pub_benchmark);
criterion_main!(benches);