OpenSource/axum
OpenSource/axum
1
0
Fork 0
mirror of https://github.com/tokio-rs/axum.git synced 2025-01-01 00:50:32 +01:00
Code Issues Projects Releases Packages Wiki Activity

update to matchit 0.5 (#843)

Browse source
This commit is contained in:
Ibraheem Ahmed 2022-03-09 03:28:19 -05:00 committed by GitHub
parent 19eda2f566
commit 04dd7617a4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 16 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
axum/Cargo.toml
View file

@ -33,7 +33,7 @@ http = "0.2.5"
http-body = "0.4.4"
hyper = { version = "0.14.14", features = ["server", "tcp", "stream"] }
itoa = "1.0.1"
matchit = "0.4.6"
matchit = "0.5.0"
memchr = "2.4.1"
mime = "0.3.16"
percent-encoding = "2.1"

34
axum/src/routing/mod.rs
View file

@ -10,6 +10,7 @@ use crate::{
BoxError,
};
use http::{Request, Uri};
use matchit::MatchError;
use std::{
borrow::Cow,
collections::HashMap,
@ -502,22 +503,17 @@ where
match self.node.at(&path) {
Ok(match_) => self.call_route(match_, req),
Err(err) => {
if err.tsr() {
let redirect_to = if let Some(without_tsr) = path.strip_suffix('/') {
with_path(req.uri(), without_tsr)
} else {
with_path(req.uri(), &format!("{}/", path))
};
let res = Redirect::permanent(redirect_to);
RouteFuture::from_response(res.into_response())
} else {
match &self.fallback {
Fallback::Default(inner) => inner.clone().call(req),
Fallback::Custom(inner) => inner.clone().call(req),
}
}
}
Err(MatchError::MissingTrailingSlash) => RouteFuture::from_response(
Redirect::permanent(with_path(req.uri(), &format!("{}/", path))).into_response(),
),
Err(MatchError::ExtraTrailingSlash) => RouteFuture::from_response(
Redirect::permanent(with_path(req.uri(), path.strip_suffix('/').unwrap()))
.into_response(),
),
Err(MatchError::NotFound) => match &self.fallback {
Fallback::Default(inner) => inner.clone().call(req),
Fallback::Custom(inner) => inner.clone().call(req),
},
}
}
}
@ -551,10 +547,10 @@ fn with_path(uri: &Uri, new_path: &str) -> Uri {
Uri::from_parts(parts).unwrap()
}
/// Wrapper around `matchit::Node` that supports merging two `Node`s.
/// Wrapper around `matchit::Router` that supports merging two `Router`s.
#[derive(Clone, Default)]
struct Node {
inner: matchit::Node<RouteId>,
inner: matchit::Router<RouteId>,
route_id_to_path: HashMap<RouteId, Arc<str>>,
path_to_route_id: HashMap<Arc<str>, RouteId>,
}
@ -579,7 +575,7 @@ impl Node {
fn at<'n, 'p>(
&'n self,
path: &'p str,
) -> Result<matchit::Match<'n, 'p, &'n RouteId>, matchit::MatchError> {
) -> Result<matchit::Match<'n, 'p, &'n RouteId>, MatchError> {
self.inner.at(path)
}
}