feat(body): add body::aggregate and body::to_bytes functions
Adds utility functions to `hyper::body` to help asynchronously collecting all the buffers of some `HttpBody` into one. - `aggregate` will collect all into an `impl Buf` without copying the contents. This is ideal if you don't need a contiguous buffer. - `to_bytes` will copy all the data into a single contiguous `Bytes` buffer.
This commit is contained in:
89
benches/body.rs
Normal file
89
benches/body.rs
Normal file
@@ -0,0 +1,89 @@
|
||||
#![feature(test)]
|
||||
#![deny(warnings)]
|
||||
|
||||
extern crate test;
|
||||
|
||||
use bytes::Buf;
|
||||
use futures_util::stream;
|
||||
use futures_util::StreamExt;
|
||||
use hyper::body::Body;
|
||||
|
||||
macro_rules! bench_stream {
|
||||
($bencher:ident, bytes: $bytes:expr, count: $count:expr, $total_ident:ident, $body_pat:pat, $block:expr) => {{
|
||||
let mut rt = tokio::runtime::Builder::new()
|
||||
.basic_scheduler()
|
||||
.build()
|
||||
.expect("rt build");
|
||||
|
||||
let $total_ident: usize = $bytes * $count;
|
||||
$bencher.bytes = $total_ident as u64;
|
||||
let __s: &'static [&'static [u8]] = &[&[b'x'; $bytes] as &[u8]; $count] as _;
|
||||
|
||||
$bencher.iter(|| {
|
||||
rt.block_on(async {
|
||||
let $body_pat = Body::wrap_stream(
|
||||
stream::iter(__s.iter()).map(|&s| Ok::<_, std::convert::Infallible>(s)),
|
||||
);
|
||||
$block;
|
||||
});
|
||||
});
|
||||
}};
|
||||
}
|
||||
|
||||
macro_rules! benches {
|
||||
($($name:ident, $bytes:expr, $count:expr;)+) => (
|
||||
mod aggregate {
|
||||
use super::*;
|
||||
|
||||
$(
|
||||
#[bench]
|
||||
fn $name(b: &mut test::Bencher) {
|
||||
bench_stream!(b, bytes: $bytes, count: $count, total, body, {
|
||||
let buf = hyper::body::aggregate(body).await.unwrap();
|
||||
assert_eq!(buf.remaining(), total);
|
||||
});
|
||||
}
|
||||
)+
|
||||
}
|
||||
|
||||
mod manual_into_vec {
|
||||
use super::*;
|
||||
|
||||
$(
|
||||
#[bench]
|
||||
fn $name(b: &mut test::Bencher) {
|
||||
bench_stream!(b, bytes: $bytes, count: $count, total, mut body, {
|
||||
let mut vec = Vec::new();
|
||||
while let Some(chunk) = body.next().await {
|
||||
vec.extend_from_slice(&chunk.unwrap());
|
||||
}
|
||||
assert_eq!(vec.len(), total);
|
||||
});
|
||||
}
|
||||
)+
|
||||
}
|
||||
|
||||
mod to_bytes {
|
||||
use super::*;
|
||||
|
||||
$(
|
||||
#[bench]
|
||||
fn $name(b: &mut test::Bencher) {
|
||||
bench_stream!(b, bytes: $bytes, count: $count, total, body, {
|
||||
let bytes = hyper::body::to_bytes(body).await.unwrap();
|
||||
assert_eq!(bytes.len(), total);
|
||||
});
|
||||
}
|
||||
)+
|
||||
}
|
||||
)
|
||||
}
|
||||
|
||||
// ===== Actual Benchmarks =====
|
||||
|
||||
benches! {
|
||||
bytes_1_000_count_2, 1_000, 2;
|
||||
bytes_1_000_count_10, 1_000, 10;
|
||||
bytes_10_000_count_1, 10_000, 1;
|
||||
bytes_10_000_count_10, 10_000, 10;
|
||||
}
|
||||
Reference in New Issue
Block a user