From 2c48ee6537416d778f97a8e327ed88cc3dff8ce3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E8=BF=90=E5=AE=B6?= Date: Thu, 3 Oct 2024 11:43:07 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E8=B7=AF=E7=94=B1=E9=94=99?= =?UTF-8?q?=E8=AF=AF=E4=BF=A1=E6=81=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- macro/src/route.rs | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/macro/src/route.rs b/macro/src/route.rs index 41295bf..9a2ff3e 100644 --- a/macro/src/route.rs +++ b/macro/src/route.rs @@ -18,7 +18,7 @@ struct Args { impl Parse for Args { fn parse(input: ParseStream) -> syn::parse::Result { let vars = Punctuated::::parse_terminated(input)?; - + Ok(Args { vars: vars.into_iter().collect(), }) @@ -31,7 +31,7 @@ impl Args { Some(var) => Ok(var.clone()), None => return Err(syn::Error::new( Span::call_site().into(), - "No Route was provided" + "route must have one argument" )) } } @@ -41,26 +41,26 @@ impl Args { pub fn gen_route(attr: TokenStream, item: TokenStream) -> TokenStream { let args = parse_macro_input!(attr as Args); let func = parse_macro_input!(item as ItemFn); - + let vis = func.vis.clone(); let ident = func.sig.ident.clone(); - + let method = args.get_arg(0).unwrap(); let route = args.get_arg(1).unwrap(); let expanded = quote! { #[allow(non_camel_case_types)] #vis struct #ident; - + impl #ident { #vis fn route() -> axum::Router { #func - + axum::Router::new().route(#route, #method (#ident)) } } }; - + expanded.into() }