Получение фактической ссылки на конечный автомат задачи

Question

Я играю с идеей вмешательства в состояние состояния внутреннего конечного автомата, но у меня возникли проблемы с поиском способа доступа к ссылке на конечный автомат в моей метод задачи.

class Test 
{ 
    async Task IndexAsync() 
    { 
     var nottheactualtype = GetType(); //This references the "Test" class, but this operation is actually located in the nested state machine class named "IndexAsync", in the method "MoveNext()". 
     var actualcalledmethod = new StackTrace().GetFrame(0).GetMethod(); //This shows the actual method currently being run: IndexAsync.MoveNext(). 
     //But how do I get the reference to my current IndexAsync class? 
    } 
} 

Как я могу получить доступ к ссылке на сгенерированный конечный автомат, который в настоящее время запущен?

Answer 1

Вы можете принять первый вызов метода из своего класса на стек как вариант:

var nottheactualtype = GetType(); 
var actualcalledmethod = new StackTrace().GetFrames().FirstOrDefault(x => x.GetMethod().ReflectedType == nottheactualtype); 

Answer 2

Один из способов сделать это - передать ссылку или экземпляр вашего класса IndexAsync в класс Test.

Другой вариант - назначить его частному полю поддержки через отражение, хотя я бы предпочел первый вариант, так как отражение может быть медленным.

Если вы разделите больше своего кода, будет легче определить, как это сделать в вашем случае.

Answer 3

Это противно, и это не гарантирует работу (это зависит от деталей реализации) - но это работает для меня ... он в основном провоцирует конечный автомат, чтобы передать продолжение на awaiter. Затем мы можем вывести государственный автомат из цели делегата продолжения.

Гадкий, уродливый, некрасивый код ... но это не работа для меня :)

using System; 
using System.Reflection; 
using System.Threading.Tasks; 
using System.Runtime.CompilerServices; 
using static System.Reflection.BindingFlags; 

public class StateMachineProvider 
{ 
    private static readonly StateMachineProvider instance = new StateMachineProvider(); 

    public static StateMachineProvider GetStateMachine() => instance; 

    public StateMachineAwaiter GetAwaiter() => new StateMachineAwaiter(); 

    public class StateMachineAwaiter : INotifyCompletion 
    { 
     private Action continuation; 

     public bool IsCompleted => continuation != null; 

     public void OnCompleted(Action continuation) 
     { 
      this.continuation = continuation; 
      // Fire the continuation in a separate task. 
      // (We shouldn't just call it synchronously.) 
      Task.Run(continuation); 
     } 

     public IAsyncStateMachine GetResult() 
     { 
      var target = continuation.Target; 
      var field = target.GetType() 
           .GetField("m_stateMachine", NonPublic | Instance); 
      return (IAsyncStateMachine) field.GetValue(target); 
     } 
    } 
} 

class Test 
{ 
    static void Main() 
    { 
     AsyncMethod().Wait(); 
    } 

    static async Task AsyncMethod() 
    { 
     int x = 10; 
     IAsyncStateMachine machine = await StateMachineProvider.GetStateMachine(); 
     Console.WriteLine($"x={x}"); // Force the use across an await boundary 
     Console.WriteLine($"State machine type: {machine.GetType()})"); 
     Console.WriteLine("Fields:"); 
     var fields = machine.GetType().GetFields(Public | NonPublic | Instance); 
     foreach (var field in fields) 
     { 
      Console.WriteLine($"{field.Name}: {field.GetValue(machine)}"); 
     } 
    } 
} 

Выход:

x=10 
State machine type: Test+<AsyncMethod>d__1) 
Fields: 
<>1__state: -1 
<>t__builder: System.Runtime.CompilerServices.AsyncTaskMethodBuilder 
<x>5__1: 10 
<machine>5__2: Test+<AsyncMethod>d__1 
<>s__3: 
<>s__4: System.Reflection.FieldInfo[] 
<>s__5: 6 
<field>5__6: System.Reflection.FieldInfo <field>5__6 
<>u__1: