jleony/nanoreth
1
0
Fork 0
mirror of https://github.com/hl-archive-node/nanoreth.git synced 2025-12-06 19:09:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(net): batch P2PStream sends (#9498)

Browse Source
This commit is contained in:
Emilia Hane
2024-07-25 20:39:36 +02:00
committed by GitHub
parent 2766c355e6
commit b020ff218e
1 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
crates/net/eth-wire/src/p2pstream.rs
View File

@ -613,19 +613,24 @@ where
/// Returns `Poll::Ready(Ok(()))` when no buffered items remain.
fn poll_flush(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
let mut this = self.project();
loop {
match ready!(this.inner.as_mut().poll_flush(cx)) {
Err(err) => return Poll::Ready(Err(err.into())),
Ok(()) => {
let poll_res = loop {
match this.inner.as_mut().poll_ready(cx) {
Poll::Pending => break Poll::Pending,
Poll::Ready(Err(err)) => break Poll::Ready(Err(err.into())),
Poll::Ready(Ok(())) => {
let Some(message) = this.outgoing_messages.pop_front() else {
return Poll::Ready(Ok(()))
break Poll::Ready(Ok(()))
};
if let Err(err) = this.inner.as_mut().start_send(message) {
return Poll::Ready(Err(err.into()))
break Poll::Ready(Err(err.into()))
}
}
}
}
};
ready!(this.inner.as_mut().poll_flush(cx))?;
poll_res
}
fn poll_close(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {