Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Improve build times by generating less IR #1192

Merged
merged 12 commits into from Jul 25, 2022
Merged

Improve build times by generating less IR #1192

Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
03f2b88
example
davidpdrsn Jul 25, 2022
d70eda5
`MethodRouter::merge`
davidpdrsn Jul 25, 2022
afd3356
`set_content_length` and `set_allow_header`
davidpdrsn Jul 25, 2022
0c80a8b
`MethodRouter::on_service_boxed_response_body`
davidpdrsn Jul 25, 2022
c4fdb10
`Router::route`
davidpdrsn Jul 25, 2022
f1a9a21
`MethodRouter::merge` again
davidpdrsn Jul 25, 2022
4fed037
`MethodRouter::on_service_boxed_response_body`
davidpdrsn Jul 25, 2022
985726b
`Router::call_route`
davidpdrsn Jul 25, 2022
2f9cd06
`MethodRouter::{layer, route_layer}`
davidpdrsn Jul 25, 2022
bbdba7d
revert example
davidpdrsn Jul 25, 2022
f4c3f57
fix test
davidpdrsn Jul 25, 2022
1bd680f
move function to method on `AllowHeader`
davidpdrsn Jul 25, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
138 changes: 101 additions & 37 deletions axum/src/routing/method_routing.rs
View file
Open in desktop
Expand Up @@ -899,29 +899,26 @@ impl<ReqBody, E> MethodRouter<ReqBody, E> {
S::Response: IntoResponse + 'static,
S::Future: Send + 'static,
{
macro_rules! set_service {
(
$filter:ident,
$svc:ident,
$allow_header:ident,
[
$(
($out:ident, $variant:ident, [$($method:literal),+])
),+
$(,)?
]
) => {
$(
if $filter.contains(MethodFilter::$variant) {
if $out.is_some() {
panic!("Overlapping method route. Cannot add two method routes that both handle `{}`", stringify!($variant))
}
$out = $svc.clone();
$(
append_allow_header(&mut $allow_header, $method);
)+
}
)+
// written using an inner function to generate less IR
fn set_service<T>(
method_name: &str,
out: &mut Option<T>,
svc: &T,
svc_filter: MethodFilter,
filter: MethodFilter,
allow_header: &mut AllowHeader,
methods: &[&'static str],
) where
T: Clone,
{
if svc_filter.contains(filter) {
if out.is_some() {
panic!("Overlapping method route. Cannot add two method routes that both handle `{}`", method_name)
}
*out = Some(svc.clone());
for method in methods {
append_allow_header(allow_header, method);
}
}
}

Expand All @@ -939,22 +936,89 @@ impl<ReqBody, E> MethodRouter<ReqBody, E> {
mut allow_header,
_request_body: _,
} = self;
let svc = Some(Route::new(svc));
set_service!(

let svc = Route::new(svc);

set_service(
davidpdrsn marked this conversation as resolved.
Show resolved Hide resolved
"GET",
&mut get,
&svc,
filter,
svc,
allow_header,
[
(get, GET, ["GET", "HEAD"]),
(head, HEAD, ["HEAD"]),
(delete, DELETE, ["DELETE"]),
(options, OPTIONS, ["OPTIONS"]),
(patch, PATCH, ["PATCH"]),
(post, POST, ["POST"]),
(put, PUT, ["PUT"]),
(trace, TRACE, ["TRACE"]),
]
MethodFilter::GET,
&mut allow_header,
&["GET", "HEAD"],
);

set_service(
"HEAD",
&mut head,
&svc,
filter,
MethodFilter::HEAD,
&mut allow_header,
&["HEAD"],
);

set_service(
"TRACE",
&mut trace,
&svc,
filter,
MethodFilter::TRACE,
&mut allow_header,
&["TRACE"],
);

set_service(
"PUT",
&mut put,
&svc,
filter,
MethodFilter::PUT,
&mut allow_header,
&["PUT"],
);

set_service(
"POST",
&mut post,
&svc,
filter,
MethodFilter::POST,
&mut allow_header,
&["POST"],
);

set_service(
"PATCH",
&mut patch,
&svc,
filter,
MethodFilter::PATCH,
&mut allow_header,
&["PATCH"],
);

set_service(
"OPTIONS",
&mut options,
&svc,
filter,
MethodFilter::OPTIONS,
&mut allow_header,
&["OPTIONS"],
);

set_service(
"DELETE",
&mut delete,
&svc,
filter,
MethodFilter::DELETE,
&mut allow_header,
&["DELETE"],
);

Self {
get,
head,
Expand Down