use bytes::Bytes;
use criterion::{BatchSize, Criterion, black_box, criterion_group, criterion_main};
use engineioxide::Packet;
fn criterion_benchmark(c: &mut Criterion) {
let mut group = c.benchmark_group("engineio_packet/decode");
group.bench_function("Decode packet ping/pong", |b| {
let packet: String = Packet::Ping.into();
b.iter_batched(
|| packet.clone(),
|p| Packet::try_from(p).unwrap(),
BatchSize::SmallInput,
)
});
group.bench_function("Decode packet ping/pong upgrade", |b| {
let packet: String = Packet::PingUpgrade.into();
b.iter_batched(
|| packet.clone(),
|p| Packet::try_from(p).unwrap(),
BatchSize::SmallInput,
)
});
group.bench_function("Decode packet message", |b| {
let packet: String = Packet::Message(black_box("Hello").into()).into();
b.iter_batched(
|| packet.clone(),
|p| Packet::try_from(p).unwrap(),
BatchSize::SmallInput,
)
});
group.bench_function("Decode packet noop", |b| {
let packet: String = Packet::Noop.into();
b.iter_batched(
|| packet.clone(),
|p| Packet::try_from(p).unwrap(),
BatchSize::SmallInput,
)
});
group.bench_function("Decode packet binary b64", |b| {
const BYTES: Bytes = Bytes::from_static(&[0x00, 0x01, 0x02, 0x03, 0x04, 0x05]);
let packet: String = Packet::Binary(BYTES).into();
b.iter_batched(
|| packet.clone(),
|p| Packet::try_from(p).unwrap(),
BatchSize::SmallInput,
)
});
group.finish();
}
criterion_group!(benches, criterion_benchmark);
criterion_main!(benches);