Python 3.5 Asyncio 和多​​个 websocket 服务器

Question

我在 Ubuntu 上使用 Python websockets 4.0.1。我想运行 2 个 websocket 服务器。我能够通过为每个线程创建 2 个线程和独立的事件循环来完成“某种工作”。通过“某种工作”，我的意思是两个 websocket 都工作并且响应大约 30 秒，然后其中一个停止。我必须重新启动该过程才能让它们再次工作。如果我只运行这两个线程中的一个或另一个，则单个 websocket 将永远工作。

我做错了什么，如何让 2 个 websocket 与 asyncio 一起工作？谢谢。

# Start VL WebSocket Task
class vlWebSocketTask (threading.Thread):
    def __init__(self):
        threading.Thread.__init__(self)
    def run(self):
        # Main while loops
        loop = asyncio.new_event_loop()
        asyncio.set_event_loop(loop)
        while True:
            try:
                print("Starting VL WebSocket Server...")
                startVLServer = websockets.serve(vlWebsocketServer, '192.168.1.3', 8777)
                asyncio.get_event_loop().run_until_complete(startVLServer)  
                asyncio.get_event_loop().run_forever()
            except Exception as ex:
                print(ex)
            time.sleep(5)

# Start IR WebSocket Task
class irWebSocketTask (threading.Thread):
    def __init__(self):
        threading.Thread.__init__(self)
    def run(self):
        loop = asyncio.new_event_loop()
        asyncio.set_event_loop(loop)
        while True:
            try:
                print("Starting IR WebSocket Server...")
                startIRServer = websockets.serve(irWebsocketServer, '192.168.1.3', 8555)
                asyncio.get_event_loop().run_until_complete(startIRServer)  
                asyncio.get_event_loop().run_forever()
            except Exception as ex:
                print(ex)
            time.sleep(5)

# Initialize VL WebSocket Task
#VLWebSocketTask = vlWebSocketTask()
#VLWebSocketTask.start()

# Initialize IR WebSocket Task
IRWebSocketTask = irWebSocketTask()
IRWebSocketTask.start()

Answer 1

您不需要线程来运行多个 asyncio 任务 - 允许多个代理共享同一个事件循环是 asyncio 的强项。您应该能够用如下代码替换这两个基于线程的类：

loop = asyncio.new_event_loop()
loop.run_until_complete(websockets.serve(vlWebsocketServer, '192.168.1.3', 8777))
loop.run_until_complete(websockets.serve(irWebsocketServer, '192.168.1.3', 8555))
loop.run_forever()

虽然混合线程和 asyncio 并不是完全错误，但正确地这样做需要注意不要混淆单独的 asyncio 实例。将线程用于 asyncio 的安全方法是使用 loop.run_in_executor()，它在单独的线程中运行同步代码而不阻塞事件循环，同时从循环中返回对象 awaitable。

注意：以上代码是在asyncio.run() 出现之前编写的，并手动旋转事件循环。在 Python 3.7 及更高版本中，可能会编写如下内容：

async def main():
    server1 = await websockets.serve(vlWebsocketServer, '192.168.1.3', 8777)
    server2 = await websockets.serve(irWebsocketServer, '192.168.1.3', 8555)
    await asyncio.gather(server1.wait_closed(), server2.wait_closed())

asyncio.run(main())