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

62 lines
1.4 KiB
Rust
Raw Normal View History

2019-06-17 19:46:21 +02:00
// <override>
2020-09-12 17:21:54 +02:00
use actix_web::{
2022-02-26 04:56:24 +01:00
error, get,
http::{header::ContentType, StatusCode},
App, HttpResponse,
2020-09-12 17:21:54 +02:00
};
use derive_more::derive::{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 MyError {
#[display("internal error")]
2019-06-17 19:46:21 +02:00
InternalError,
2020-09-12 17:21:54 +02:00
#[display("bad request")]
2019-06-17 19:46:21 +02:00
BadClientData,
2020-09-12 17:21:54 +02:00
#[display("timeout")]
2019-06-17 19:46:21 +02:00
Timeout,
}
impl error::ResponseError for MyError {
fn error_response(&self) -> HttpResponse {
2022-02-26 04:56:24 +01:00
HttpResponse::build(self.status_code())
.insert_header(ContentType::html())
.body(self.to_string())
}
fn status_code(&self) -> StatusCode {
2019-06-17 19:46:21 +02:00
match *self {
MyError::InternalError => StatusCode::INTERNAL_SERVER_ERROR,
MyError::BadClientData => StatusCode::BAD_REQUEST,
MyError::Timeout => StatusCode::GATEWAY_TIMEOUT,
2019-06-17 19:46:21 +02:00
}
}
}
2020-09-12 17:21:54 +02:00
#[get("/")]
async fn index() -> Result<&'static str, MyError> {
2019-06-17 19:46:21 +02:00
Err(MyError::BadClientData)
}
// </override>
2020-09-12 17:21:54 +02:00
#[get("/e2")]
async fn error2() -> Result<&'static str, MyError> {
Err(MyError::InternalError)
}
2020-09-12 17:21:54 +02:00
#[get("/e3")]
async fn error3() -> Result<&'static str, MyError> {
Err(MyError::Timeout)
}
2019-06-25 03:18:30 +02:00
2020-09-12 17:21:54 +02:00
#[actix_web::main]
async fn main() -> std::io::Result<()> {
2019-06-25 03:18:30 +02:00
use actix_web::HttpServer;
2020-09-12 17:21:54 +02:00
HttpServer::new(|| App::new().service(index).service(error2).service(error3))
2022-02-26 04:56:24 +01:00
.bind(("127.0.0.1", 8080))?
2020-09-12 17:21:54 +02:00
.run()
.await
2019-06-25 03:18:30 +02:00
}