Re-enable writev support in TcpStreams (#1956)

1 files changed, 9 insertions, 0 deletions

diff --git a/tokio/src/net/tcp/split.rs b/tokio/src/net/tcp/split.rs
index 2b337c08..6034d4ef 100644
--- a/tokio/src/net/tcp/split.rs
+++ b/tokio/src/net/tcp/split.rs
@@ -11,6 +11,7 @@
 use crate::io::{AsyncRead, AsyncWrite};
 use crate::net::TcpStream;
 
+use bytes::Buf;
 use std::io;
 use std::mem::MaybeUninit;
 use std::net::Shutdown;
@@ -55,6 +56,14 @@ impl AsyncWrite for WriteHalf<'_> {
         self.0.poll_write_priv(cx, buf)
     }
 
+    fn poll_write_buf<B: Buf>(
+        self: Pin<&mut Self>,
+        cx: &mut Context<'_>,
+        buf: &mut B,
+    ) -> Poll<io::Result<usize>> {
+        self.0.poll_write_buf_priv(cx, buf)
+    }
+
     #[inline]
     fn poll_flush(self: Pin<&mut Self>, _: &mut Context<'_>) -> Poll<io::Result<()>> {
         // tcp flush is a no-op