add support for async-std

Author: spencerjibz

.gitignore

@@ -1,3 +1,4 @@
 /target
 /Cargo.lock
-*.idea/
+*.idea/
+.vscode

Cargo.toml

@@ -14,18 +14,28 @@ readme = "./README.md"
 [dependencies]
 anyhow = "1.0.75"
 bincode = "1.3.3"
+cfg-if = "1.0.0"
 futures = "0.3.29"
 lazy_static = "1.4.0"
 tokio-test = "0.4.3"
 
+[dependencies.async-std]
+version = "1.12.0"
+optional = true
+
+features = ["alloc"]
 [dependencies.serde]
 version = "1.0.190"
 features = ["derive"]
-
 [dependencies.tokio]
 version = "1.33.0"
-features = ["full"]
+features = ["rt", "macros", "rt-multi-thread"]
+optional = true
 
 [dependencies.uuid]
 version = "1.5.0"
 features = ["v4"]
+
+[features]
+use-async-std = ["async-std"]
+default = ["tokio"]

src/lib.rs

@@ -113,11 +113,9 @@
         License: MIT
 */
 
+use futures::future::{BoxFuture, Future, FutureExt};
 use serde::{Deserialize, Serialize};
 use std::{collections::HashMap, sync::Arc};
-use tokio::task::{self};
-
-use futures::future::{BoxFuture, Future, FutureExt};
 use uuid::Uuid;
 
 pub type AsyncCB = dyn Fn(Vec<u8>) -> BoxFuture<'static, ()> + Send + Sync + 'static;
@@ -169,14 +167,34 @@ impl AsyncEventEmitter {
 
                 match listener.limit {
                     None => {
-                        callback_handlers.push(task::spawn(async move {
-                            callback(bytes).await;
+                        cfg_if::cfg_if! {
+                          if  #[cfg(feature = "use-async-std")] {
+                            use async_std::task::spawn;
+                          callback_handlers.push(spawn(async move {
+                         callback(bytes).await;
                         }));
+                          } else {
+                             use tokio::spawn;
+                         callback_handlers
+                             .push(spawn(async move { callback(bytes).await }));
+                          }
+
+                         };
                     }
                     Some(limit) => {
                         if limit != 0 {
-                            callback_handlers
-                                .push(task::spawn(async move { callback(bytes).await }));
+                            cfg_if::cfg_if! {
+                              if  #[cfg(feature = "use-async-std")] {
+                              callback_handlers.push(async_std::task::spawn(async move {
+                             callback(bytes).await;
+                            }));
+                              } else {
+                             callback_handlers
+                                 .push(tokio::spawn(async move { callback(bytes).await }));
+                              }
+
+                             };
+
                             listener.limit = Some(limit - 1);
                         } else {
                             listeners_to_remove.push(index);
@@ -192,7 +210,7 @@ impl AsyncEventEmitter {
         }
 
         for handles in callback_handlers {
-            handles.await?;
+            handles.await;
         }
 
         Ok(())