1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72
extern crate proc_macro; use proc_macro::TokenStream; use proc_quote::quote; use syn::ReturnType; #[proc_macro_attribute] pub fn fn_handler(_: TokenStream, input: TokenStream) -> TokenStream { let mut input = syn::parse_macro_input!(input as syn::ItemFn); let attrs = &input.attrs; let vis = &input.vis; let sig = &mut input.sig; let body = &input.block; let name = &sig.ident; if sig.asyncness.is_none() { return syn::Error::new_spanned(sig.fn_token, "only async fn is supported") .to_compile_error() .into(); } match sig.inputs.len() { 3 => { let ts: TokenStream = quote! {_conf: ::std::sync::Arc<::salvo::ServerConfig>}.into(); sig.inputs.insert(0, syn::parse_macro_input!(ts as syn::FnArg)); } 4 => {} _ => { return syn::Error::new_spanned(&sig.inputs, "numbers of fn is not supports") .to_compile_error() .into() } } let sdef = quote! { #[allow(non_camel_case_types)] #vis struct #name; impl #name { #(#attrs)* #sig { #body } } }; match sig.output { ReturnType::Default => { (quote! { #sdef #[async_trait] impl salvo::Handler for #name { async fn handle(&self, conf: ::std::sync::Arc<::salvo::ServerConfig>, req: &mut ::salvo::Request, depot: &mut ::salvo::Depot, res: &mut ::salvo::Response) { Self::#name(conf, req, depot, res).await } } }).into() }, ReturnType::Type(_, _) => { (quote! { #sdef #[async_trait] impl salvo::Handler for #name { async fn handle(&self, conf: ::std::sync::Arc<::salvo::ServerConfig>, req: &mut ::salvo::Request, depot: &mut ::salvo::Depot, res: &mut ::salvo::Response) { match Self::#name(conf.clone(), req, depot, res).await { Ok(writer) => ::salvo::Writer::write(writer, conf.clone(), req, depot, res).await, Err(err) => ::salvo::Writer::write(err, conf.clone(), req, depot, res).await, } } } }).into() } } }