与asyncio有两个无限任务

时间:2018-09-06 08:21:21

标签: python python-3.x multithreading python-asyncio

我是python的新手,我必须创建一个程序来阻止Web套接字和管道,因此我需要两个异步函数。 这些函数中的每一个都以不同的方式调用其他方法,以详细说明接收到的json的内容。 即我在套接字线程上收到一条消息,得到消息并抛出一个新线程来详细说明该消息。
这是实际的代码:

import asyncio
import sys
import json
import websockets

# Keep listening from web socket and pipe


async def socket_receiver():
    """Listening from web socket"""
    file_socket = open(r"SocketReceived.txt", "w")
    header = {"Authorization": r"Basic XXXXXXXXXXXXXX="}
    async with websockets.connect(
            'wss://XXXXXXXXX', extra_headers=header) as web_socket:
        print("SOCKET receiving:")
        greeting = await web_socket.recv()
        json_message = json.loads(greeting)
        file_socket.write(json_message)
        print(json_message)

    file_socket.close()

async def pipe_receiver():
    """Listening from pipe"""
    file_pipe = open(r"ipeReceived.txt", "w")
    while True:
        print("PIPE receiving:")
        line = sys.stdin.readline()
        if not line:
            break

        jsonObj = json.loads(line);
        file_pipe.write(jsonObj['prova'] + '\n')
        # jsonValue = json.dump(str(line), file);
        sys.stdout.flush()

    file_pipe.close()
asyncio.get_event_loop().run_until_complete(socket_receiver())
asyncio.get_event_loop().run_until_complete(pipe_receiver())

run_until_complete方法在我的情况下永远保持不变(它等待功能结束),因此只有套接字启动。 如何同时启动两者?谢谢

1 个答案:

答案 0 :(得分:2)

asyncio.gather 可以解决问题,唯一的一点是两个函数应该共享相同的事件循环,并且应该完全异步。

asyncio.get_event_loop().run_until_complete(
    asyncio.gather( socket_receiver(),pipe_receiver()))

通过快速阅读pipe_receiver,您将在sys.stdin.readline调用中挂起事件循环,请考虑使用aioconsole异步处理输入。