Question

我正在研究一个Python-3程序，该程序试图做两件事：（1）从外部网络套接字（无阻塞）读取数据（类型1），并且（2）在常规UDP套接字（非阻塞）上接收数据（类型2）

在很长一段时间内，websocket和UDP套接字上都没有数据。因此，我试图使两种数据类型的读取/接收都无阻塞。我正在尝试使用Asyncio和Websockets对websocket执行此操作。

不幸的是，只要websocket上没有数据（类型1），以下代码就会挂起。它阻止其余代码执行。我在做什么错了？

预先感谢所有帮助。

import asyncio
import websockets
import socket

IP_STRATUX =    "ws://192.168.86.201/traffic"

# Method to retrieve data (Type 1) from websocket
async def readWsStratux(inURL):
    async with websockets.connect(inURL, close_timeout=0.1) as websocket:
        try:
            data = await websocket.recv()
            return data
        except websocket.error:
            return None

if __name__ == "__main__":
    # Socket to receive data (Type 2)
    sockPCC = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    sockPCC.bind((PCC_IP, PCC_PORT))
    sockPCC.setblocking(0)

    while True:
        print('-----MAIN LOOP-----')
        data1 = asyncio.get_event_loop().run_until_complete(
            readWsStratux(IP_STRATUX))
        print(f'Data 1: {data1}')
        data2, addr = sockPCC.recvfrom(1024)
        print(f'Data 2: {data2}')

Answer 1

问题在于run_until_complete会执行所说的操作，然后运行提供的协程直到返回。您需要创建一个协程，派生两个独立的任务，每个任务都在“后台”运行其自己的协程。一个任务将处理从websocket的读取，另一个任务将处理UDP数据。这两个协程都可以填充您的主协程读取的队列。

websocket协程看起来与您已经拥有的协程非常相似，但是将无限循环推入协程，并将数据传输到调用方提供的队列中：

async def readWsStratux(inURL, queue):
    while True:
        async with websockets.connect(inURL, close_timeout=0.1) as ws:
            try:
                data = await ws.recv()
                await queue.put(('websocket', data))
            except websockets.error:
                return None

接下来，您将需要类似的协程来执行UDP。与其手动创建一个非阻塞套接字，不如使用asyncio的support for UDP。您可以从文档中以example class的简化版本开始：

class ClientProtocol:
    def __init__(self, queue):
        self.queue = queue

    def datagram_received(self, data, addr):
        self.queue.put_nowait(('udp', data))

    def connection_lost(self, exc):
        self.queue.put_nowait(('udp', b''))

async def read_udp(queue):
    transport, protocol = await loop.create_datagram_endpoint(
        lambda: ClientProtocol(queue),
        remote_addr=(PCC_IP, PCC_PORT))
    # wait until canceled
    try:
        await asyncio.get_event_loop().create_future()
    except asyncio.CancelledError:
        transport.close()
        raise

有了这两个选项，您就可以编写主要的协程以生成它们，并在它们运行时从队列中收集数据：

async def read_both(in_url):
    queue = asyncio.Queue()
    # spawn two workers in parallel, and have them send
    # data to our queue
    ws_task = asyncio.create_task(readWsStratux(in_url, queue))
    udp_task = asyncio.create_task(read_udp(queue))

    while True:
        source, data = await queue.get()
        if source == 'ws':
            print('from websocket', data)
        elif source == 'udp':
            if data == b'':
                break  # lost the UDP connection
            print('from UDP', data)

    # terminate the workers
    ws_task.cancel()
    udp_task.cancel()

您的主程序现在包括对read_both的简单调用：

if __name__ == "__main__":
    asyncio.get_event_loop().run_until_complete(read_both(IP_STRATUX))

请注意，上面的代码未经测试，可能包含错字。

使用Asyncio的无阻塞Websocket接收

1 个答案: