fix: correctly handle async (#7)

fast · Jun 19, 2024 · fc49f28 · fc49f28
1 parent f843fc8
commit fc49f28
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "logcall"
-version = "0.1.8"
+version = "0.1.9"
 edition = "2021"
 authors = ["andylokandy <[email protected]>"]
 description = "An attribute macro that logs the function return value."

diff --git a/src/lib.rs b/src/lib.rs
@@ -187,8 +187,7 @@ fn gen_block(
                     #[allow(clippy::useless_format)]
                     let __input_string = format!(#input_format);
                     #[allow(unknown_lints)]
-                    #[allow(clippy::redundant_closure_call)]
-                    let __ret_value = (move || #block)();
+                    let __ret_value = async { #block }.await;
                     #log;
                     __ret_value
                 );
@@ -211,6 +210,7 @@ fn gen_block(
                     let __input_string = format!(#input_format);
                     #[allow(unknown_lints)]
                     #[allow(clippy::redundant_closure_call)]
+                    #[allow(clippy::let_unit_value)]
                     let __ret_value = (move || #block)();
                     #log;
                     __ret_value
@@ -258,8 +258,7 @@ fn gen_block(
                     #[allow(clippy::useless_format)]
                     let __input_string = format!(#input_format);
                     #[allow(unknown_lints)]
-                    #[allow(clippy::redundant_closure_call)]
-                    let __ret_value = (move || #block)();
+                    let __ret_value = async { #block }.await;
                     match __ret_value {
                         #ok_arm
                         #err_arm
@@ -283,6 +282,7 @@ fn gen_block(
                     let __input_string = format!(#input_format);
                     #[allow(unknown_lints)]
                     #[allow(clippy::redundant_closure_call)]
+                    #[allow(clippy::let_unit_value)]
                     let __ret_value = (move || #block)();
                     match __ret_value {
                         #ok_arm

diff --git a/tests/ui/ok/minitrace.rs b/tests/ui/ok/minitrace.rs
@@ -4,7 +4,9 @@ fn f() {}
 
 #[logcall::logcall]
 #[minitrace::trace]
-async fn g() {}
+async fn g() {
+    std::future::ready(1).await;
+}
 
 fn main() {
     f();