Set the listener slot to None after finishing

Signed-off-by: John Nunley <dev@notgull.net>
smol-rs · Dec 21, 2023 · c13adc3 · c13adc3
1 parent 7929559
commit c13adc3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/lib.rs b/src/lib.rs
@@ -458,7 +458,11 @@ impl<'a, 'evl> Strategy<'evl> for NonBlocking<'a> {
         event_listener: &mut Option<L>,
         context: &mut Self::Context,
     ) -> Poll<T> {
-        Pin::new(event_listener.as_mut().expect("`event_listener` should never be `None`")).poll(context)
+        let poll = Pin::new(event_listener.as_mut().expect("`event_listener` should never be `None`")).poll(context);
+        if poll.is_ready() {
+            *event_listener = None;
+        }
+        poll
     }
 }