redirectionio_actix_proxy/
error.rs

1use crate::forwarder::ForwardError;
2use actix_http::StatusCode;
3use actix_web::{Error as ActixError, ResponseError};
4use std::error::Error;
5use std::fmt;
6
7#[derive(Debug)]
8pub enum ProxyError {
9    CannotReadRequestBody(ActixError),
10    ForwardError(ForwardError),
11    NoPeer,
12}
13
14impl fmt::Display for ProxyError {
15    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
16        match self {
17            Self::CannotReadRequestBody(e) => write!(f, "error when reading request body: {}", e),
18            Self::ForwardError(e) => write!(f, "error when forwarding request: {}", e),
19            Self::NoPeer => f.write_str("no peer found"),
20        }
21    }
22}
23
24impl Error for ProxyError {
25    fn source(&self) -> Option<&(dyn Error + 'static)> {
26        match self {
27            Self::CannotReadRequestBody(err) => Some(err),
28            Self::ForwardError(err) => Some(err),
29            Self::NoPeer => None,
30        }
31    }
32}
33
34impl ResponseError for ProxyError {
35    fn status_code(&self) -> StatusCode {
36        match self {
37            Self::CannotReadRequestBody(err) => err.as_response_error().status_code(),
38            Self::ForwardError(err) => err.status_code(),
39            Self::NoPeer => StatusCode::NOT_FOUND,
40        }
41    }
42}