在调用 asyncio 的相关方法时需要注意线程安全,因为 asyncio 并不是线程安全的。在同一时间多个线程调用 asyncio 方法会导致竞争访问同一个事件循环,可能会发生错误。
call_soon 和 call_soon_threadsafe 都是在事件循环中注册回调函数,以便在尽可能快的时间内执行它们。它们最大的区别在于它们的线程安全性。call_soon 是线程不安全的,因为它假设是在主循环线程中被调用。任何在其他线程中调用 call_soon 的操作都可能与主循环的状态发生冲突,从而产生争用。
而 call_soon_threadsafe 将回调任务重新排队以便在 async 事件循环中执行,并确保了线程安全性。因此,可以在任何线程上调用它,而不会影响事件循环的状态。
下面是一个示例程序,说明如何使用 call_soon_threadsafe 方法:
import asyncio
async def my_task():
print('Task is running')
await asyncio.sleep(1)
print('Task is done')
def call_from_thread(loop):
print('Thread is calling the event loop')
loop.call_soon_threadsafe(asyncio.create_task, my_task())
if __name__ == '__main__':
loop = asyncio.get_event_loop()
threading.Thread(target=call_from_thread, args=(loop,)).start()
loop.run_forever()
在上面的代码中,我们启动了一个新线程来调用相关方法,该方法将任务添加到主线程事件循环中。由于使用了 call_soon_threadsafe 方法,因此我们可以通过该方法从任何线程安全地调用 asyncio 事件循环。