0.2.5: proxy function macros

Author: hackermondev

Cargo.lock

@@ -1,7 +1,7 @@
 [package]
 name = "batched"
 description = "rust macro util for batching expensive operations"
-version = "0.2.3"
+version = "0.2.5"
 edition = "2024"
 license = "MIT"
 readme = "../README.md"
@@ -11,4 +11,4 @@ keywords = ["batch", "performance", "efficient"]
 
 [dependencies]
 anyhow = "1.0.98"
-batched_derive = { version = "0.2.1", path = "../batched_derive" }
+batched_derive = { version = "0.2.5", path = "../batched_derive" }

batched/Cargo.toml

@@ -1,7 +1,7 @@
 [package]
 name = "batched_derive"
 description = "rust macro util for batching expensive operations"
-version = "0.2.1"
+version = "0.2.5"
 edition = "2024"
 license = "MIT"
 readme = "../README.md"

batched_derive/Cargo.toml

@@ -36,6 +36,7 @@ fn build_identifiers(call_function: &Function) -> Identifiers {
 pub fn build_code(call_function: Function, options: Attributes) -> TokenStream {
     let identifiers = build_identifiers(&call_function);
 
+    let macros = &call_function.macros;
     let visibility = &call_function.visibility;
     let arg = &call_function.batched_arg;
     let arg_name: TokenStream = syn::parse_str(&call_function.batched_arg_name).unwrap();
@@ -123,6 +124,7 @@ pub fn build_code(call_function: Function, options: Attributes) -> TokenStream {
     quote! {
         #executor
 
+        #(#macros)*
         async fn #inner_batched(#arg) -> #returned {
             let result = async { #inner_body };
             let result = result.await;
@@ -246,7 +248,7 @@ fn build_executor(
                     }
 
                     if return_channels.is_empty() {
-                        break;
+                        continue;
                     }
 
                     let mut data = vec![];

batched_derive/src/builder.rs

@@ -9,6 +9,7 @@ use crate::utils::expr_to_u64;
 
 #[derive(Debug)]
 pub struct Function {
+    pub macros: Vec<TokenStream>,
     pub identifier: String,
     pub visibility: TokenStream,
     pub inner: TokenStream,
@@ -53,6 +54,12 @@ impl Function {
     pub fn parse(tokens: TokenStream) -> Self {
         let function: ItemFn = syn::parse2(tokens).expect("invalid function");
 
+        let macros = function
+            .attrs
+            .into_iter()
+            .map(|attr| attr.into_token_stream())
+            .collect();
+
         let visibility = function.vis.into_token_stream();
         let identifier = function.sig.ident.to_string();
         let args = function.sig.inputs;
@@ -156,6 +163,7 @@ impl Function {
         let batched_arg_type = batched_arg_type.unwrap();
 
         Self {
+            macros,
             identifier,
             visibility,
             batched_arg,

batched_derive/src/parse.rs

@@ -9,11 +9,12 @@ anyhow = "1.0.98"
 batched = { path = "../batched" }
 batched_derive = { path = "../batched_derive" }
 tokio = { version = "1.44.2", features = ["full"] }
+tracing = "0.1.41"
 
 [[test]]
-name = "macros"
-path = "macros.rs"
+name = "macro"
+path = "src/macro.rs"
 
 [[test]]
 name = "error"
-path = "error.rs"
+path = "src/error.rs"

tests/Cargo.toml

@@ -47,10 +47,11 @@ async fn window() {
         numbers.iter().sum()
     }
 
-    let before = Instant::now();
+    let start = Instant::now();
     add_multiple(vec![1, 1, 1]).await;
-    let after = before.elapsed();
-    assert!(after.as_secs() == 1);
+
+    let elapsed = start.elapsed();
+    assert!(elapsed.as_secs() == 1);
 }
 
 #[tokio::test]
@@ -79,5 +80,4 @@ async fn returned_iterator_with_error() {
 
     let result = add_each(2).await.unwrap();
     assert!(result == 3);
-}
-
+}