1
0
mirror of https://github.com/actix/actix-website synced 2025-03-11 02:42:43 +01:00
actix-website/examples/errors/src/override_error.rs

60 lines
1.5 KiB
Rust
Raw Normal View History

use actix_web::{web, App};
2019-06-17 13:46:21 -04:00
// <override>
use actix_http::ResponseBuilder;
use actix_web::{error, http::header, http::StatusCode, HttpResponse};
2019-06-17 13:46:21 -04:00
use failure::Fail;
#[derive(Fail, Debug)]
enum MyError {
#[fail(display = "internal error")]
InternalError,
#[fail(display = "bad request")]
BadClientData,
#[fail(display = "timeout")]
Timeout,
}
impl error::ResponseError for MyError {
fn error_response(&self) -> HttpResponse {
ResponseBuilder::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 13:46:21 -04:00
match *self {
MyError::InternalError => StatusCode::INTERNAL_SERVER_ERROR,
MyError::BadClientData => StatusCode::BAD_REQUEST,
MyError::Timeout => StatusCode::GATEWAY_TIMEOUT,
2019-06-17 13:46:21 -04:00
}
}
}
async fn index() -> Result<&'static str, MyError> {
2019-06-17 13:46:21 -04:00
Err(MyError::BadClientData)
}
// </override>
async fn error2() -> Result<&'static str, MyError> {
Err(MyError::InternalError)
}
async fn error3() -> Result<&'static str, MyError> {
Err(MyError::Timeout)
}
2019-06-24 21:18:30 -04:00
#[actix_rt::main]
async fn main() -> std::io::Result<()> {
2019-06-24 21:18:30 -04:00
use actix_web::HttpServer;
HttpServer::new(|| {
App::new()
.route("/", web::get().to(index))
.route("/e2", web::get().to(error2))
.route("/e3", web::get().to(error3))
})
.bind("127.0.0.1:8088")?
2019-06-24 21:18:30 -04:00
.run()
.await
2019-06-24 21:18:30 -04:00
}