From df20c162ae1308c07073b6a67c8ba4202f52d208 Mon Sep 17 00:00:00 2001 From: Sunjay Varma Date: Sat, 12 Dec 2020 08:47:35 -0800 Subject: sync: add blocking_recv method to UnboundedReceiver, similar to Receiver::blocking_recv (#3262) --- tokio/src/sync/mpsc/unbounded.rs | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) diff --git a/tokio/src/sync/mpsc/unbounded.rs b/tokio/src/sync/mpsc/unbounded.rs index fe882d5b..48fbca96 100644 --- a/tokio/src/sync/mpsc/unbounded.rs +++ b/tokio/src/sync/mpsc/unbounded.rs @@ -122,6 +122,36 @@ impl UnboundedReceiver { poll_fn(|cx| self.poll_recv(cx)).await } + /// Blocking receive to call outside of asynchronous contexts. + /// + /// # Panics + /// + /// This function panics if called within an asynchronous execution + /// context. + /// + /// # Examples + /// + /// ``` + /// use std::thread; + /// use tokio::sync::mpsc; + /// + /// #[tokio::main] + /// async fn main() { + /// let (tx, mut rx) = mpsc::unbounded_channel::(); + /// + /// let sync_code = thread::spawn(move || { + /// assert_eq!(Some(10), rx.blocking_recv()); + /// }); + /// + /// let _ = tx.send(10); + /// sync_code.join().unwrap(); + /// } + /// ``` + #[cfg(feature = "sync")] + pub fn blocking_recv(&mut self) -> Option { + crate::future::block_on(self.recv()) + } + /// Attempts to return a pending value on this receiver without blocking. /// /// This method will never block the caller in order to wait for data to -- cgit v1.2.3