forked/hyper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(client): always wait on reads for pooled connections

Browse Source
This commit is contained in:
Sean McArthur
2017-11-28 16:17:20 -08:00
parent 60d0eaf891
commit 9f21241002
2 changed files with 15 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/proto/conn.rs
View File

@@ -102,16 +102,7 @@ where I: AsyncRead + AsyncWrite,
pub fn can_read_head(&self) -> bool {
match self.state.reading {
Reading::Init => {
if T::should_read_first() {
true
} else {
match self.state.writing {
Writing::Init => false,
_ => true,
}
}
}
Reading::Init => true,
_ => false,
}
}
@@ -245,19 +236,13 @@ where I: AsyncRead + AsyncWrite,
Reading::Closed => false,
};
let wants_write = match self.state.writing {
match self.state.writing {
Writing::Continue(..) |
Writing::Body(..) |
Writing::Ending(..) => return,
Writing::Init => true,
Writing::KeepAlive => false,
Writing::Closed => false,
};
// if the client is at Reading::Init and Writing::Init,
// it's not actually looking for a read, but a write.
if wants_write && !T::should_read_first() {
return;
Writing::Init |
Writing::KeepAlive |
Writing::Closed => (),
}
if !self.io.is_read_blocked() {

15
src/proto/dispatch.rs
View File

@@ -1,3 +1,5 @@
use std::io;
use futures::{Async, AsyncSink, Future, Poll, Sink, Stream};
use futures::sync::{mpsc, oneshot};
use tokio_io::{AsyncRead, AsyncWrite};
@@ -64,7 +66,7 @@ where
} else {
None
};
self.dispatch.recv_msg(Ok((head, body))).expect("recv_msg with Ok shouldn't error");
self.dispatch.recv_msg(Ok((head, body)))?;
},
Ok(Async::Ready(None)) => {
// read eof, conn will start to shutdown automatically
@@ -306,10 +308,13 @@ where
fn recv_msg(&mut self, msg: ::Result<(Self::RecvItem, Option<Body>)>) -> ::Result<()> {
match msg {
Ok((msg, body)) => {
let res = super::response::from_wire(msg, body);
let cb = self.callback.take().expect("recv_msg without callback");
let _ = cb.send(Ok(res));
Ok(())
if let Some(cb) = self.callback.take() {
let res = super::response::from_wire(msg, body);
let _ = cb.send(Ok(res));
Ok(())
} else {
Err(::Error::Io(io::Error::new(io::ErrorKind::InvalidData, "response received without matching request")))
}
},
Err(err) => {
if let Some(cb) = self.callback.take() {