midgard/planetwars.dev
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

make sure that all pushed data is actually written

Browse source
This commit is contained in:
Ilion Beyst 2022-06-30 20:28:37 +02:00
parent d7e4a1fd5c
commit 4d1c0a3289
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
planetwars-server/src/modules/registry.rs
View file

@ -227,8 +227,8 @@ async fn patch_upload(
.await
.unwrap();
while let Some(Ok(chunk)) = stream.next().await {
let n_bytes = file.write(&chunk).await.unwrap();
len += n_bytes;
file.write_all(&chunk).await.unwrap();
len += chunk.len();
}
Ok(Response::builder()
@ -270,9 +270,10 @@ async fn put_upload(
.unwrap();
while let Some(Ok(chunk)) = stream.next().await {
let n_bytes = file.write(&chunk).await.unwrap();
_len += n_bytes;
file.write_all(&chunk).await.unwrap();
_len += chunk.len();
}
let digest = params.digest.strip_prefix("sha256:").unwrap();
// TODO: check the digest
let target_path = PathBuf::from(REGISTRY_PATH).join("sha256").join(&digest);