Django异步代码中替换thread_locals逻辑的可行方案咨询

1. 直接传递请求上下文参数 #

这是最直观可靠的方案，虽然改动量可能较大：

• 将原本通过thread_locals获取的请求关联ID、缓存信息等，直接作为参数传递给需要的函数、方法和类
• 同步和异步代码通用，没有隐式依赖，调试和维护更清晰
• 缺点：若项目中依赖thread_locals的代码量大，需要逐一梳理并修改调用链路，初期成本较高

2. 基于Redis的分布式上下文存储 #

针对你考虑的Redis方案，键管理的具体思路如下：

• 生成唯一请求ID：在请求入口（如全局中间件）生成UUID作为请求唯一标识，同步请求存在request.request_id，异步请求存在request.scope['request_id']中
• 统一键命名规则：用请求ID作为Redis键的核心标识，比如格式为req_ctx:{request_id}，所有请求相关缓存信息可存在该键对应的Hash结构或JSON值中
• 上下文传递：sync_to_async/async_to_sync调用时，请求对象（或请求ID）会被适配器自动传递，因此同步/异步代码都能通过请求对象拿到ID，进而访问Redis中的上下文数据
• 过期策略：为每个Redis键设置与请求超时匹配的过期时间（如30秒），避免无效数据堆积

示例代码片段：

import uuid
from django.core.cache import cache

class RequestIDMiddleware:
    def __init__(self, get_response):
        self.get_response = get_response

    def __call__(self, request):
        request.request_id = str(uuid.uuid4())
        # 初始化空上下文（可选）
        cache.set(f"req_ctx:{request.request_id}", {}, timeout=30)
        response = self.get_response(request)
        # 请求结束后主动清理（可选，依赖过期策略也可省略）
        cache.delete(f"req_ctx:{request.request_id}")
        return response

# 业务代码中获取上下文
def get_request_context(request):
    return cache.get(f"req_ctx:{request.request_id}", {})

3. 自定义异步上下文管理器 #

针对异步代码占比较高的场景，可封装上下文管理器存储请求信息：

• 利用Python的contextlib.asynccontextmanager定义异步上下文，在请求处理时注入上下文数据
• 同步代码通过sync_to_async调用时，Django会自动复制当前线程的上下文到协程中
• 适合局部业务流程的上下文管理，避免全局上下文污染

示例代码片段：

from contextlib import asynccontextmanager
import contextvars

request_ctx = contextvars.ContextVar("request_ctx", default={})

@asynccontextmanager
async def request_context(request):
    token = request_ctx.set({"request_id": request.request_id, "user_id": request.user.id})
    try:
        yield
    finally:
        request_ctx.reset(token)

# 异步视图中使用
async def async_view(request):
    async with request_context(request):
        # 业务逻辑中获取上下文
        ctx = request_ctx.get()
        # ...业务处理

4. 扩展Django请求对象属性 #

利用Django原生请求对象直接存储上下文：

• 同步请求：直接给request添加自定义属性，如request.custom_cache = {}
• 异步请求：往request.scope中添加自定义字段，如request.scope['custom_cache'] = {}
• sync_to_async和async_to_sync适配器会自动传递请求对象，因此同步/异步代码都能访问这些属性
• 优点：无需额外依赖，利用Django原生机制，性能无损耗
• 缺点：上下文仅存活于当前请求生命周期，无法跨进程共享

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