nordic-dev.net/embassy
2
0
Fork 0
mirror of https://github.com/embassy-rs/embassy.git synced 2024-11-25 00:02:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Expose poll_ready_to_{send,receive} in Sender/Receiver

Browse Source
This commit is contained in:
Ruben De Smet 2023-08-09 11:50:26 +02:00
parent c1da2c0219
commit b658f10db9
No known key found for this signature in database
GPG Key ID: 1AE26A210C14115B
1 changed files with 39 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
embassy-sync/src/channel.rs
View File

@ -65,6 +65,13 @@ where
pub fn try_send(&self, message: T) -> Result<(), TrySendError<T>> {
self.channel.try_send(message)
}
/// Allows a poll_fn to poll until the channel is ready to send
///
/// See [`Channel::poll_ready_to_send()`]
pub fn poll_ready_to_send(&self, cx: &mut Context<'_>) -> bool {
self.channel.poll_ready_to_send(cx)
}
}
/// Send-only access to a [`Channel`] without knowing channel size.
@ -106,6 +113,13 @@ impl<'ch, T> DynamicSender<'ch, T> {
pub fn try_send(&self, message: T) -> Result<(), TrySendError<T>> {
self.channel.try_send_with_context(message, None)
}
/// Allows a poll_fn to poll until the channel is ready to send
///
/// See [`Channel::poll_ready_to_send()`]
pub fn poll_ready_to_send(&self, cx: &mut Context<'_>) -> bool {
self.channel.poll_ready_to_send(cx)
}
}
/// Receive-only access to a [`Channel`].
@ -144,6 +158,13 @@ where
pub fn try_recv(&self) -> Result<T, TryRecvError> {
self.channel.try_recv()
}
/// Allows a poll_fn to poll until the channel is ready to receive
///
/// See [`Channel::poll_ready_to_receive()`]
pub fn poll_ready_to_receive(&self, cx: &mut Context<'_>) -> bool {
self.channel.poll_ready_to_receive(cx)
}
}
/// Receive-only access to a [`Channel`] without knowing channel size.
@ -173,6 +194,13 @@ impl<'ch, T> DynamicReceiver<'ch, T> {
pub fn try_recv(&self) -> Result<T, TryRecvError> {
self.channel.try_recv_with_context(None)
}
/// Allows a poll_fn to poll until the channel is ready to receive
///
/// See [`Channel::poll_ready_to_receive()`]
pub fn poll_ready_to_receive(&self, cx: &mut Context<'_>) -> bool {
self.channel.poll_ready_to_receive(cx)
}
}
impl<'ch, M, T, const N: usize> From<Receiver<'ch, M, T, N>> for DynamicReceiver<'ch, T>
@ -286,6 +314,9 @@ trait DynamicChannel<T> {
fn try_send_with_context(&self, message: T, cx: Option<&mut Context<'_>>) -> Result<(), TrySendError<T>>;
fn try_recv_with_context(&self, cx: Option<&mut Context<'_>>) -> Result<T, TryRecvError>;
fn poll_ready_to_send(&self, cx: &mut Context<'_>) -> bool;
fn poll_ready_to_receive(&self, cx: &mut Context<'_>) -> bool;
}
/// Error returned by [`try_recv`](Channel::try_recv).
@ -492,6 +523,14 @@ where
fn try_recv_with_context(&self, cx: Option<&mut Context<'_>>) -> Result<T, TryRecvError> {
Channel::try_recv_with_context(self, cx)
}
fn poll_ready_to_send(&self, cx: &mut Context<'_>) -> bool {
Channel::poll_ready_to_send(self, cx)
}
fn poll_ready_to_receive(&self, cx: &mut Context<'_>) -> bool {
Channel::poll_ready_to_receive(self, cx)
}
}
#[cfg(test)]