使用Python的asyncio时,可以把一个同步的函数放到线程池中执行从而避免这个函数阻塞asyncio自身的事件循环。比如可以把requests库的请求放进去
async def to_thread_do_request(url):
return await asyncio.to_thread(requests.get, url)
这个to_thread_do_request
方法就不会造成asyncio的阻塞,反而下面这样直接调用一个下面这样把requests.get简单包在一个async函数里则会造成的协程的阻塞
async def do_request(url):
return requests.get(url)
不过在使用asyncio.to_thread
函数时,我们需要注意一点,即asyncio的上下文,也就是通过contextvars来保存的内容通过asyncio.to_thread
来执行时与调用者否是一致的?
这个可以大致看一下asyncio.to_thread
的实现,我们会发现,对于context来说,这个to_thread
会先调用contextvars.copy_context
方法把当前的context整体复制到ctx
变量中,然后再通过ctx.run
来执行其实际任务。这样通过asyncio.to_thread
来执行的函数,与调用方的context是一致的,可以放心的把一些需要获取contextvars的工作放在里面跑。