From 025511baa654bf664c3d3e684364707bc6db5ed2 Mon Sep 17 00:00:00 2001 From: Emilia Hane Date: Mon, 27 Nov 2023 20:01:00 +0100 Subject: [PATCH] Fix sink implementation (#5594) --- crates/net/eth-wire/src/p2pstream.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/crates/net/eth-wire/src/p2pstream.rs b/crates/net/eth-wire/src/p2pstream.rs index 8d407c4872..1d3f660643 100644 --- a/crates/net/eth-wire/src/p2pstream.rs +++ b/crates/net/eth-wire/src/p2pstream.rs @@ -578,8 +578,7 @@ where Ok(()) } - /// Returns Poll::Ready(Ok(())) when no buffered items remain and the sink has been successfully - /// closed. + /// Returns `Poll::Ready(Ok(()))` when no buffered items remain. fn poll_flush(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll> { let mut this = self.project(); loop { @@ -599,6 +598,7 @@ where fn poll_close(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll> { ready!(self.as_mut().poll_flush(cx))?; + ready!(self.project().inner.poll_close(cx))?; Poll::Ready(Ok(())) }