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")?