| | | 1 | | using Elsa.Mediator.Contracts; |
| | | 2 | | using JetBrains.Annotations; |
| | | 3 | | using ActivityCompleted = Elsa.Workflows.Signals.ActivityCompleted; |
| | | 4 | | |
| | | 5 | | namespace Elsa.Workflows.Behaviors; |
| | | 6 | | |
| | | 7 | | /// <summary> |
| | | 8 | | /// Implements a behavior that invokes "child completed" callbacks on parent activities. |
| | | 9 | | /// </summary> |
| | | 10 | | [UsedImplicitly] |
| | | 11 | | public class ScheduledChildCallbackBehavior : Behavior |
| | | 12 | | { |
| | | 13 | | /// <inheritdoc /> |
| | 31419 | 14 | | public ScheduledChildCallbackBehavior(IActivity owner) : base(owner) |
| | | 15 | | { |
| | 31419 | 16 | | OnSignalReceived<ActivityCompleted>(OnActivityCompletedAsync); |
| | 31419 | 17 | | } |
| | | 18 | | |
| | | 19 | | private async ValueTask OnActivityCompletedAsync(ActivityCompleted signal, SignalContext context) |
| | | 20 | | { |
| | 12784 | 21 | | var activityExecutionContext = context.ReceiverActivityExecutionContext; |
| | 12784 | 22 | | var childActivityExecutionContext = context.SenderActivityExecutionContext; |
| | 12784 | 23 | | var childActivityNode = childActivityExecutionContext.ActivityNode; |
| | 12784 | 24 | | var callbackEntry = activityExecutionContext.WorkflowExecutionContext.PopCompletionCallback(activityExecutionCon |
| | | 25 | | |
| | 12784 | 26 | | if (callbackEntry?.CompletionCallback != null) |
| | | 27 | | { |
| | 2731 | 28 | | var completedContext = new ActivityCompletedContext(activityExecutionContext, childActivityExecutionContext, |
| | 2731 | 29 | | var tag = callbackEntry.Tag; |
| | 2731 | 30 | | completedContext.TargetContext.Tag = tag; |
| | | 31 | | |
| | 2731 | 32 | | var mediator = activityExecutionContext.GetRequiredService<IMediator>(); |
| | 2731 | 33 | | var invokingActivityCallbackNotification = new Notifications.InvokingActivityCallback(activityExecutionConte |
| | 2731 | 34 | | await mediator.SendAsync(invokingActivityCallbackNotification, context.CancellationToken); |
| | 2731 | 35 | | await callbackEntry.CompletionCallback(completedContext); |
| | 2731 | 36 | | } |
| | 12784 | 37 | | } |
| | | 38 | | } |