public Task Do(Action actionToRun)
{
TaskCompletionSource <object> tcs = new TaskCompletionSource <object>();
m_ActorEngine.Post(() => {
try
{
actionToRun();
tcs.SetResult(1);
return(Task.CompletedTask);
}
catch (Exception e)
{
tcs.SetException(e);
return(tcs.Task);
}
});
return(tcs.Task.AsNonReturningTask());
}
public void Intercept(IInvocation invocation)
{
bool containsReturnValue = invocation.Method.ReturnType.IsConstructedGenericType;
var tcs = new TaskCompletionSource <object>();
m_ActorEngine.Post(() =>
{
try
{
// when target method throws an exception TargetInvocationException will be invoked!
var taskReturned = (Task)invocation.Method.Invoke(m_T, invocation.Arguments);
if (taskReturned.IsFaulted)
{
tcs.SetException(taskReturned.Exception);
}
taskReturned.ContinueWith(previous =>
{
if (previous.IsFaulted)
{
tcs.SetException(previous.Exception);
}
else if (previous.IsCanceled)
{
tcs.SetCanceled();
}
else if (previous.IsCompleted)
{
tcs.SetResult(containsReturnValue ? ((dynamic)previous).Result :"No result");
}
else
{
tcs.SetException(previous.Exception);
}
});
}
catch (TargetInvocationException ite)
{
tcs.SetException(ite.InnerException);
}
return(tcs.Task);
});
if (containsReturnValue)
{
invocation.ReturnValue = tcs.Task.ChangeTaskType(invocation.Method.ReturnType);
}
else
{
invocation.ReturnValue = tcs.Task.AsNonReturningTask();
}
}