lune-org/lune
1
1
Fork 0
mirror of https://github.com/lune-org/lune.git synced 2025-05-04 10:43:57 +01:00
Code Issues Projects Releases Packages Wiki Activity

Preallocate incoming body

Browse source
This commit is contained in:
Filip Tibell 2025-04-26 21:10:27 +02:00
parent f5993c6505
commit ece2c1adf2
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
crates/lune-std-net/src/shared/response.rs
View file

@ -2,7 +2,7 @@ use futures_lite::prelude::*;
use http_body_util::BodyStream; use http_body_util::BodyStream;
use hyper::{ use hyper::{
body::{Bytes, Incoming}, body::{Body, Bytes, Incoming},
HeaderMap, Response as HyperResponse, HeaderMap, Response as HyperResponse,
}; };
@ -23,8 +23,10 @@ impl Response {
) -> LuaResult<Self> { ) -> LuaResult<Self> {
let (parts, body) = incoming.into_parts(); let (parts, body) = incoming.into_parts();
let size = body.size_hint().lower() as usize;
let buffer = Vec::<u8>::with_capacity(size);
let body = BodyStream::new(body) let body = BodyStream::new(body)
.try_fold(Vec::<u8>::new(), |mut body, chunk| { .try_fold(buffer, |mut body, chunk| {
if let Some(chunk) = chunk.data_ref() { if let Some(chunk) = chunk.data_ref() {
body.extend_from_slice(chunk); body.extend_from_slice(chunk);
} }