Hopiu/lychee
1
0
Fork 0
mirror of https://github.com/Hopiu/lychee.git synced 2026-03-17 05:00:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Extract function and add SAFETY note

Browse source
This commit is contained in:
Thomas Zahner 2024-04-11 11:33:38 +02:00
parent a3184190b8
commit d5b9b84db6
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
lychee-lib/src/checker.rs
View file

@ -37,7 +37,7 @@ impl Checker {
let mut retries: u64 = 0;
let mut wait_time = self.retry_wait_time;
let mut status = self.check_default(request.try_clone().unwrap()).await; // TODO: try_clone
let mut status = self.check_default(clone_unwrap(&request)).await;
while retries < self.max_retries {
if status.is_success() || !status.should_retry() {
return status;
@ -45,7 +45,7 @@ impl Checker {
retries += 1;
tokio::time::sleep(wait_time).await;
wait_time = wait_time.saturating_mul(2);
status = self.check_default(request.try_clone().unwrap()).await; // TODO: try_clone
status = self.check_default(clone_unwrap(&request)).await;
}
status
}
@ -59,6 +59,12 @@ impl Checker {
}
}
/// SAFETY: unwrapping the `try_clone` of `reqwest::Request` is safe because a request only fails to be cloned when `body` of `Request` is a stream
/// and `body` cannot be a stream as long as the `stream` feature is disabled.
fn clone_unwrap(request: &Request) -> Request {
request.try_clone().unwrap()
}
#[async_trait]
impl Chainable<Request, Status> for Checker {
async fn chain(&mut self, input: Request) -> ChainResult<Request, Status> {