diff options
author | Sean McArthur <sean@seanmonstar.com> | 2020-11-18 10:41:47 -0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-11-18 10:41:47 -0800 |
commit | 34fcef258b84d17f8d418b39eb61fa07fa87c390 (patch) | |
tree | 98e37fc2d6fa541a9e499331df86ba3d1b7b6e3a /tokio/src/net/tcp/stream.rs | |
parent | 7d11aa866837eea50a6f1e0ef7e24846a653cbf1 (diff) |
io: add vectored writes to `AsyncWrite` (#3149)
This adds `AsyncWrite::poll_write_vectored`, and implements it for
`TcpStream` and `UnixStream`.
Refs: #3135.
Diffstat (limited to 'tokio/src/net/tcp/stream.rs')
-rw-r--r-- | tokio/src/net/tcp/stream.rs | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/tokio/src/net/tcp/stream.rs b/tokio/src/net/tcp/stream.rs index de7b4213..28118f73 100644 --- a/tokio/src/net/tcp/stream.rs +++ b/tokio/src/net/tcp/stream.rs @@ -832,6 +832,14 @@ impl TcpStream { ) -> Poll<io::Result<usize>> { self.io.poll_write(cx, buf) } + + pub(super) fn poll_write_vectored_priv( + &self, + cx: &mut Context<'_>, + bufs: &[io::IoSlice<'_>], + ) -> Poll<io::Result<usize>> { + self.io.poll_write_vectored(cx, bufs) + } } impl TryFrom<std::net::TcpStream> for TcpStream { @@ -867,6 +875,18 @@ impl AsyncWrite for TcpStream { self.poll_write_priv(cx, buf) } + fn poll_write_vectored( + self: Pin<&mut Self>, + cx: &mut Context<'_>, + bufs: &[io::IoSlice<'_>], + ) -> Poll<io::Result<usize>> { + self.poll_write_vectored_priv(cx, bufs) + } + + fn is_write_vectored(&self) -> bool { + true + } + #[inline] fn poll_flush(self: Pin<&mut Self>, _: &mut Context<'_>) -> Poll<io::Result<()>> { // tcp flush is a no-op |