如何在.NET中创建跨执行上下文以保存关联ID等信息

你的问题核心在于AsyncLocal的初始化逻辑和引用类型共享导致的上下文冲突：

• 原代码中AsyncLocal<Dictionary<string, string>>初始化时直接赋值了同一个Dictionary实例，所有异步分支默认继承这个引用类型实例，多线程并行处理时不同任务会共用同一个字典，造成CorrelationId互相覆盖。
• ContextHolder的??=逻辑仅在Value为null时创建新字典，但异步分支会继承父上下文的Value引用，不会触发新实例创建。
• ResetContext调用Clear()会清空共享字典，进一步加剧上下文混乱。

public static class InternalContext
{
    private static readonly AsyncLocal<Dictionary<string, string>> _internalContextHolder = new();

    private static Dictionary<string, string> ContextHolder
    {
        get
        {
            // 确保每个异步/线程上下文拥有独立字典实例，避免继承父上下文引用
            var context = _internalContextHolder.Value;
            if (context == null)
            {
                context = new Dictionary<string, string>();
                _internalContextHolder.Value = context;
            }
            return context;
        }
    }

    public const string CORRELATION_ID_KEY = "correlationId";

    public static string CorrelationId
    {
        get
        {
            if (!ContextHolder.TryGetValue(CORRELATION_ID_KEY, out var currentValue))
            {
                currentValue = Guid.NewGuid().ToString();
                ContextHolder[CORRELATION_ID_KEY] = currentValue;
            }
            return currentValue;
        }
        set
        {
            if (value == null)
            {
                ContextHolder.Remove(CORRELATION_ID_KEY);
            }
            else
            {
                ContextHolder[CORRELATION_ID_KEY] = value;
            }
        }
    }

    public static void ResetContext()
    {
        // 直接置空，后续访问时自动创建全新实例，彻底隔离上下文状态
        _internalContextHolder.Value = null;
    }
}

• 移除AsyncLocal初始化时的默认Value赋值，避免所有上下文共享同一字典。
• 调整ContextHolder获取逻辑：当前上下文无字典实例时，创建新实例并赋值给AsyncLocal.Value，确保每个异步/线程上下文拥有独立存储。
• CorrelationId的setter处理null值时改为移除键，避免字典残留无效键值。
• ResetContext直接将AsyncLocal.Value设为null，后续访问时自动生成新字典，彻底避免上下文污染。

并行处理SQS消息时，只需确保每个任务内的上下文操作被正确隔离：

public virtual async Task ExecuteAsync(CancellationToken stoppingToken)
{
    IEnumerable<QueueMessage> messagesList = await GetMessages();

    var tasks = messagesList.Select(msg => Task.Run(async () =>
    {
        try
        {
            InternalContext.CorrelationId = msg.MessageAttributes["correlationId"];
            // 执行需要传递CorrelationId的HttpClient调用等操作
        }
        finally
        {
            // 确保无论是否异常都重置上下文
            InternalContext.ResetContext();
        }
    }, stoppingToken));
    
    await Task.WhenAll(tasks);
}

该方案可直接适配以下场景：

• HTTP请求：在请求入口（如中间件）设置CorrelationId，整个请求生命周期内的异步操作可安全获取。
• 消息队列（SQS/RabbitMQ等）：消费消息时读取并设置消息携带的CorrelationId，处理流程自动传递。
• SNS主题：发布消息时将当前上下文的CorrelationId放入消息属性，订阅端消费时读取并设置到上下文。

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