ConnectionResetError («Соединение потеряно») Python Asyncio Server - PullRequest
Новая
       22

ConnectionResetError («Соединение потеряно») Python Asyncio Server

0 голосов
/ 30 апреля 2019

Я пишу сервер redis для клиента redis, подключающегося через порт 9000 (используя redis-py).

Я написал простой тестовый код, чтобы проверить, получает ли сервер команду клиента. Мой код выглядит следующим образом:

client.py 

import redis

r = redis.Redis(
    host="127.0.0.1",
    port=9000
)

v = r.set('foo', 1)
print(v)

v = r.set('bar', '1')

server.py 

import asyncio

HOST = "127.0.0.1"
PORT = 15000


async def exo_redis(reader, writer):

    try:
        while True:
            data = await reader.readuntil(b'\r\n')
            # data = await reader.read(1024)
            if not data:
                break
            print("Data: ", data)
            response = "+OK\r\n"
            writer.write(response.encode())
            await writer.drain()

        writer.close()

    except asyncio.streams.IncompleteReadError:
        pass


def start_server():
    loop = asyncio.get_event_loop()
    coro = asyncio.start_server(exo_redis, HOST, PORT)
    server = loop.run_until_complete(coro)
    print("Serving on ", HOST, "port: ", PORT)
    try:
        loop.run_forever()
    except KeyboardInterrupt:
        print("\nShutting Down Server...\n")
    finally:
        server.close()
        loop.run_until_complete(server.wait_closed())
        loop.close()


if __name__ == '__main__':
    start_server()

После запуска я получаю вывод со следующим исключением: 

Data:  b'*3\r\n'
Data:  b'$3\r\n'
Data:  b'SET\r\n'
Data:  b'$3\r\n'
Data:  b'foo\r\n'
Task exception was never retrieved
future: <Task finished coro=<exo_redis() done, defined at app.py:13> exception=ConnectionResetError('Connection lost')>
Traceback (most recent call last):
  File "app.py", line 23, in exo_redis
    await writer.drain()
  File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/asyncio/streams.py", line 348, in drain
    await self._protocol._drain_helper()
  File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/asyncio/streams.py", line 202, in _drain_helper
    raise ConnectionResetError('Connection lost')
ConnectionResetError: Connection lost
Data:  b'*3\r\n'
Data:  b'$3\r\n'
Data:  b'SET\r\n'
Data:  b'$3\r\n'
Data:  b'bar\r\n'
Data:  b'$1\r\n'
Data:  b'1\r\n'

По какой-то причине первая команда, отправленная клиентом, выдает исключение, а вторая - нет.

Если я использую reader.read (1024) (закомментированный в коде), я получаю следующий вывод: 

Data:  b'*3\r\n$3\r\nSET\r\n$3\r\nfoo\r\n$1\r\n1\r\n'
Data:  b'*3\r\n$3\r\nSET\r\n$3\r\nbar\r\n$1\r\n1\r\n'

Иногда это работает, и исключение не выдается, но иногда я получаю это исключение, когда команда не читается полностью. Любые идеи о том, как это можно исправить, и я могу гарантировать, что вся команда принимается на стороне сервера?

...