mirror of https://github.com/fafhrd91/actix-net
100 lines
3.0 KiB
Rust
100 lines
3.0 KiB
Rust
#![allow(missing_docs)]
|
|
|
|
use bytes::BytesMut;
|
|
use criterion::{criterion_group, criterion_main, Criterion};
|
|
|
|
const INPUT: &[u8] = include_bytes!("./lorem.txt");
|
|
const PARTIAL_CHUNK: [u8; 64] = [b'a'; 64];
|
|
const PARTIAL_CHUNKS: usize = 128;
|
|
|
|
fn bench_lines_codec(c: &mut Criterion) {
|
|
let mut decode_group = c.benchmark_group("lines decode");
|
|
|
|
decode_group.bench_function("actix", |b| {
|
|
b.iter(|| {
|
|
use actix_codec::Decoder as _;
|
|
|
|
let mut codec = actix_codec::LinesCodec::default();
|
|
let mut buf = BytesMut::from(INPUT);
|
|
while let Ok(Some(_bytes)) = codec.decode_eof(&mut buf) {}
|
|
});
|
|
});
|
|
|
|
decode_group.bench_function("tokio", |b| {
|
|
b.iter(|| {
|
|
use tokio_util::codec::Decoder as _;
|
|
|
|
let mut codec = tokio_util::codec::LinesCodec::new();
|
|
let mut buf = BytesMut::from(INPUT);
|
|
while let Ok(Some(_bytes)) = codec.decode_eof(&mut buf) {}
|
|
});
|
|
});
|
|
|
|
decode_group.finish();
|
|
|
|
let mut partial_decode_group = c.benchmark_group("lines decode partial");
|
|
|
|
partial_decode_group.bench_function("actix", |b| {
|
|
b.iter(|| {
|
|
use actix_codec::Decoder as _;
|
|
|
|
let mut codec = actix_codec::LinesCodec::default();
|
|
let mut buf = BytesMut::with_capacity(PARTIAL_CHUNK.len() * PARTIAL_CHUNKS + 1);
|
|
|
|
for _ in 0..PARTIAL_CHUNKS {
|
|
buf.extend_from_slice(&PARTIAL_CHUNK);
|
|
assert!(codec.decode(&mut buf).unwrap().is_none());
|
|
}
|
|
|
|
buf.extend_from_slice(b"\n");
|
|
assert!(codec.decode(&mut buf).unwrap().is_some());
|
|
});
|
|
});
|
|
|
|
partial_decode_group.bench_function("tokio", |b| {
|
|
b.iter(|| {
|
|
use tokio_util::codec::Decoder as _;
|
|
|
|
let mut codec = tokio_util::codec::LinesCodec::new();
|
|
let mut buf = BytesMut::with_capacity(PARTIAL_CHUNK.len() * PARTIAL_CHUNKS + 1);
|
|
|
|
for _ in 0..PARTIAL_CHUNKS {
|
|
buf.extend_from_slice(&PARTIAL_CHUNK);
|
|
assert!(codec.decode(&mut buf).unwrap().is_none());
|
|
}
|
|
|
|
buf.extend_from_slice(b"\n");
|
|
assert!(codec.decode(&mut buf).unwrap().is_some());
|
|
});
|
|
});
|
|
|
|
partial_decode_group.finish();
|
|
|
|
let mut encode_group = c.benchmark_group("lines encode");
|
|
|
|
encode_group.bench_function("actix", |b| {
|
|
b.iter(|| {
|
|
use actix_codec::Encoder as _;
|
|
|
|
let mut codec = actix_codec::LinesCodec::default();
|
|
let mut buf = BytesMut::new();
|
|
codec.encode("123", &mut buf).unwrap();
|
|
});
|
|
});
|
|
|
|
encode_group.bench_function("tokio", |b| {
|
|
b.iter(|| {
|
|
use tokio_util::codec::Encoder as _;
|
|
|
|
let mut codec = tokio_util::codec::LinesCodec::new();
|
|
let mut buf = BytesMut::new();
|
|
codec.encode("123", &mut buf).unwrap();
|
|
});
|
|
});
|
|
|
|
encode_group.finish();
|
|
}
|
|
|
|
criterion_group!(benches, bench_lines_codec);
|
|
criterion_main!(benches);
|