tokio-rs/axum
1
0
Fork 0
mirror of https://github.com/tokio-rs/axum.git synced 2026-08-24 00:00:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix debug_handler for Result<impl IntoResponse, Error> and friends (#588)

Browse Source 
Fixes #587
This commit is contained in:
David Pedersen
2021-12-04 18:24:58 +01:00
committed by GitHub
parent 628921bcad
commit cbb34869d8
3 changed files with 165 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
axum-debug/src/lib.rs
+30 -10
View File
@@ -251,6 +251,24 @@ mod debug_handler {
};
let span = ty.span();
let declare_inputs = item_fn
.sig
.inputs
.iter()
.filter_map(|arg| match arg {
FnArg::Receiver(_) => None,
FnArg::Typed(pat_ty) => {
let pat = &pat_ty.pat;
let ty = &pat_ty.ty;
Some(quote! {
let #pat: #ty = panic!();
})
}
})
.collect::<TokenStream>();
let block = &item_fn.block;
let make_value_name = format_ident!(
"__axum_debug_check_{}_into_response_make_value",
item_fn.sig.ident
@@ -259,18 +277,18 @@ mod debug_handler {
let make = if item_fn.sig.asyncness.is_some() {
quote_spanned! {span=>
#[allow(warnings)]
async fn #make_value_name() -> #ty { panic!() }
}
} else if let syn::Type::ImplTrait(_) = &**ty {
// lets just assume it returns `impl Future`
quote_spanned! {span=>
#[allow(warnings)]
fn #make_value_name() -> #ty { async { panic!() } }
async fn #make_value_name() -> #ty {
#declare_inputs
#block
}
}
} else {
quote_spanned! {span=>
#[allow(warnings)]
fn #make_value_name() -> #ty { panic!() }
fn #make_value_name() -> #ty {
#declare_inputs
#block
}
}
};
@@ -285,7 +303,7 @@ mod debug_handler {
let value = #receiver #make_value_name().await;
fn check<T>(_: T)
where T: ::axum::response::IntoResponse
{}
{}
check(value);
}
}
@@ -296,9 +314,11 @@ mod debug_handler {
#make
let value = #make_value_name().await;
fn check<T>(_: T)
where T: ::axum::response::IntoResponse
where T: ::axum::response::IntoResponse
{}
check(value);
}
}