调用asyncio.to_thread后上下文依然一致吗

使用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的工作放在里面跑。

你可能感兴趣的:(python,asyncio)