Wake the task before returning Poll::Pending

mattsse · Feb 14, 2024 · cbc4db5 · cbc4db5
1 parent fb54899
commit cbc4db5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/conn.rs b/src/conn.rs
@@ -154,8 +154,9 @@ impl<T: EventMessage + Unpin> Stream for Connection<T> {
                 Poll::Ready(Some(ready))
             }
             Some(Ok(WsMessage::Close(_))) => Poll::Ready(None),
+            // ignore ping and pong
             Some(Ok(WsMessage::Ping(_))) | Some(Ok(WsMessage::Pong(_))) => {
-                // ignore pings
+                cx.waker().wake_by_ref();
                 Poll::Pending
             }
             Some(Ok(msg)) => Poll::Ready(Some(Err(CdpError::UnexpectedWsMessage(msg)))),