如何在异步函数中更新ContextVar并让修改持久化？

我发现在异步函数中引用contextvars.ContextVar时，更新其值后修改无法持久化，像是操作了原变量的副本。请问有没有办法按引用传递上下文？或者是不是我用错工具了？

示例代码 #

import asyncio
from contextvars import ContextVar
import uuid
from fastapi import FastAPI
app = FastAPI()

req_context = ContextVar("req_context")

def do_something():
    print("uuid in do something: ", req_context.get())

async def do_something_async():
    req_context.set({
        **req_context.get(),
        "async1": "1",
    })
    print("uuid in do something async: ", req_context.get())

async def do_something_async2():
    req_context.set({
        **req_context.get(),
        "async2": "done",
    })
    print("uuid in do something async2: ", req_context.get())

@app.get("/")
async def root():
    print("=====\n\n0uuid: ", req_context.get({}))
    req_context.set({"REQ_ID": uuid.uuid4()})
    print("1uuid: ", req_context.get())
    do_something()
    await asyncio.gather(
        do_something_async(),
        do_something_async2(),
    )
    print("2uuid: ", req_context.get())
    return {"message": "Hello World"}

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=8000)

运行输出 #

0uuid:  {}
1uuid:  {'REQ_ID': UUID('e5efe0a0-5975-43b9-aff3-9e618ffb8be0')}
uuid in do something:  {'REQ_ID': UUID('e5efe0a0-5975-43b9-aff3-9e618ffb8be0')}
uuid in do something async:  {'REQ_ID': UUID('e5efe0a0-5975-43b9-aff3-9e618ffb8be0'), 'async1': '1'}
uuid in do something async2:  {'REQ_ID': UUID('e5efe0a0-5975-43b9-aff3-9e618ffb8be0'), 'async2': 'done'}

# Notice the mutations do not get saved to the original context

2uuid:  {'REQ_ID': UUID('e5efe0a0-5975-43b9-aff3-9e618ffb8be0')}

原因分析 #

ContextVar的核心设计是上下文隔离：当在异步子任务（比如asyncio.gather启动的任务）中调用set()时，修改的是当前子任务上下文内的变量副本，而非父任务的根上下文值。这是异步框架避免任务间状态污染的关键特性，并非工具用错。

你的代码中，do_something_async和do_something_async2是独立的异步任务，它们从root任务继承初始上下文，但各自的set()操作仅作用于自身任务的上下文，不会回溯修改父任务的上下文。

解决办法 #

1. 使用可变对象存储上下文 #

将ContextVar存储的不可变字典改为可变字典，直接修改字典内容而不调用set()，所有共享该上下文的任务都会看到修改：

import asyncio
from contextvars import ContextVar
import uuid
from fastapi import FastAPI
app = FastAPI()

req_context = ContextVar("req_context")

def do_something():
    print("uuid in do something: ", req_context.get())

async def do_something_async():
    ctx = req_context.get()
    ctx["async1"] = "1"  # 直接修改可变字典
    print("uuid in do something async: ", req_context.get())

async def do_something_async2():
    ctx = req_context.get()
    ctx["async2"] = "done"  # 直接修改可变字典
    print("uuid in do something async2: ", req_context.get())

@app.get("/")
async def root():
    print("=====\n\n0uuid: ", req_context.get({}))
    req_context.set({"REQ_ID": uuid.uuid4()})  # 存入可变字典
    print("1uuid: ", req_context.get())
    do_something()
    await asyncio.gather(
        do_something_async(),
        do_something_async2(),
    )
    print("2uuid: ", req_context.get())
    return {"message": "Hello World"}

2. 明确传递上下文结果 #

如果不需要全局共享，让子任务返回更新后的上下文，在父任务中合并后统一set()：

async def do_something_async():
    ctx = {**req_context.get(), "async1": "1"}
    print("uuid in do something async: ", ctx)
    return ctx

async def do_something_async2():
    ctx = {**req_context.get(), "async2": "done"}
    print("uuid in do something async2: ", ctx)
    return ctx

@app.get("/")
async def root():
    print("=====\n\n0uuid: ", req_context.get({}))
    req_context.set({"REQ_ID": uuid.uuid4()})
    print("1uuid: ", req_context.get())
    do_something()
    # 获取子任务返回的上下文
    result1, result2 = await asyncio.gather(
        do_something_async(),
        do_something_async2(),
    )
    # 合并结果（可根据需求调整合并策略）
    merged_ctx = {**req_context.get(), **result1, **result2}
    req_context.set(merged_ctx)
    print("2uuid: ", req_context.get())
    return {"message": "Hello World"}

工具选择建议 #

• 若需要异步任务间隔离的上下文，ContextVar是正确工具；
• 若需要异步任务间共享可修改状态，可变对象方案更合适；
• 若是FastAPI请求级上下文管理，也可通过依赖注入实现：

  from fastapi import Depends
  
  def get_req_context():
      ctx = req_context.get({"REQ_ID": uuid.uuid4()})
      req_context.set(ctx)
      return ctx
  
  @app.get("/")
  async def root(ctx: dict = Depends(get_req_context)):
      ctx["key"] = "value"
      return {"message": "Hello World"}
  

内容的提问来源于stack exchange，提问作者Jiew Meng