xerbalind/planetwars.dev
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages 1 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 .await
.unwrap(); .unwrap();
while let Some(Ok(chunk)) = stream.next().await { while let Some(Ok(chunk)) = stream.next().await {
let n_bytes = file.write(&chunk).await.unwrap(); file.write_all(&chunk).await.unwrap();
len += n_bytes; len += chunk.len();
} }
Ok(Response::builder() Ok(Response::builder()
@ -270,9 +270,10 @@ async fn put_upload(
.unwrap(); .unwrap();
while let Some(Ok(chunk)) = stream.next().await { while let Some(Ok(chunk)) = stream.next().await {
let n_bytes = file.write(&chunk).await.unwrap(); file.write_all(&chunk).await.unwrap();
_len += n_bytes; _len += chunk.len();
} }
let digest = params.digest.strip_prefix("sha256:").unwrap(); let digest = params.digest.strip_prefix("sha256:").unwrap();
// TODO: check the digest // TODO: check the digest
let target_path = PathBuf::from(REGISTRY_PATH).join("sha256").join(&digest); let target_path = PathBuf::from(REGISTRY_PATH).join("sha256").join(&digest);