Merge pull request #136 from EspressoSystems/rm/add-try-send

Add `try_send`

Author: rob-maron

Cargo.toml

@@ -2,7 +2,7 @@
 name = "async-compatibility-layer"
 description = "an abstraction layer for using both async-std and tokio"
 authors = ["Espresso Systems <[email protected]>"]
-version = "1.2.0"
+version = "1.2.1"
 edition = "2021"
 license = "MIT"
 

src/channel.rs

@@ -26,7 +26,9 @@ compile_error!(
     "async_channel_impl = 'tokio' requires tokio runtime, e. g. async_executor_impl = 'tokio'"
 );
 
-pub use bounded::{bounded, BoundedStream, Receiver, RecvError, SendError, Sender, TryRecvError};
+pub use bounded::{
+    bounded, BoundedStream, Receiver, RecvError, SendError, Sender, TryRecvError, TrySendError,
+};
 pub use oneshot::{oneshot, OneShotReceiver, OneShotRecvError, OneShotSender, OneShotTryRecvError};
 pub use unbounded::{
     unbounded, UnboundedReceiver, UnboundedRecvError, UnboundedSendError, UnboundedSender,

src/channel/bounded.rs

@@ -5,7 +5,7 @@ use futures::Stream;
 /// inner module, used to group feature-specific imports
 #[cfg(async_channel_impl = "tokio")]
 mod inner {
-    pub use tokio::sync::mpsc::error::{SendError, TryRecvError};
+    pub use tokio::sync::mpsc::error::{SendError, TryRecvError, TrySendError};
 
     use tokio::sync::mpsc::{Receiver as InnerReceiver, Sender as InnerSender};
 
@@ -47,7 +47,7 @@ mod inner {
 /// inner module, used to group feature-specific imports
 #[cfg(async_channel_impl = "flume")]
 mod inner {
-    pub use flume::{RecvError, SendError, TryRecvError};
+    pub use flume::{RecvError, SendError, TryRecvError, TrySendError};
 
     use flume::{r#async::RecvStream, Receiver as InnerReceiver, Sender as InnerSender};
 
@@ -77,7 +77,7 @@ mod inner {
 /// inner module, used to group feature-specific imports
 #[cfg(not(any(async_channel_impl = "flume", async_channel_impl = "tokio")))]
 mod inner {
-    pub use async_std::channel::{RecvError, SendError, TryRecvError};
+    pub use async_std::channel::{RecvError, SendError, TryRecvError, TrySendError};
 
     use async_std::channel::{Receiver as InnerReceiver, Sender as InnerSender};
 
@@ -121,6 +121,15 @@ impl<T> Sender<T> {
 
         result
     }
+
+    /// Try to send a value over the channel. Will return immediately if the channel is full.
+    ///
+    /// # Errors
+    /// - If the channel is full
+    /// - If the channel is dropped
+    pub fn try_send(&self, msg: T) -> Result<(), TrySendError<T>> {
+        self.0.try_send(msg)
+    }
 }
 
 impl<T> Receiver<T> {