Merge pull request #20 from fibonacci1729/async

Author: suneetnangia

Cargo.lock

@@ -2,7 +2,7 @@ use chrono::{DateTime, Utc};
 use spin_mqtt_sdk::{mqtt_component, Payload};
 
 #[mqtt_component]
-fn handle_message(message: Payload) -> anyhow::Result<()> {
+async fn handle_message(message: Payload) -> anyhow::Result<()> {
     let datetime: DateTime<Utc> = std::time::SystemTime::now().into();
     let formatted_time = datetime.format("%Y-%m-%d %H:%M:%S.%f").to_string();
 

examples/mqtt-app/Cargo.lock

@@ -8,5 +8,6 @@ include = ["../spin-mqtt.wit"]
 name = "spin_mqtt_sdk"
 
 [dependencies]
+spin-executor = "3.0.1"
 spin-mqtt-macro = { path = "macro" }
 wit-bindgen = { workspace = true }

examples/mqtt-app/src/lib.rs

@@ -7,6 +7,7 @@ const WIT_PATH: &str = concat!(env!("CARGO_MANIFEST_DIR"), "/../../spin-mqtt.wit
 pub fn mqtt_component(_attr: TokenStream, item: TokenStream) -> TokenStream {
     let func = syn::parse_macro_input!(item as syn::ItemFn);
     let func_name = &func.sig.ident;
+    let await_postfix = func.sig.asyncness.map(|_| quote!(.await));
     let preamble = preamble();
 
     quote!(
@@ -17,13 +18,15 @@ pub fn mqtt_component(_attr: TokenStream, item: TokenStream) -> TokenStream {
             }
             impl self::preamble::Guest for preamble::Mqtt {
                 fn handle_message(payload: ::spin_mqtt_sdk::Payload) -> ::std::result::Result<(), ::spin_mqtt_sdk::Error> {
-                    match super::#func_name(payload) {
-                        ::std::result::Result::Ok(()) => ::std::result::Result::Ok(()),
-                        ::std::result::Result::Err(e) => {
-                            eprintln!("{}", e);
-                            ::std::result::Result::Err(::spin_mqtt_sdk::Error::Other(e.to_string()))
-                        },
-                    }
+                    ::spin_mqtt_sdk::executor::run(async move {
+                        match super::#func_name(payload)#await_postfix {
+                            ::std::result::Result::Ok(()) => ::std::result::Result::Ok(()),
+                            ::std::result::Result::Err(e) => {
+                                eprintln!("{}", e);
+                                ::std::result::Result::Err(::spin_mqtt_sdk::Error::Other(e.to_string()))
+                            },
+                        }
+                    })
                 }
             }
         }

sdk/Cargo.toml

@@ -1,5 +1,8 @@
 pub use spin_mqtt_macro::mqtt_component;
 
+#[doc(hidden)]
+pub use spin_executor as executor;
+
 #[doc(hidden)]
 pub mod wit {
     #![allow(missing_docs)]