use std::{fmt, time::Duration};
use http::StatusCode;
use http_body_util::BodyExt;
use hyper::body::Incoming;
pub type Result<T> = std::result::Result<T, Error>;
#[derive(Debug, Clone)]
pub enum Error {
Generic(String),
Timeout {
elapsed: Option<Duration>,
timeout: Option<Duration>,
},
Http(HttpErrorContent<crate::models::HttpErrorOut>),
Validation(HttpErrorContent<crate::models::HttpValidationError>),
}
impl Error {
pub(crate) fn generic(err: impl std::error::Error) -> Self {
Self::Generic(format!("{err:?}"))
}
pub(crate) async fn from_response(
status_code: StatusCode,
body: Incoming,
timeout: Option<Duration>,
) -> Self {
match body.collect().await {
Ok(collected) => {
let bytes = collected.to_bytes();
if status_code == StatusCode::UNPROCESSABLE_ENTITY {
Self::Validation(HttpErrorContent {
status: StatusCode::UNPROCESSABLE_ENTITY,
payload: serde_json::from_slice(&bytes).ok(),
})
} else {
Error::Http(HttpErrorContent {
status: status_code,
payload: serde_json::from_slice(&bytes).ok(),
})
}
}
Err(e) if e.is_timeout() => Self::Timeout {
elapsed: None,
timeout,
},
Err(e) => Self::Generic(e.to_string()),
}
}
}
impl fmt::Display for Error {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match self {
Error::Generic(s) => s.fmt(f),
Self::Timeout {
elapsed: Some(elapsed),
timeout: Some(timeout),
} => format!("Request timeout after {elapsed:?} (timeout {timeout:?})").fmt(f),
Self::Timeout {
timeout: Some(timeout),
..
} => format!("Request timeout (threshold {timeout:?})").fmt(f),
Self::Timeout { .. } => "Request timeout".fmt(f),
Error::Http(e) => format!("Http error (status={}) {:?}", e.status, e.payload).fmt(f),
Error::Validation(e) => format!("Validation error {:?}", e.payload).fmt(f),
}
}
}
impl std::error::Error for Error {}
#[derive(Debug, Clone)]
pub struct HttpErrorContent<T> {
pub status: StatusCode,
pub payload: Option<T>,
}