tyler/proxmox-backup
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

src/client/http_client.rs - h2 download: implement flow control

Browse Source
This commit is contained in:
Dietmar Maurer 2019-06-28 07:02:43 +02:00
parent cf925b94ab
commit 984a7c3502
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/client/http_client.rs
View File

@ -1001,10 +1001,14 @@ impl H2Client {
.and_then(|_| { bail!("unknown error"); }) .and_then(|_| { bail!("unknown error"); })
) )
} else { } else {
let mut body = resp.into_body();
let mut release_capacity = body.release_capacity().clone();
future::Either::B( future::Either::B(
resp.into_body() body
.map_err(Error::from) .map_err(Error::from)
.fold(output, move |mut acc, chunk| { .fold(output, move |mut acc, chunk| {
let _ = release_capacity.release_capacity(chunk.len());
acc.write_all(&chunk)?; acc.write_all(&chunk)?;
Ok::<_, Error>(acc) Ok::<_, Error>(acc)
}) })