NyxErinys/crunchy-cli
1
0
Fork 0
mirror of https://github.com/crunchy-labs/crunchy-cli.git synced 2026-01-21 12:12:00 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add error handling and retry attempts

Browse source 
Handles cases where the segments fail to download and sometimes get stuck by introducing a timeout and retrying on failure.
This commit is contained in:
Alexandru Dracea 2022-12-28 15:44:45 +02:00 • committed by GitHub
parent c5940a240c
commit 240e5563a3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 49 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

51
crunchy-cli-core/src/cli/utils.rs
View file

@ -76,8 +76,55 @@ pub async fn download_segments(
let thread_count = count.clone();
join_set.spawn(async move {
for (i, segment) in thread_segments.into_iter().enumerate() {
let response = thread_client.get(&segment.url).send().await?;
let mut buf = response.bytes().await?.to_vec();
let response_res = thread_client
.get(&segment.url)
.timeout(Duration::from_secs(60u64))
.send()
.await;
let verfified_response = match response_res {
Ok(x) => x,
Err(y) => panic!("This is likely a netowrking error: {}", y),
};
let possible_error_in_response = verfified_response.bytes().await;
let mut buf = if let Ok(r) = possible_error_in_response {
r.to_vec()
} else {
debug!(
"Segment Failed to download: {}, retrying.",
num + (i * cpus)
);
let mut resp = thread_client
.get(&segment.url)
.timeout(Duration::from_secs(60u64))
.send()
.await
.unwrap()
.bytes()
.await;
if resp.is_err() {
let mut retry_ctr = 1;
loop {
debug!(
"Segment Failed to download: {}, retry {}.",
num + (i * cpus),
retry_ctr
);
resp = thread_client
.get(&segment.url)
.timeout(Duration::from_secs(60u64))
.send()
.await
.unwrap()
.bytes()
.await;
if resp.is_ok() {
break;
}
retry_ctr += 1;
}
}
resp.unwrap().to_vec()
};
buf = VariantSegment::decrypt(buf.borrow_mut(), segment.key)?.to_vec();
debug!(