Add a poll_with_strategy method to easy_wrapper! (#60)

smol-rs · May 27, 2023 · 5180532 · 5180532
1 parent d6065ed
commit 5180532
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/strategy/src/lib.rs b/strategy/src/lib.rs
@@ -212,6 +212,14 @@ macro_rules! easy_wrapper {
                 use $crate::EventListenerFuture;
                 self._inner.into_inner().wait()
             }
+
+            pub(crate) fn poll_with_strategy<'__strategy, __S: $crate::Strategy<'__strategy>>(
+                self: ::core::pin::Pin<&'__strategy mut Self>,
+                strategy: &mut __S,
+                context: &mut __S::Context,
+            ) -> ::core::task::Poll<$output> {
+                self.project()._inner.get_pin_mut().poll_with_strategy(strategy, context)
+            }
         }
 
         impl $(<