refactor read_from_io

src/client/parser.rs

@@ -8,7 +8,7 @@ use std::mem;
 use error::{ParseError, PayloadError};
 
 use server::h1decoder::EncodingDecoder;
-use server::{utils, IoStream};
+use server::IoStream;
 
 use super::response::ClientMessage;
 use super::ClientResponse;
@@ -39,7 +39,7 @@ impl HttpResponseParser {
     {
         // if buf is empty parse_message will always return NotReady, let's avoid that
         if buf.is_empty() {
-            match utils::read_from_io(io, buf) {
+            match io.read_available(buf) {
                 Ok(Async::Ready(0)) => return Err(HttpResponseParserError::Disconnect),
                 Ok(Async::Ready(_)) => (),
                 Ok(Async::NotReady) => return Ok(Async::NotReady),
@@ -59,7 +59,7 @@ impl HttpResponseParser {
                     if buf.capacity() >= MAX_BUFFER_SIZE {
                         return Err(HttpResponseParserError::Error(ParseError::TooLarge));
                     }
-                    match utils::read_from_io(io, buf) {
+                    match io.read_available(buf) {
                         Ok(Async::Ready(0)) => {
                             return Err(HttpResponseParserError::Disconnect)
                         }
@@ -83,7 +83,7 @@ impl HttpResponseParser {
         if self.decoder.is_some() {
             loop {
                 // read payload
-                let (not_ready, stream_finished) = match utils::read_from_io(io, buf) {
+                let (not_ready, stream_finished) = match io.read_available(buf) {
                     Ok(Async::Ready(0)) => (false, true),
                     Err(err) => return Err(err.into()),
                     Ok(Async::NotReady) => (true, false),

src/server/channel.rs

@@ -7,7 +7,7 @@ use futures::{Async, Future, Poll};
 use tokio_io::{AsyncRead, AsyncWrite};
 
 use super::settings::WorkerSettings;
-use super::{h1, h2, utils, HttpHandler, IoStream};
+use super::{h1, h2, HttpHandler, IoStream};
 
 const HTTP2_PREFACE: [u8; 14] = *b"PRI * HTTP/2.0";
 
@@ -139,7 +139,7 @@ where
                 ref mut io,
                 ref mut buf,
             )) => {
-                match utils::read_from_io(io, buf) {
+                match io.read_available(buf) {
                     Ok(Async::Ready(0)) | Err(_) => {
                         debug!("Ignored premature client disconnection");
                         settings.remove_channel();

src/server/mod.rs

@@ -2,7 +2,7 @@
 use std::net::Shutdown;
 use std::{io, time};
 
-use bytes::BytesMut;
+use bytes::{BufMut, BytesMut};
 use futures::{Async, Poll};
 use tokio_io::{AsyncRead, AsyncWrite};
 use tokio_tcp::TcpStream;
@@ -18,7 +18,6 @@ pub(crate) mod helpers;
 pub(crate) mod settings;
 pub(crate) mod shared;
 mod srv;
-pub(crate) mod utils;
 mod worker;
 
 pub use self::settings::ServerSettings;
@@ -37,6 +36,9 @@ use httpresponse::HttpResponse;
 /// max buffer size 64k
 pub(crate) const MAX_WRITE_BUFFER_SIZE: usize = 65_536;
 
+const LW_BUFFER_SIZE: usize = 4096;
+const HW_BUFFER_SIZE: usize = 32_768;
+
 /// Create new http server with application factory.
 ///
 /// This is shortcut for `server::HttpServer::new()` method.
@@ -213,6 +215,27 @@ pub trait IoStream: AsyncRead + AsyncWrite + 'static {
     fn set_nodelay(&mut self, nodelay: bool) -> io::Result<()>;
 
     fn set_linger(&mut self, dur: Option<time::Duration>) -> io::Result<()>;
+
+    fn read_available(&mut self, buf: &mut BytesMut) -> Poll<usize, io::Error> {
+        unsafe {
+            if buf.remaining_mut() < LW_BUFFER_SIZE {
+                buf.reserve(HW_BUFFER_SIZE);
+            }
+            match self.read(buf.bytes_mut()) {
+                Ok(n) => {
+                    buf.advance_mut(n);
+                    Ok(Async::Ready(n))
+                }
+                Err(e) => {
+                    if e.kind() == io::ErrorKind::WouldBlock {
+                        Ok(Async::NotReady)
+                    } else {
+                        Err(e)
+                    }
+                }
+            }
+        }
+    }
 }
 
 impl IoStream for TcpStream {

src/server/utils.rs

@@ -1,31 +0,0 @@
-use bytes::{BufMut, BytesMut};
-use futures::{Async, Poll};
-use std::io;
-
-use super::IoStream;
-
-const LW_BUFFER_SIZE: usize = 4096;
-const HW_BUFFER_SIZE: usize = 32_768;
-
-pub fn read_from_io<T: IoStream>(
-    io: &mut T, buf: &mut BytesMut,
-) -> Poll<usize, io::Error> {
-    unsafe {
-        if buf.remaining_mut() < LW_BUFFER_SIZE {
-            buf.reserve(HW_BUFFER_SIZE);
-        }
-        match io.read(buf.bytes_mut()) {
-            Ok(n) => {
-                buf.advance_mut(n);
-                Ok(Async::Ready(n))
-            }
-            Err(e) => {
-                if e.kind() == io::ErrorKind::WouldBlock {
-                    Ok(Async::NotReady)
-                } else {
-                    Err(e)
-                }
-            }
-        }
-    }
-}