protected override Task InterceptAsync(IInvocation invocation, Task proceed)
{
var scope = _methodIdentityProvider.BeginNextMethodIdentityScope(_mehodIdentityKey);
proceed.ContinueWith(t => scope.Dispose());
return(proceed);
}
public IMethodInvocationResult Intercept(ISyncMethodInvocation methodInvocation)
{
using (_methodIdentityProvider.BeginNextMethodIdentityScope(_mehodIdentityKey))
{
return(methodInvocation.InvokeNext());
}
}
public object Invoke(IInvocationInfo invocationInfo)
{
using (_methodIdentityProvider.BeginNextMethodIdentityScope(_mehodIdentityKey))
{
return(invocationInfo.Proceed());
}
}
public void Intercept(IInvocation invocation)
{
using (_methodIdentityProvider.BeginNextMethodIdentityScope(_mehodIdentityKey))
{
invocation.Proceed();
}
}
public IMethodReturn Invoke(IMethodInvocation input, GetNextInterceptionBehaviorDelegate getNext)
{
using (_methodIdentityProvider.BeginNextMethodIdentityScope(_mehodIdentityKey))
{
return(getNext().Invoke(input, getNext));
}
}
public Task <IMethodInvocationResult> InterceptAsync(IAsyncMethodInvocation methodInvocation)
{
var scope = _methodIdentityProvider.BeginNextMethodIdentityScope(_mehodIdentityKey);
var res = methodInvocation.InvokeNextAsync();
res.ContinueWith(t => scope.Dispose());
return(res);
}