actix/actix-web
1
0
Fork 0
mirror of https://github.com/fafhrd91/actix-web synced 2024-11-24 00:21:08 +01:00
Code Issues Releases Wiki Activity

Fix Client Request with custom Body Stream halting on certain size requests #176

Browse Source
This commit is contained in:
Nikolay Kim 2018-04-15 10:22:09 -07:00
parent a9ea649348
commit 58cc0dfbc5
3 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
CHANGES.md
View File

@ -9,6 +9,8 @@
* Fix StaticFiles does not support percent encoded paths #177
* Fix Client Request with custom Body Stream halting on certain size requests #176
## 0.5.1 (2018-04-12)

13
src/client/pipeline.rs
View File

@ -184,6 +184,7 @@ impl Future for SendRequest {
parser: Some(HttpResponseParser::default()),
parser_buf: BytesMut::new(),
disconnected: false,
body_completed: false,
drain: None,
decompress: None,
should_decompress: self.req.response_decompress(),
@ -217,6 +218,7 @@ impl Future for SendRequest {
pub(crate) struct Pipeline {
body: IoBody,
body_completed: bool,
conn: Option<Connection>,
writer: HttpClientWriter,
parser: Option<HttpResponseParser>,
@ -394,7 +396,7 @@ impl Pipeline {
IoBody::Payload(mut body) => match body.poll()? {
Async::Ready(None) => {
self.writer.write_eof()?;
self.disconnected = true;
self.body_completed = true;
break;
}
Async::Ready(Some(chunk)) => {
@ -421,8 +423,7 @@ impl Pipeline {
for frame in vec {
match frame {
Frame::Chunk(None) => {
// info.context = Some(ctx);
self.disconnected = true;
self.body_completed = true;
self.writer.write_eof()?;
break 'outter;
}
@ -451,7 +452,7 @@ impl Pipeline {
}
}
IoBody::Done => {
self.disconnected = true;
self.body_completed = true;
done = true;
break;
}
@ -472,7 +473,9 @@ impl Pipeline {
.poll_completed(self.conn.as_mut().unwrap(), false)
{
Ok(Async::Ready(_)) => {
if self.disconnected {
if self.disconnected
|| (self.body_completed && self.writer.is_completed())
{
self.write_state = RunningState::Done;
} else {
self.write_state.resume();

4
src/client/writer.rs
View File

@ -61,6 +61,10 @@ impl HttpClientWriter {
self.buffer.take();
}
pub fn is_completed(&mut self) -> bool {
self.buffer.is_empty()
}
// pub fn keepalive(&self) -> bool {
// self.flags.contains(Flags::KEEPALIVE) &&
// !self.flags.contains(Flags::UPGRADE) }