从asyncio StreamReader读取

Question

我正在尝试扩展使用流式读取器/写入器（代码）的python asyncio HTTP服务器示例。如果我理解正确，示例处理程序将从读取器读取100个字节，并将其通过写入器回显给客户端。我正在尝试读取100个字节以上的内容...读取直到没有更多内容可读取为止。

我尝试让read（）函数尽可能多地读取，

data = yield from reader.read()

但这似乎永远无法实现。因此，我尝试读取块，直到达到EOF，

while not reader.at_eof():
    data += yield from reader.read(100)

尽管这会检索更多数据，但似乎在读取调用时阻塞，而不是退出while循环。

如何使用流阅读器从客户端获取整个消息？

Answer 1

您应该检查是否StreamReader.read返回一个空字节对象来发出 EOF 信号：

data = bytearray()
while True:
    chunk = yield from reader.read(100)
    if not chunk:
        break
    data += chunk

另外，aiohttp如果您需要功能齐全的 HTTP 客户端，请考虑使用。