深究Python中的asyncio库线程池

  • Z时代
  • 2024-01-10
  • 分类：综合

python

在同步线程中使用的run_in_executor就如它方法的名字所示，把协程放到了一个执行器里面，可以在一个线程池，也可以在一个进程池。另外还可以使用run_coroutine_threadsafe在其他线程执行协程（这是线程安全的）。

多线程

def start_loop(loop):
    asyncio.set_event_loop(loop)
    loop.run_forever()
def shutdown(loop):
    loop.stop()
async def b1():
    new_loop = asyncio.new_event_loop()
    t = Thread(target=start_loop, args=(new_loop,))
    t.start()
    future = asyncio.run_coroutine_threadsafe(a(), new_loop)
    print(future)
    print(f'Result: {future.result(timeout=2)}')
    new_loop.call_soon_threadsafe(partial(shutdown, new_loop))
In : await b1()
<Future at 0x107edf4e0 state=pending>
Result: A

这里面有几个细节要注意:

协程应该从另一个线程中调用，而非事件循环运行所在线程，所以用asyncio.new_event_loop()新建一个事件循环

在执行协程前要确保新创建的事件循环是运行着的，所以需要用start_loop之类的方式启动循环

接着就可以用asyncio.run_coroutine_threadsafe执行协程a了，它返回了一个Future对象

可以通过输出感受到future一开始是pending的，因为协程a里面会sleep 1秒才返回结果

用future.result(timeout=2)就可以获得结果，设置timeout的值要大于a协程执行时间，要不然会抛出TimeoutError

一开始我们创建的新的事件循环跑在一个线程里面，由于loop.run_forever会阻塞程序关闭，所以需要结束时杀掉线程，所以用call_soon_threadsafe回调函数shutdown去停止事件循环

这里再说一下call_soon_threadsafe，看名字就知道它是线程安全版本的call_soon，其实就是在另外一个线程里面调度回调。BTW， 其实asyncio.run_coroutine_threadsafe底层也是用的它。

以上是 深究Python中的asyncio库线程池 的全部内容， 来源链接： utcz.com/z/521712.html

回到顶部