This commit is contained in:
Youhei Sakurai
2015-02-27 02:44:47 +09:00
parent bd6c3f64c1
commit 77bb0b74ed
3 changed files with 20 additions and 1 deletions

View File

@@ -1332,7 +1332,12 @@ class HTTPHandler(ProtocolHandler):
# incrementally:
h = flow.response._assemble_head(preserve_transfer_encoding=True)
self.c.client_conn.send(h)
for chunk in http.read_http_body_chunked(self.c.server_conn.rfile,
for chunk in hasattr(flow.response.stream, "__call__") and \
flow.response.stream(http.read_http_body_chunked(self.c.server_conn.rfile,
flow.response.headers,
self.c.config.body_size_limit, flow.request.method,
flow.response.code, False, 4096)) or \
http.read_http_body_chunked(self.c.server_conn.rfile,
flow.response.headers,
self.c.config.body_size_limit, flow.request.method,
flow.response.code, False, 4096):