我想用 django 连接到内部 http 服务,我需要缓冲这些服务的 http 响应的输出,因为有些内容非常大。
我正在使用 python 3.6、django 2.0http.client
和以下代码:
class HTTPStreamIterAndClose():
def __init__(self, conn, res, buffsize):
self.conn = conn
self.res = res
self.buffsize = buffsize
self.length = 1
bytes_length = int(res.getheader('Content-Length'))
if buffsize < bytes_length:
self.length = math.ceil(bytes_length/buffsize)
def __iter__(self):
return self
def __next__(self):
buff = self.res.read(self.buffsize)
if buff is b'':
self.res.close()
self.conn.close()
raise StopIteration
else:
return buff
def __len__(self):
return self.length
def passthru_http_service(request, server, timeout, path):
serv = HTTPService(server, timeout)
res = serv.request(path)
response = StreamingHttpResponse(
HTTPStreamIterAndClose(serv.connection, res, 200),
content_type='application/json'
)
response['Content-Length'] = res.getheader('Content-Length')
return response
并且响应为空,我使用以下方法测试迭代器:
b''.join(HTTPStreamIterAndClose(serv.connection, res, 200)
一切正常,我不知道为什么不工作。