C#中如何判断方法是否处于异步执行流？实现IsInAsyncContext方法

要实现IsInAsyncContext方法识别程序是否处于异步执行流内，我们可以利用AsyncLocal<T>的异步上下文状态传递特性，结合显式标记异步上下文入口的方式完成，同时避免竞态条件。

实现思路 #

1. 用AsyncLocal<bool>跟踪当前上下文的异步状态，默认值为false（对应同步根上下文）。
2. 在异步方法的入口处显式将AsyncLocal的值设为true，由于AsyncLocal会在异步流中自动复制状态，每个异步调用都会拥有独立的状态实例，不会引发竞态条件。
3. IsInAsyncContext方法直接返回AsyncLocal的值，即可判断当前是否处于异步执行流。

代码实现 #

public static class AsyncContextDetector
{
    // AsyncLocal用于跟踪异步上下文状态，默认值为false
    private static readonly AsyncLocal<bool> _isInAsyncContext = new AsyncLocal<bool>();

    /// <summary>
    /// 判断当前是否处于异步执行流中
    /// </summary>
    /// <returns>true表示处于异步执行流，false表示处于同步根上下文</returns>
    public static bool IsInAsyncContext()
    {
        return _isInAsyncContext.Value;
    }

    /// <summary>
    /// 标记进入异步上下文，应在异步方法的开头调用
    /// </summary>
    public static void MarkAsAsyncContext()
    {
        _isInAsyncContext.Value = true;
    }
}

// 测试代码
public static class Program
{
    public static void Main()
    {
        Console.WriteLine(AsyncContextDetector.IsInAsyncContext()); // 输出: False
        AsyncMethod().GetAwaiter().GetResult();
        Console.WriteLine(AsyncContextDetector.IsInAsyncContext()); // 输出: False
    }

    private static async Task AsyncMethod()
    {
        AsyncContextDetector.MarkAsAsyncContext();
        Console.WriteLine(AsyncContextDetector.IsInAsyncContext()); // 输出: True
        SyncMethod();
        await Task.Delay(100);
        // await后仍处于异步上下文
        SyncMethod();
    }

    private static void SyncMethod()
    {
        Console.WriteLine(AsyncContextDetector.IsInAsyncContext()); 
        // 从AsyncMethod调用时输出True，从Main调用时输出False
    }
}

解决竞态条件的关键 #

你之前遇到的AsyncLocal竞态问题，根源是在同步根上下文（如Main）中设置了AsyncLocal的值，导致所有异步调用都继承了同一个状态实例。现在的方案仅在异步方法入口设置AsyncLocal的值：

• 每个异步调用启动时，AsyncLocal会创建独立的状态副本，并行的异步方法之间不会共享状态，彻底避免竞态条件。
• 同步根上下文的AsyncLocal值始终为默认的false，不会被异步流的状态污染。

拓展：自动标记异步上下文（可选） #

如果不想在每个异步方法中手动调用MarkAsAsyncContext，可以通过AOP（如AspectCore、PostSharp）实现自动拦截异步方法并标记上下文，示例伪代码如下：

// 自定义特性标记需要自动处理的异步方法
[AttributeUsage(AttributeTargets.Method)]
public class AsyncContextMarkerAttribute : Attribute
{
}

// AOP拦截器逻辑
public class AsyncContextInterceptor : IInterceptor
{
    public async Task Invoke(AspectContext context, AspectDelegate next)
    {
        AsyncContextDetector.MarkAsAsyncContext();
        await next(context);
    }
}

// 使用时只需给异步方法加特性
[AsyncContextMarker]
private static async Task AsyncMethod()
{
    // 无需手动调用MarkAsAsyncContext
    Console.WriteLine(AsyncContextDetector.IsInAsyncContext()); // 输出: True
}

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