OpenSource/axum
OpenSource/axum
1
0
Fork 0
mirror of https://github.com/tokio-rs/axum.git synced 2024-11-23 07:39:25 +01:00
Code Issues Projects Releases Packages Wiki Activity

Avoid cloning the state in layer

Browse source
This commit is contained in:
novacrazy 2024-10-10 11:35:04 +02:00 committed by Jonas Platte
parent 296dfe1a40
commit c417a28142
No known key found for this signature in database
GPG key ID: 7D261D771D915378
3 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
axum/src/routing/path_router.rs
View file

@ -357,7 +357,7 @@ where
Endpoint::MethodRouter(method_router) => {
Ok(method_router.call_with_state(req, state))
}
Endpoint::Route(route) => Ok(route.clone().call(req)),
Endpoint::Route(route) => Ok(route.clone().call_owned(req)),
}
}
// explicitly handle all variants in case matchit adds

6
axum/src/routing/route.rs
View file

@ -42,6 +42,12 @@ impl<E> Route<E> {
)))
}
/// Variant of [`Route::call`] that takes ownership of the route to avoid cloning.
pub(crate) fn call_owned(self, req: Request<Body>) -> RouteFuture<E> {
let req = req.map(Body::new);
RouteFuture::from_future(self.oneshot_inner_owned(req))
}
pub(crate) fn oneshot_inner(
&mut self,
req: Request,

2
axum/src/routing/tests/mod.rs
View file

@ -940,7 +940,7 @@ async fn state_isnt_cloned_too_much_in_layer() {
client.get("/").await;
assert_eq!(state.count(), 4);
assert_eq!(state.count(), 3);
}
#[crate::test]