summaryrefslogtreecommitdiffstats
path: root/tokio/src/net/tcp/split.rs
diff options
context:
space:
mode:
Diffstat (limited to 'tokio/src/net/tcp/split.rs')
-rw-r--r--tokio/src/net/tcp/split.rs12
1 files changed, 9 insertions, 3 deletions
diff --git a/tokio/src/net/tcp/split.rs b/tokio/src/net/tcp/split.rs
index 28c94eb4..296b469d 100644
--- a/tokio/src/net/tcp/split.rs
+++ b/tokio/src/net/tcp/split.rs
@@ -60,7 +60,7 @@ impl ReadHalf<'_> {
/// # Examples
///
/// ```no_run
- /// use tokio::io;
+ /// use tokio::io::{self, ReadBuf};
/// use tokio::net::TcpStream;
///
/// use futures::future::poll_fn;
@@ -70,6 +70,7 @@ impl ReadHalf<'_> {
/// let mut stream = TcpStream::connect("127.0.0.1:8000").await?;
/// let (mut read_half, _) = stream.split();
/// let mut buf = [0; 10];
+ /// let mut buf = ReadBuf::new(&mut buf);
///
/// poll_fn(|cx| {
/// read_half.poll_peek(cx, &mut buf)
@@ -80,7 +81,11 @@ impl ReadHalf<'_> {
/// ```
///
/// [`TcpStream::poll_peek`]: TcpStream::poll_peek
- pub fn poll_peek(&mut self, cx: &mut Context<'_>, buf: &mut [u8]) -> Poll<io::Result<usize>> {
+ pub fn poll_peek(
+ &mut self,
+ cx: &mut Context<'_>,
+ buf: &mut ReadBuf<'_>,
+ ) -> Poll<io::Result<usize>> {
self.0.poll_peek(cx, buf)
}
@@ -124,7 +129,8 @@ impl ReadHalf<'_> {
/// [`read`]: fn@crate::io::AsyncReadExt::read
/// [`AsyncReadExt`]: trait@crate::io::AsyncReadExt
pub async fn peek(&mut self, buf: &mut [u8]) -> io::Result<usize> {
- poll_fn(|cx| self.poll_peek(cx, buf)).await
+ let mut buf = ReadBuf::new(buf);
+ poll_fn(|cx| self.poll_peek(cx, &mut buf)).await
}
}