mirror of
https://github.com/tokio-rs/axum.git
synced 2024-12-29 15:49:16 +01:00
2be79168d8
If a route existed but had no handler for the method, the code used in the 404 example wouldn't catch it.
52 lines
1.3 KiB
Rust
52 lines
1.3 KiB
Rust
//! Run with
|
|
//!
|
|
//! ```not_rust
|
|
//! cargo run --example 404
|
|
//! ```
|
|
|
|
use axum::{
|
|
body::{box_body, Body, BoxBody},
|
|
prelude::*,
|
|
};
|
|
use http::{Response, StatusCode};
|
|
use std::net::SocketAddr;
|
|
use tower::util::MapResponseLayer;
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
// Set the RUST_LOG, if it hasn't been explicitly defined
|
|
if std::env::var("RUST_LOG").is_err() {
|
|
std::env::set_var("RUST_LOG", "404=debug")
|
|
}
|
|
tracing_subscriber::fmt::init();
|
|
|
|
// build our application with a route
|
|
let app = route("/", get(handler))
|
|
// make sure this is added as the very last thing
|
|
.layer(MapResponseLayer::new(map_404));
|
|
|
|
// run it
|
|
let addr = SocketAddr::from(([127, 0, 0, 1], 3000));
|
|
tracing::debug!("listening on {}", addr);
|
|
axum::Server::bind(&addr)
|
|
.serve(app.into_make_service())
|
|
.await
|
|
.unwrap();
|
|
}
|
|
|
|
async fn handler() -> response::Html<&'static str> {
|
|
response::Html("<h1>Hello, World!</h1>")
|
|
}
|
|
|
|
fn map_404(response: Response<BoxBody>) -> Response<BoxBody> {
|
|
if response.status() == StatusCode::NOT_FOUND
|
|
|| response.status() == StatusCode::METHOD_NOT_ALLOWED
|
|
{
|
|
return Response::builder()
|
|
.status(StatusCode::NOT_FOUND)
|
|
.body(box_body(Body::from("nothing to see here")))
|
|
.unwrap();
|
|
}
|
|
|
|
response
|
|
}
|