From 3167a55853e047410c8e74c02e8dfd90e5966dc6 Mon Sep 17 00:00:00 2001
From: Mark Lodato <mlodato517@gmail.com>
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 @@
 
 // <error-handler>
 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<B>(mut res: dev::ServiceResponse<B>) -> Result<ErrorHandlerResponse<B>> {
+fn add_error_header<B>(mut res: dev::ServiceResponse<B>) -> Result<ErrorHandlerResponse<B>> {
     res.response_mut().headers_mut().insert(
         http::header::CONTENT_TYPE,
         http::HeaderValue::from_static("Error"),
@@ -14,18 +14,15 @@ fn render_500<B>(mut res: dev::ServiceResponse<B>) -> Result<ErrorHandlerRespons
 
 #[actix_web::main]
 async fn main() -> 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")?