smol-rs
/
event-listener
mirror of https://github.com/smol-rs/event-listener
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

feat: Move Future impl to forwarding macro 

Signed-off-by: John Nunley <dev@notgull.net>
This commit is contained in:
John Nunley 2023-12-19 19:43:29 -08:00 committed by John Nunley
parent e0fefc28b1
commit 6fc00c0f3f
1 changed files with 10 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
src/lib.rs
View File

@ -833,6 +833,7 @@ pub trait Listener<T>: Future<Output = T> + __sealed::Sealed {
fn same_event(&self, other: &Self) -> bool;
}
/// Implement the `Listener` trait using the underlying `InnerListener`.
macro_rules! forward_impl_to_listener {
($gen:ident => $ty:ty) => {
impl<$gen> crate::Listener<$gen> for $ty {
@ -869,6 +870,15 @@ macro_rules! forward_impl_to_listener {
core::ptr::eq::<Inner<$gen>>(&*self.listener().event, &*other.listener().event)
}
}
impl<$gen> Future for $ty {
type Output = $gen;
#[inline]
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<$gen> {
self.listener_mut().poll_internal(cx)
}
}
};
}
@ -922,14 +932,6 @@ impl<T> EventListener<T> {
forward_impl_to_listener! { T => EventListener<T> }
impl<T> Future for EventListener<T> {
type Output = T;
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
self.listener.as_mut().poll_internal(cx)
}
}
/// Create a stack-based event listener for an [`Event`].
#[macro_export]
macro_rules! listener {
@ -1376,13 +1378,4 @@ pub mod __private {
}
forward_impl_to_listener! { T => StackListener<'_, '_, T> }
impl<T> Future for StackListener<'_, '_, T> {
type Output = T;
#[inline]
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
self.listener_mut().poll_internal(cx)
}
}
}