move flush in write_to_file function

This commit is contained in:
marin postma 2021-06-23 16:34:07 +02:00
parent 880fc069bd
commit f68c257452
No known key found for this signature in database
GPG Key ID: 6088B7721C3E39F9
1 changed files with 4 additions and 1 deletions

View File

@ -145,11 +145,15 @@ where
D: AsRef<[u8]> + Sized + 'static,
{
let mut file_len = 0;
while let Some(bytes) = payload.recv().await {
let bytes = bytes?;
file_len += bytes.as_ref().len();
file.write_all(bytes.as_ref()).await?;
}
file.flush().await?;
Ok(file_len)
}
@ -157,7 +161,6 @@ where
match file_len {
Ok(len) if len > 0 => {
file.flush().await?;
let file = file.into_std().await;
Some((file, update_file_id))
}