python - Asyncio - create_task 阻塞线程

Question

我正在尝试创建一个 Python 脚本，它将从 websocket 连接接收消息，并且每次收到新消息时，它都需要在后台运行 asyncio 任务。

为了“模拟”这个过程，我制作了一个阻塞函数，它用 while True 语句开始计数。预期的输出是，每次从 ws 连接收到新消息时，都会开始新的计数，但就我而言，一旦运行脚本，计数函数就会阻止整个代码。我该如何解决这个问题？

这是我尝试过的:

import asyncio
import websockets
import json
import time

#this is the blocking function..
def counter():
    count = 0
    while True:
        print(count)
        count += 1
        time.sleep(0.5)
    
async def main():
    while True:
        try:
            async with websockets.connect('MY-URL') as websocket:

                while True:
                    msg = await asyncio.wait_for(websocket.recv(), 500)
                   
                    try:
                        data = json.loads(msg)
                        await loop.create_task(counter())

                    except Exception as e:
                        print(e)
        
        except Exception as e:
            print(e)


loop = asyncio.get_event_loop()
loop.run_until_complete(main())

Answer 1

这里有两个主要问题。您的第一个问题是，当您尝试将其传递给 create_task 时，您在 counter 中创建了一个无限循环，然后调用它。这样 create_task 甚至不会被调用。 第二个明显的问题是，您尝试将一个方法传递给 create_task，而它需要一个协程。 使用 async def 再次将您的 counter 方法定义为协程，并将 time.sleep 替换为 asyncio.sleep 我认为它可能会起作用。

作为一般注意事项:您不能在与事件循环相同的线程中包含阻塞代码。这意味着永远不要在异步代码中使用 time.sleep...