我正在尝试使用ProcessPoolExecutor()将用于自动化测试的TCP echo client and server组合到单个模块中,并且按预期工作。

唯一的问题是我无法完成事件循环。我可以看到执行程序的目标run_client()的最后一行的调试输出,但是看起来执行程序本身仍然处于阻塞状态。

代码:

import asyncio
import concurrent.futures
from concurrent.futures import ProcessPoolExecutor

async def server_handle_echo(reader, writer):
    data = await reader.read(100)
    message = data.decode()
    addr = writer.get_extra_info('peername')
    print("Received %r from %r" % (message, addr))

    print("Send: %r" % message)
    writer.write(data)
    await writer.drain()

    print("Close the client socket")
    writer.close()


async def echo_client_handler(message, loop):
    reader, writer = await asyncio.open_connection('127.0.0.1', 8888,
                                                   loop=loop)

    print('Send: %r' % message)
    writer.write(message.encode())

    data = await reader.read(100)
    print('Received: %r' % data.decode())

    print('Close the socket')
    writer.close()



def run_client():
    message = 'Hello World!'
    loop = asyncio.get_event_loop()
    loop.run_until_complete(echo_client_handler(message, loop))
    loop.close()
    print('run_client last line')

executor = ProcessPoolExecutor(1)
loop = asyncio.get_event_loop()

coro = asyncio.start_server(server_handle_echo, '127.0.0.1', 8888, loop=loop)
server = loop.run_until_complete(coro)

# Serve requests until Ctrl+C is pressed
print('Serving on {}'.format(server.sockets[0].getsockname()))
try:
    #loop.run_forever()
    client = asyncio.ensure_future(loop.run_in_executor(executor, run_client))
    loop.run_until_complete(client)

except KeyboardInterrupt:
    pass

# Close the server
server.close()
loop.run_until_complete(server.wait_closed())
loop.close()

输出:
Serving on ('127.0.0.1', 8888)
Send: 'Hello World!'
Received 'Hello World!' from ('127.0.0.1', 51157)
Send: 'Hello World!'
Close the client socket
Received: 'Hello World!'
Close the socket
run_client last line

此输出后,它将进入msg循环,等待IO。

期待您的帮助。抱歉,我是异步一日专家:)

最佳答案

您不能为子进程中运行的客户端使用相同的事件循环,需要new loop:

def run_client():
    loop = asyncio.new_event_loop()
    asyncio.set_event_loop(loop)
    [...]

关于python - Python 3异步: run_until_complete() blocks when waiting for ProcessPoolExecutor job done,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/45303040/

10-11 22:32