fix chunking bug

This commit is contained in:
cy 2025-04-13 23:52:31 -04:00
parent b1134d5d6e
commit 57a7ab944b
Signed by: cy
SSH key fingerprint: SHA256:o/geVWV4om1QhUSkKvDQeW/eAihwnjyXkqMwrVdbuts
3 changed files with 17 additions and 18 deletions

View file

@ -2,7 +2,7 @@ use std::{
fs,
iter::once,
sync::{
Arc, Mutex,
Arc,
atomic::{AtomicUsize, Ordering},
},
};
@ -13,7 +13,7 @@ use aws_sdk_s3 as s3;
use futures::future::join_all;
use nix_compat::narinfo::{self, SigningKey};
use tokio::sync::{RwLock, Semaphore, mpsc};
use tracing::{debug, info, trace, warn};
use tracing::{debug, info, trace};
use url::Url;
use crate::{Cli, path_info::PathInfo, uploader::Uploader};
@ -132,7 +132,6 @@ impl NixCp {
async fn upload(&'static self, mut rx: mpsc::Receiver<PathInfo>) -> Result<()> {
let upload_count = AtomicUsize::new(0);
let failures: Arc<Mutex<Vec<String>>> = Arc::new(Mutex::new(Vec::new()));
let permits = Arc::new(Semaphore::new(10));
let mut uploads = Vec::with_capacity(10);
@ -155,7 +154,12 @@ impl NixCp {
});
uploads.push(fut);
} else {
join_all(uploads).await;
join_all(uploads)
.await
.into_iter()
.flatten()
.collect::<Result<Vec<_>>>()?;
println!("uploaded: {}", upload_count.load(Ordering::Relaxed));
println!(
"skipped because of signature match: {}",
@ -165,14 +169,6 @@ impl NixCp {
"skipped because of upstream hit: {}",
self.upstream_hit_count.load(Ordering::Relaxed)
);
let failures = failures.lock().unwrap();
if !failures.is_empty() {
warn!("failed to upload these paths: ");
for failure in failures.iter() {
warn!("{}", failure);
}
}
break;
}
}