6

Fody を使用して、メソッドからスローされたすべての例外を共通の例外形式でラップしようとしています。

そこで、次のような必要なインターフェイス宣言とクラス実装を追加しました。

using System;
using System.Diagnostics;
using System.Reflection;
using System.Threading.Tasks;

[module: MethodDecorator]

public interface IMethodDecorator
{
  void Init(object instance, MethodBase method, object[] args);
  void OnEntry();
  void OnExit();
  void OnException(Exception exception);
  void OnTaskContinuation(Task t);
}


[AttributeUsage(
    AttributeTargets.Module |
    AttributeTargets.Method |
    AttributeTargets.Assembly |
    AttributeTargets.Constructor, AllowMultiple = true)]
public class MethodDecorator : Attribute, IMethodDecorator
{
  public virtual void Init(object instance, MethodBase method, object[] args) { }

  public void OnEntry()
  {
    Debug.WriteLine("base on entry");
  }

  public virtual void OnException(Exception exception)
  {
    Debug.WriteLine("base on exception");
  }

  public void OnExit()
  {
    Debug.WriteLine("base on exit");
  }

  public void OnTaskContinuation(Task t)
  {
    Debug.WriteLine("base on continue");
  }
}

そして、このようなドメイン実装

using System;
using System.Diagnostics;
using System.Linq;
using System.Reflection;
using System.Runtime.ExceptionServices;

namespace CC.Spikes.AOP.Fody
{
  public class FodyError : MethodDecorator
  {
    public string TranslationKey { get; set; }
    public Type ExceptionType { get; set; }

    public override void Init(object instance, MethodBase method, object[] args)
    {
      SetProperties(method);
    }

    private void SetProperties(MethodBase method)
    {
      var attribute = method.CustomAttributes.First(n => n.AttributeType.Name == nameof(FodyError));
      var translation = attribute
        .NamedArguments
        .First(n => n.MemberName == nameof(TranslationKey))
        .TypedValue
        .Value
          as string;

      var exceptionType = attribute
        .NamedArguments
        .First(n => n.MemberName == nameof(ExceptionType))
        .TypedValue
        .Value
          as Type;


      TranslationKey = translation;
      ExceptionType = exceptionType;
    }

    public override void OnException(Exception exception)
    {
      Debug.WriteLine("entering fody error exception");
      if (exception.GetType() != ExceptionType)
      {
        Debug.WriteLine("rethrowing fody error exception");
        //rethrow without losing stacktrace
        ExceptionDispatchInfo.Capture(exception).Throw();
      }

      Debug.WriteLine("creating new fody error exception");
      throw new FodyDangerException(TranslationKey, exception);

    }
  }

  public class FodyDangerException : Exception
  {
    public string CallState { get; set; }
    public FodyDangerException(string message, Exception error) : base(message, error)
    {

    }
  }
}

これは、同期コードではうまく機能します。ただし、非同期コードの場合、他のすべての IMethodDecorator が実行されても ( OnExit、 、 などOnTaskContinuation)、例外ハンドラーはスキップされます。

たとえば、次のテスト クラスを見てください。

public class FodyTestStub
{ 

  [FodyError(ExceptionType = typeof(NullReferenceException), TranslationKey = "EN_WHATEVER")]
  public async Task ShouldGetErrorAsync()
  {
    await Task.Delay(200);
    throw new NullReferenceException();
  }

  public async Task ShouldGetErrorAsync2()
  {
    await Task.Delay(200);
    throw new NullReferenceException();
  }
}

ShouldGetErrorAsync次の IL コードが生成されることがわかります。

// CC.Spikes.AOP.Fody.FodyTestStub
[FodyError(ExceptionType = typeof(NullReferenceException), TranslationKey = "EN_WHATEVER"), DebuggerStepThrough, AsyncStateMachine(typeof(FodyTestStub.<ShouldGetErrorAsync>d__3))]
public Task ShouldGetErrorAsync()
{
    MethodBase methodFromHandle = MethodBase.GetMethodFromHandle(methodof(FodyTestStub.ShouldGetErrorAsync()).MethodHandle, typeof(FodyTestStub).TypeHandle);
    FodyError fodyError = (FodyError)Activator.CreateInstance(typeof(FodyError));
    object[] args = new object[0];
    fodyError.Init(this, methodFromHandle, args);
    fodyError.OnEntry();
    Task task;
    try
    {
        FodyTestStub.<ShouldGetErrorAsync>d__3 <ShouldGetErrorAsync>d__ = new FodyTestStub.<ShouldGetErrorAsync>d__3();
        <ShouldGetErrorAsync>d__.<>4__this = this;
        <ShouldGetErrorAsync>d__.<>t__builder = AsyncTaskMethodBuilder.Create();
        <ShouldGetErrorAsync>d__.<>1__state = -1;
        AsyncTaskMethodBuilder <>t__builder = <ShouldGetErrorAsync>d__.<>t__builder;
        <>t__builder.Start<FodyTestStub.<ShouldGetErrorAsync>d__3>(ref <ShouldGetErrorAsync>d__);
        task = <ShouldGetErrorAsync>d__.<>t__builder.Task;
        fodyError.OnExit();
    }
    catch (Exception exception)
    {
        fodyError.OnException(exception);
        throw;
    }
    return task;
}

そしてShouldGetErrorAsync2生成します：

    // CC.Spikes.AOP.Fody.FodyTestStub
[DebuggerStepThrough, AsyncStateMachine(typeof(FodyTestStub.<ShouldGetErrorAsync2>d__4))]
public Task ShouldGetErrorAsync2()
{
    FodyTestStub.<ShouldGetErrorAsync2>d__4 <ShouldGetErrorAsync2>d__ = new FodyTestStub.<ShouldGetErrorAsync2>d__4();
    <ShouldGetErrorAsync2>d__.<>4__this = this;
    <ShouldGetErrorAsync2>d__.<>t__builder = AsyncTaskMethodBuilder.Create();
    <ShouldGetErrorAsync2>d__.<>1__state = -1;
    AsyncTaskMethodBuilder <>t__builder = <ShouldGetErrorAsync2>d__.<>t__builder;
    <>t__builder.Start<FodyTestStub.<ShouldGetErrorAsync2>d__4>(ref <ShouldGetErrorAsync2>d__);
    return <ShouldGetErrorAsync2>d__.<>t__builder.Task;
}

を呼び出すとShouldGetErrorAsync、Fody が呼び出しをインターセプトし、メソッド本体を try キャッチでラップします。ただし、メソッドが非同期の場合、fodyError.OnTaskContinuation(task)andfodyError.OnExit()がまだ呼び出されていても、catch ステートメントにヒットすることはありません。

一方、ShouldGetErrorAsyncIL にエラー処理ブロックがなくても、エラーは問題なく処理されます。

私の質問は、エラー ブロックを適切に挿入し、非同期エラーが傍受されるようにするために、Fody がどのように IL を生成する必要があるかということです。

これは、問題を再現するテストを含むレポです

4

2 に答える 2

1

「キックオフ」メソッドのコンテンツの周りにのみ try-catch を配置しています。これは、最初に再スケジュールが必要になる時点までしか保護されません (非同期メソッドが最初に実行されると、「キックオフ」メソッドは終了します)。再スケジュールする必要があるため、非同期メソッドが再開されたときにスタックに存在しません)。

IAsyncStateMachine.MoveNext()代わりに、ステート マシンで実装するメソッドを変更することを検討する必要があります。SetException(Exception)特に、非同期メソッド ビルダー ( AsyncVoidMethodBuilder、AsyncTaskMethodBuilderまたは)の呼び出しを探しAsyncTaskMethodBuilder<TResult>、例外を渡す直前にラップします。

于 2016-02-01T08:53:09.257 に答える
1

await確かに、非同期メソッドは単純に見えますよね? :) その抽象化にリークが見つかりました - メソッドは通常、最初のものが見つかるとすぐに戻りawait、例外ヘルパーはそれ以降の例外をインターセプトする方法がありません。

OnExceptionを実装し、メソッドからの戻り値を処理する必要があります。メソッドが返され、タスクが完了していない場合、タスクでエラーの継続を終了する必要があります。これは、例外を処理したい方法で処理する必要があります。Fody の連中はそれを考えました - それがそのOnTaskContinuation目的です。Task.Exceptionタスクに潜んでいる例外があるかどうかを確認し、必要に応じて処理する必要があります。

これは、ロギングなどを行っているときに例外を再スローしたい場合にのみ機能すると思います-例外を別のものに置き換えることはできません。あなたはそれをテストする必要があります:)

于 2016-02-01T11:37:33.993 に答える