private void AppCompletionCleanup(NativeActivityContext context, CompensationExtension compensationExtension, CompensationTokenData compensationToken)
{
Fx.Assert(compensationToken != null, "CompensationTokenData must be valid");
// Remove the token from the parent!
if (compensationToken.ParentCompensationId != CompensationToken.RootCompensationId)
{
CompensationTokenData parentToken = compensationExtension.Get(compensationToken.ParentCompensationId);
Fx.Assert(parentToken != null, "parentToken must be valid");
parentToken.ExecutionTracker.Remove(compensationToken);
}
else
{
// remove from workflow root...
CompensationTokenData parentToken = compensationExtension.Get(CompensationToken.RootCompensationId);
Fx.Assert(parentToken != null, "parentToken must be valid");
parentToken.ExecutionTracker.Remove(compensationToken);
}
compensationToken.RemoveBookmark(context, CompensationBookmarkName.Canceled);
compensationToken.RemoveBookmark(context, CompensationBookmarkName.Compensated);
// Remove the token from the extension...
compensationExtension.Remove(compensationToken.CompensationId);
}
private void OnCancellation(NativeActivityContext context, Bookmark bookmark, object value)
{
CompensationExtension compensationExtension = context.GetExtension <CompensationExtension>();
Fx.Assert(compensationExtension != null, "CompensationExtension must be valid");
long compensationId = (long)value;
Fx.Assert(compensationId != CompensationToken.RootCompensationId, "CompensationId must be passed when resuming the Completed bookmark");
CompensationTokenData compensationToken = compensationExtension.Get(compensationId);
Fx.Assert(compensationToken != null, "CompensationTokenData must be valid");
Fx.Assert(compensationToken.CompensationState == CompensationState.Canceling, "CompensationState should be in Canceling state");
if (TD.CompensationStateIsEnabled())
{
TD.CompensationState(compensationToken.DisplayName, compensationToken.CompensationState.ToString());
}
// remove bookmarks.
compensationToken.RemoveBookmark(context, CompensationBookmarkName.OnCompensation);
compensationToken.RemoveBookmark(context, CompensationBookmarkName.OnConfirmation);
_currentCompensationToken.Set(context, new CompensationToken(compensationToken));
if (CancellationHandler != null)
{
context.ScheduleActivity(CancellationHandler, new CompletionCallback(this.OnCancellationHandlerComplete), new FaultCallback(OnExceptionFromHandler));
}
else
{
if (compensationToken.ExecutionTracker.Count > 0)
{
context.ScheduleActivity(DefaultCompensation, new CompletionCallback(this.OnCompensationComplete));
}
else
{
this.InternalOnCompensationComplete(context, compensationExtension, compensationToken);
}
}
}
// Successfully received Confirmed response.
private void OnConfirmed(NativeActivityContext context, Bookmark bookmark, object value)
{
CompensationExtension compensationExtension = context.GetExtension <CompensationExtension>();
Fx.Assert(compensationExtension != null, "CompensationExtension must be valid");
CompensationToken compensationToken = Target.Get(context);
Fx.Assert(compensationToken != null, "compensationToken must be valid");
// The compensationToken should be a valid one at this point. Ensure its validated in Confirm activity.
CompensationTokenData tokenData = compensationExtension.Get(compensationToken.CompensationId);
Fx.Assert(tokenData != null, "The compensationToken should be a valid one at this point. Ensure its validated in Confirm activity.");
tokenData.CompensationState = CompensationState.Confirmed;
if (TD.CompensationStateIsEnabled())
{
TD.CompensationState(tokenData.DisplayName, tokenData.CompensationState.ToString());
}
// Remove the token from the parent!
if (tokenData.ParentCompensationId != CompensationToken.RootCompensationId)
{
CompensationTokenData parentToken = compensationExtension.Get(tokenData.ParentCompensationId);
Fx.Assert(parentToken != null, "parentToken must be valid");
parentToken.ExecutionTracker.Remove(tokenData);
}
else
{
// remove from workflow root...
CompensationTokenData parentToken = compensationExtension.Get(CompensationToken.RootCompensationId);
Fx.Assert(parentToken != null, "parentToken must be valid");
parentToken.ExecutionTracker.Remove(tokenData);
}
tokenData.RemoveBookmark(context, CompensationBookmarkName.Confirmed);
// Remove the token from the extension...
compensationExtension.Remove(compensationToken.CompensationId);
}
