mirror of
https://github.com/LemmyNet/lemmy.git
synced 2024-06-10 01:19:21 +00:00
442369a041
Adjust dockerfiles, fix cargo.toml and remove unused deps Merge branch 'main' into move-websocket-to-workspace Move api code into workspace Move apub to separate workspace Move websocket code into separate workspace Some code cleanup Remove websocket dependency on API Co-authored-by: Felix Ableitner <me@nutomic.com> Reviewed-on: https://yerbamate.dev/LemmyNet/lemmy/pulls/107
44 lines
1,009 B
Rust
44 lines
1,009 B
Rust
use crate::LemmyError;
|
|
use anyhow::anyhow;
|
|
use std::future::Future;
|
|
use thiserror::Error;
|
|
|
|
#[derive(Clone, Debug, Error)]
|
|
#[error("Error sending request, {0}")]
|
|
struct SendError(pub String);
|
|
|
|
#[derive(Clone, Debug, Error)]
|
|
#[error("Error receiving response, {0}")]
|
|
pub struct RecvError(pub String);
|
|
|
|
pub async fn retry<F, Fut, T>(f: F) -> Result<T, LemmyError>
|
|
where
|
|
F: Fn() -> Fut,
|
|
Fut: Future<Output = Result<T, reqwest::Error>>,
|
|
{
|
|
retry_custom(|| async { Ok((f)().await) }).await
|
|
}
|
|
|
|
async fn retry_custom<F, Fut, T>(f: F) -> Result<T, LemmyError>
|
|
where
|
|
F: Fn() -> Fut,
|
|
Fut: Future<Output = Result<Result<T, reqwest::Error>, LemmyError>>,
|
|
{
|
|
let mut response = Err(anyhow!("connect timeout").into());
|
|
|
|
for _ in 0u8..3 {
|
|
match (f)().await? {
|
|
Ok(t) => return Ok(t),
|
|
Err(e) => {
|
|
if e.is_timeout() {
|
|
response = Err(SendError(e.to_string()).into());
|
|
continue;
|
|
}
|
|
return Err(SendError(e.to_string()).into());
|
|
}
|
|
}
|
|
}
|
|
|
|
response
|
|
}
|