mdebray/tvl-depot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: Correctly configure body sizes before setting read_function

Browse source
This commit is contained in:
Vincent Ambo 2019-02-26 14:33:42 +01:00
parent 718d945753
commit b71b44a672
1 changed files with 15 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
src/lib.rs
View file

@ -111,15 +111,22 @@ impl <'a> Request<'a> {
let mut headers = HashMap::new();
let mut body = vec![];
// Optionally set content type if a body payload is
// configured.
match self.body {
Body::Bytes { content_type, .. } => self.header("Content-Type", content_type),
Body::NoBody => Ok(&mut self),
// Optionally set content type if a body payload is configured
// and configure the expected body size.
match self.body {
Body::Bytes { content_type, data } => {
self.handle.post_field_size(data.len() as u64)?;
self.headers.append(&format!("Content-Type: {}", content_type))?;
},
#[cfg(feature = "json")]
Body::Json(..) => self.header("Content-Type", "application/json"),
}?;
Body::Json(ref data) => {
self.handle.post_field_size(data.len() as u64)?;
self.headers.append("Content-Type: application/json")?;
},
Body::NoBody => (),
};
// Configure headers on the request:
self.handle.http_headers(self.headers)?;
@ -146,7 +153,7 @@ impl <'a> Request<'a> {
})?,
// Do nothing if there is no body ...
Body::NoBody => {},
Body::NoBody => (),
};
// Read one header per invocation. Request processing is