1
0
mirror of https://github.com/actix/actix-website synced 2024-12-18 09:53:12 +01:00
actix-website/examples/errors/src/recommend_two.rs

45 lines
1.2 KiB
Rust
Raw Normal View History

2019-06-17 19:46:21 +02:00
// <recommend-two>
2020-09-12 17:21:54 +02:00
use actix_web::{
dev::HttpResponseBuilder, error, get, http::header, http::StatusCode, App, HttpResponse,
HttpServer,
};
use derive_more::{Display, Error};
2019-06-17 19:46:21 +02:00
2020-09-12 17:21:54 +02:00
#[derive(Debug, Display, Error)]
2019-06-17 19:46:21 +02:00
enum UserError {
2020-09-12 17:21:54 +02:00
#[display(fmt = "An internal error occurred. Please try again later.")]
2019-06-17 19:46:21 +02:00
InternalError,
}
impl error::ResponseError for UserError {
fn error_response(&self) -> HttpResponse {
2020-09-12 17:21:54 +02:00
HttpResponseBuilder::new(self.status_code())
.set_header(header::CONTENT_TYPE, "text/html; charset=utf-8")
.body(self.to_string())
}
fn status_code(&self) -> StatusCode {
2019-06-17 19:46:21 +02:00
match *self {
UserError::InternalError => StatusCode::INTERNAL_SERVER_ERROR,
2019-06-17 19:46:21 +02:00
}
}
}
2020-09-12 17:21:54 +02:00
#[get("/")]
async fn index() -> Result<&'static str, UserError> {
2020-09-12 17:21:54 +02:00
do_thing_that_fails().map_err(|_e| UserError::InternalError)?;
2019-06-17 19:46:21 +02:00
Ok("success!")
}
// </recommend-two>
2019-06-25 03:18:30 +02:00
2020-09-12 17:21:54 +02:00
fn do_thing_that_fails() -> Result<(), std::io::Error> {
2019-06-25 03:18:30 +02:00
Err(std::io::Error::new(std::io::ErrorKind::Other, "some error"))
}
2020-09-12 17:21:54 +02:00
#[actix_web::main]
async fn main() -> std::io::Result<()> {
2020-09-12 17:21:54 +02:00
HttpServer::new(|| App::new().service(index))
.bind("127.0.0.1:8080")?
2019-06-25 03:18:30 +02:00
.run()
.await
}