From 3167a55853e047410c8e74c02e8dfd90e5966dc6 Mon Sep 17 00:00:00 2001 From: Mark Lodato Date: Wed, 22 Dec 2021 13:36:35 -0500 Subject: [PATCH] Improve error handler example (#248) --- examples/middleware/src/errorhandler.rs | 19 ++++++++----------- 1 file changed, 8 insertions(+), 11 deletions(-) diff --git a/examples/middleware/src/errorhandler.rs b/examples/middleware/src/errorhandler.rs index 3a510e2..1e83033 100644 --- a/examples/middleware/src/errorhandler.rs +++ b/examples/middleware/src/errorhandler.rs @@ -2,9 +2,9 @@ // use actix_web::middleware::errhandlers::{ErrorHandlerResponse, ErrorHandlers}; -use actix_web::{dev, http, HttpResponse, Result}; +use actix_web::{dev, http, web, App, HttpResponse, HttpServer, Result}; -fn render_500(mut res: dev::ServiceResponse) -> Result> { +fn add_error_header(mut res: dev::ServiceResponse) -> Result> { res.response_mut().headers_mut().insert( http::header::CONTENT_TYPE, http::HeaderValue::from_static("Error"), @@ -14,18 +14,15 @@ fn render_500(mut res: dev::ServiceResponse) -> Result std::io::Result<()> { - use actix_web::{web, App, HttpServer}; - HttpServer::new(|| { App::new() - .wrap( - ErrorHandlers::new() - .handler(http::StatusCode::INTERNAL_SERVER_ERROR, render_500), - ) + .wrap(ErrorHandlers::new().handler( + http::StatusCode::INTERNAL_SERVER_ERROR, + add_error_header, + )) .service( - web::resource("/test") - .route(web::get().to(|| HttpResponse::Ok())) - .route(web::head().to(|| HttpResponse::MethodNotAllowed())), + web::resource("/") + .route(web::get().to(HttpResponse::InternalServerError)), ) }) .bind("127.0.0.1:8080")?