1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
use http::{header, Request, Response};
use output::ResponseBody;
use super::{CritError, HttpError};
pub trait ErrorHandler {
fn handle_error(&self, err: &dyn HttpError, request: &Request<()>) -> Result<Response<ResponseBody>, CritError>;
}
impl<F, T> ErrorHandler for F
where
F: Fn(&dyn HttpError, &Request<()>) -> Result<Response<T>, CritError>,
T: Into<ResponseBody>,
{
fn handle_error(&self, err: &dyn HttpError, request: &Request<()>) -> Result<Response<ResponseBody>, CritError> {
(*self)(err, request).map(|res| res.map(Into::into))
}
}
#[derive(Debug, Default)]
pub struct DefaultErrorHandler {
_priv: (),
}
impl DefaultErrorHandler {
pub fn new() -> DefaultErrorHandler {
Default::default()
}
}
impl ErrorHandler for DefaultErrorHandler {
fn handle_error(&self, err: &dyn HttpError, _: &Request<()>) -> Result<Response<ResponseBody>, CritError> {
Response::builder()
.status(err.status_code())
.header(header::CONNECTION, "close")
.header(header::CACHE_CONTROL, "no-cache")
.body(err.to_string().into())
.map_err(|e| {
format_err!("failed to construct an HTTP error response: {}", e)
.compat()
.into()
})
}
}