ExternalVendorCode/onefuzz
1
0
Fork 0
mirror of https://github.com/microsoft/onefuzz.git synced 2025-06-12 10:08:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

add retry to download_input (#676)

Browse Source
This commit is contained in:
bmc-msft
2021-03-16 12:17:28 -04:00
committed by GitHub
parent 7ebdeac537
commit f0e5823c0a
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/agent/onefuzz-agent/src/tasks/utils.rs
View File

@ -4,7 +4,8 @@
use anyhow::Result; use anyhow::Result;
use async_trait::async_trait; use async_trait::async_trait;
use onefuzz::{http::ResponseExt, jitter::delay_with_jitter}; use onefuzz::{http::ResponseExt, jitter::delay_with_jitter};
use reqwest::Url; use reqwest::{Client, Url};
use reqwest_retry::SendRetry;
use std::path::{Path, PathBuf}; use std::path::{Path, PathBuf};
use std::time::Duration; use std::time::Duration;
use tokio::{fs, io}; use tokio::{fs, io};
@ -13,7 +14,9 @@ pub async fn download_input(input_url: Url, dst: impl AsRef<Path>) -> Result<Pat
let file_name = input_url.path_segments().unwrap().last().unwrap(); let file_name = input_url.path_segments().unwrap().last().unwrap();
let file_path = dst.as_ref().join(file_name); let file_path = dst.as_ref().join(file_name);
let resp = reqwest::get(input_url) let resp = Client::new()
.get(input_url)
.send_retry_default()
.await? .await?
.error_for_status_with_body() .error_for_status_with_body()
.await?; .await?;