深究python中的asyncio库-线程池

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

python中的asyncio库-线程池" alt="深究python中的asyncio库-线程池" src="//www.dapan.cc/wp-content/uploads/2022/11/1668480937-a35a0460d362f6e.jpg">

多线程

defstart_loop(loop):
asyncio.set_event_loop(loop)
loop.run_forever()
defshutdown(loop):
loop.stop()
asyncdefb1():
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:awaitb1()
<Futureat0x107edf4e0state=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去停止事件循环