actix/examples
1
0
Fork 0
mirror of https://github.com/actix/examples synced 2025-06-29 02:10:36 +02:00
Code Issues Releases Wiki Activity

Demonstrate how to render error pages using templates.

Browse Source
This commit is contained in:
Pieter Frenssen
2020-05-19 18:21:39 +03:00
parent bc6f614f78
commit 986614d698
8 changed files with 128 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
template_tera/src/main.rs
View File

@ -1,6 +1,10 @@
use std::collections::HashMap;
use actix_web::{error, middleware, web, App, Error, HttpResponse, HttpServer};
use actix_http::{body::Body, Response};
use actix_web::dev::ServiceResponse;
use actix_web::http::StatusCode;
use actix_web::middleware::errhandlers::{ErrorHandlerResponse, ErrorHandlers};
use actix_web::{error, middleware, web, App, Error, HttpResponse, HttpServer, Result};
use tera::Tera;
// store tera template in application state
@ -35,8 +39,53 @@ async fn main() -> std::io::Result<()> {
.data(tera)
.wrap(middleware::Logger::default()) // enable logger
.service(web::resource("/").route(web::get().to(index)))
.service(web::scope("").wrap(error_handlers()))
})
.bind("127.0.0.1:8080")?
.run()
.await
}
// Custom error handlers, to return HTML responses when an error occurs.
fn error_handlers() -> ErrorHandlers<Body> {
ErrorHandlers::new().handler(StatusCode::NOT_FOUND, not_found)
}
// Error handler for a 404 Page not found error.
fn not_found<B>(res: ServiceResponse<B>) -> Result<ErrorHandlerResponse<B>> {
let response = get_error_response(&res, "Page not found");
Ok(ErrorHandlerResponse::Response(
res.into_response(response.into_body()),
))
}
// Generic error handler.
fn get_error_response<B>(res: &ServiceResponse<B>, error: &str) -> Response<Body> {
let request = res.request();
// Provide a fallback to a simple plain text response in case an error occurs during the
// rendering of the error page.
let fallback = |e: &str| {
Response::build(res.status())
.content_type("text/plain")
.body(e.to_string())
};
let tera = request.app_data::<web::Data<Tera>>().map(|t| t.get_ref());
match tera {
Some(tera) => {
let mut context = tera::Context::new();
context.insert("error", error);
context.insert("status_code", res.status().as_str());
let body = tera.render("error.html", &context);
match body {
Ok(body) => Response::build(res.status())
.content_type("text/html")
.body(body),
Err(_) => fallback(error),
}
}
None => fallback(error),
}
}