private IAsyncResult BeginFinalWork(object sender, EventArgs e, AsyncCallback cb, object extradata)
{
var result = new StageAsyncResult(cb, extradata, () => { });
TaskCompletionSource <object> tcs = TakeLastCompletionSource();
if (tcs != null)
{
tcs.TrySetResult(null);
}
if (_state.OriginalTask != null)
{
_state.OriginalTask
.Then(() =>
{
_state.CallContext.OnEnd();
CallContextAsyncResult.End(_state.CallContext.AsyncResult);
result.TryComplete();
})
.Catch(error =>
{
result.Fail(ErrorState.Capture(error.Exception));
return(error.Handled());
});
}
else
{
result.TryComplete();
}
result.InitialThreadReturning();
return(result);
}
public IAsyncResult BeginEvent(object sender, EventArgs e, AsyncCallback cb, object extradata)
{
if (_result != null)
{
throw new InvalidOperationException();
}
if (_context.PreventNextStage)
{
var noop = new StageAsyncResult(cb, extradata, () => { });
noop.TryComplete();
noop.InitialThreadReturning();
return(noop);
}
_context.PreventNextStage = true;
_responseShouldEnd = true;
_context.PushExecutingStage(this);
Func <IDictionary <string, object>, Task> entryPoint = _stage.EntryPoint ?? _context.PrepareInitialContext((HttpApplication)sender);
IDictionary <string, object> environment = _context.TakeLastEnvironment();
TaskCompletionSource <object> tcs = _context.TakeLastCompletionSource();
var result = new StageAsyncResult(cb, extradata, () =>
{
var application = ((HttpApplication)sender);
if (_responseShouldEnd)
{
application.CompleteRequest();
}
});
_result = result;
environment[Constants.IntegratedPipelineCurrentStage] = _stage.Name;
try
{
entryPoint.Invoke(environment)
.CopyResultToCompletionSource(tcs, null)
.ContinueWith(t => result.TryComplete(), TaskContinuationOptions.ExecuteSynchronously)
.Catch(ci => ci.Handled());
}
catch (Exception ex)
{
result.Fail(ErrorState.Capture(ex));
tcs.TrySetException(ex);
return(result);
}
result.InitialThreadReturning();
return(result);
}
private async Task DoFinalWork(StageAsyncResult result)
{
try
{
await _state.OriginalTask;
_state.CallContext.OnEnd();
CallContextAsyncResult.End(_state.CallContext.AsyncResult);
result.TryComplete();
}
catch (Exception ex)
{
_state.CallContext.AbortIfHeaderSent();
result.Fail(ErrorState.Capture(ex));
}
}
private async Task DoFinalWork(StageAsyncResult result)
{
try
{
await _state.OriginalTask;
_state.CallContext.OnEnd();
CallContextAsyncResult.End(_state.CallContext.AsyncResult);
result.TryComplete();
}
catch (Exception ex)
{
_state.CallContext.AbortIfHeaderSent();
result.Fail(ErrorState.Capture(ex));
}
}
private IAsyncResult BeginFinalWork(object sender, EventArgs e, AsyncCallback cb, object extradata)
{
var result = new StageAsyncResult(cb, extradata, () => { });
TaskCompletionSource<object> tcs = TakeLastCompletionSource();
if (tcs != null)
{
tcs.TrySetResult(null);
}
if (_state.OriginalTask != null)
{
_state.OriginalTask
.Then(() =>
{
_state.CallContext.OnEnd();
CallContextAsyncResult.End(_state.CallContext.AsyncResult);
result.TryComplete();
})
.Catch(error =>
{
result.Fail(ErrorState.Capture(error.Exception));
return error.Handled();
});
}
else
{
result.TryComplete();
}
result.InitialThreadReturning();
return result;
}
public IAsyncResult BeginEvent(object sender, EventArgs e, AsyncCallback cb, object extradata)
{
if (_result != null)
{
throw new InvalidOperationException();
}
if (_context.PreventNextStage)
{
var noop = new StageAsyncResult(cb, extradata, () => { });
noop.TryComplete();
noop.InitialThreadReturning();
return noop;
}
_context.PreventNextStage = true;
_responseShouldEnd = true;
_context.PushExecutingStage(this);
Func<IDictionary<string, object>, Task> entryPoint = _stage.EntryPoint ?? _context.PrepareInitialContext((HttpApplication)sender);
IDictionary<string, object> environment = _context.TakeLastEnvironment();
TaskCompletionSource<object> tcs = _context.TakeLastCompletionSource();
var result = new StageAsyncResult(cb, extradata, () =>
{
var application = ((HttpApplication)sender);
if (_responseShouldEnd)
{
application.CompleteRequest();
}
});
_result = result;
environment[Constants.IntegratedPipelineCurrentStage] = _stage.Name;
try
{
entryPoint.Invoke(environment)
.CopyResultToCompletionSource(tcs, null)
.ContinueWith(t => result.TryComplete(), TaskContinuationOptions.ExecuteSynchronously)
.Catch(ci => ci.Handled());
}
catch (Exception ex)
{
result.Fail(ErrorState.Capture(ex));
tcs.TrySetException(ex);
return result;
}
result.InitialThreadReturning();
return result;
}