.NET异步方法能否不创建嵌套ExecutionContext？或实现环境事务？

我有如下类，用于启动事务并将其存储为环境事务：

class AmbientTransaction
{
    public AsyncLocal<IDbTransaction> AmbientTransaction { get; } = new();

    public IDbTransaction BeginAndSetAmbientTransaction(IDbConnection connection)
    {
        var tx = connection.BeginTransaction();
        AmbientTransaction.Value = tx;
        return tx;
    }
}

同步方法BeginAndSetAmbientTransaction运行正常，但添加的异步方法无法工作：

// 此方法无法正常工作
public async Task<IDbTransaction> BeginAndSetAmbientTransactionAsync(IDbConnection connection)
{
    var tx = await connection.BeginTransactionAsync();
    AmbientTransaction.Value = tx;
    return tx;
}

原因是异步方法会创建嵌套的ExecutionContext，对AsyncLocal的修改在方法返回后无法保留到调用方的上下文中。

1. 理解AsyncLocal在异步场景中的行为 #

AsyncLocal的修改会和ExecutionContext绑定，异步方法中await操作会复制当前的ExecutionContext生成新的子上下文。你在await之后设置AsyncLocal.Value，这个修改只存在于当前方法后续的子上下文中，不会回传到调用方的原始上下文里——这就是异步方法失效的核心原因。

而同步方法能正常工作，是因为全程没有上下文切换，修改的就是调用方共享的同一个ExecutionContext。

2. 调整实现方式：用回调封装事务上下文 #

如果要让后续的异步操作能访问到这个环境事务，最可靠的方式是通过回调函数，把需要在事务上下文中执行的逻辑封装起来，确保所有相关代码都在设置了AsyncLocal的上下文中运行：

public async Task<T> ExecuteInAmbientTransactionAsync<T>(IDbConnection connection, Func<IDbTransaction, Task<T>> execute)
{
    using var transaction = await connection.BeginTransactionAsync();
    var originalTransaction = AmbientTransaction.Value;
    try
    {
        // 在执行回调前设置环境事务
        AmbientTransaction.Value = transaction;
        // 回调内的所有异步操作都会继承这个上下文
        var result = await execute(transaction);
        await transaction.CommitAsync();
        return result;
    }
    catch
    {
        await transaction.RollbackAsync();
        throw;
    }
    finally
    {
        // 恢复原来的环境事务，避免污染后续上下文
        AmbientTransaction.Value = originalTransaction;
    }
}

使用时直接把需要在事务中执行的逻辑传入回调：

var ambientTxManager = new AmbientTransaction();
await ambientTxManager.ExecuteInAmbientTransactionAsync(connection, async tx =>
{
    // 这里的所有异步操作都能通过AmbientTransaction.Value拿到当前事务
    await connection.ExecuteAsync("INSERT INTO ...", new { ... });
    return someResult;
});

3. 为什么不能直接修改调用方的上下文？ #

.NET的ExecutionContext设计为不可变的副本传递，异步方法内部的上下文是调用方上下文的复制体，修改副本不会影响原始上下文。所以无法做到让异步方法直接修改调用方的AsyncLocal值——必须通过回调的方式，让业务逻辑在同一个事务上下文内执行。

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