如何验证AsyncLocal<T>处于同一异步上下文？替代ThreadLocal的绑定方案

首先明确你的核心需求：要让StorageBookmark的Dispose操作和创建时处于同一个异步上下文，保证AsyncLocal<T>.Value指向的对象和创建书签时一致，就像之前用Thread.CurrentThread绑定线程的逻辑那样。

核心方案：基于ExecutionContext实现上下文绑定 #

AsyncLocal<T>的上下文流转完全依赖ExecutionContext——这是.NET异步编程中上下文传递的核心机制，比SynchronizationContext的适用范围更广：因为不是所有异步场景都有自定义同步上下文（比如控制台应用、线程池后台线程），但ExecutionContext在所有异步流转中都会被自动捕获和传递，能覆盖所有场景。

具体实现步骤 #

1. 创建StorageBookmark时，捕获当前的ExecutionContext快照
2. 在Dispose时，对比当前ExecutionContext和快照是否一致
   注：ExecutionContext本身是不可变对象，捕获后不会被修改，所以可以安全地作为上下文标识。

修改后的代码示例 #

class Storage<T> { 
    private AsyncLocal<ImmutableStack<T>> stackHolder = new AsyncLocal<ImmutableStack<T>>(); 
    
    public IDisposable Push(T item) { 
        var bookmark = new StorageBookmark<T>(this); 
        // 初始化默认栈（避免AsyncLocal初始值为null）
        stackHolder.Value = stackHolder.Value ?? ImmutableStack<T>.Empty;
        stackHolder.Value = stackHolder.Value.Push(item); 
        return bookmark; 
    } 
    
    private class StorageBookmark<TInner> : IDisposable { 
        private Storage<TInner> owner; 
        private ImmutableStack<TInner> snapshot; 
        private ExecutionContext boundContext; 
        
        public StorageBookmark(Storage<TInner> owner) { 
            this.owner = owner; 
            this.snapshot = owner.stackHolder.Value ?? ImmutableStack<T>.Empty; 
            // 捕获当前执行上下文快照
            this.boundContext = ExecutionContext.Capture(); 
        } 
        
        public void Dispose() { 
            // 检查当前上下文是否与绑定的上下文匹配
            if (!ExecutionContext.Equals(ExecutionContext.Capture(), boundContext)) { 
                throw new InvalidOperationException("Bookmark crossed async context boundary"); 
            } 
            owner.stackHolder.Value = snapshot; 
        } 
    } 
}

关于SynchronizationContext的局限性 #

SynchronizationContext.Current确实能在部分场景（比如UI线程）标识上下文，但它有明显的短板：

• 无同步上下文场景失效：控制台应用、线程池线程等环境中SynchronizationContext.Current为null，无法区分不同的异步流。
• 同步上下文可能被替换：使用ConfigureAwait(false)等操作会改变当前同步上下文，导致原本的标识逻辑失效。

所以优先选择ExecutionContext，它能覆盖所有异步场景，是更可靠的上下文标识方案。

潜在陷阱提醒 #

1. 捕获时机必须准确：必须在创建书签的同一异步流中捕获ExecutionContext，不能延迟到其他线程或异步分支，否则会绑定到错误的上下文。
2. 注意AsyncLocal的初始值：AsyncLocal<T>默认值为null，使用时要记得初始化默认栈（比如示例中的?? ImmutableStack<T>.Empty），避免空引用异常。
3. 不可变性保证安全：ExecutionContext是不可变对象，对比时直接用ExecutionContext.Equals即可，不用担心上下文被篡改。

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