crates/utils/src/request.rs

   1 use std::future::Future;
   2
   3 #[tracing::instrument(skip_all)]
   4 pub async fn retry<F, Fut, T>(f: F) -> Result<T, reqwest_middleware::Error>
   5 where
   6   F: Fn() -> Fut,
   7   Fut: Future<Output = Result<T, reqwest_middleware::Error>>,
   8 {
   9   retry_custom(|| async { Ok((f)().await) }).await
  10 }
  11
  12 #[tracing::instrument(skip_all)]
  13 async fn retry_custom<F, Fut, T>(f: F) -> Result<T, reqwest_middleware::Error>
  14 where
  15   F: Fn() -> Fut,
  16   Fut: Future<Output = Result<Result<T, reqwest_middleware::Error>, reqwest_middleware::Error>>,
  17 {
  18   let mut response: Option<Result<T, reqwest_middleware::Error>> = None;
  19
  20   for _ in 0u8..3 {
  21     match (f)().await? {
  22       Ok(t) => return Ok(t),
  23       Err(reqwest_middleware::Error::Reqwest(e)) => {
  24         if e.is_timeout() {
  25           response = Some(Err(reqwest_middleware::Error::Reqwest(e)));
  26           continue;
  27         }
  28         return Err(reqwest_middleware::Error::Reqwest(e));
  29       }
  30       Err(otherwise) => {
  31         return Err(otherwise);
  32       }
  33     }
  34   }
  35
  36   response.expect("retry http request")
  37 }