| | | 1 | | using Elsa.Extensions; |
| | | 2 | | using Elsa.Workflows.Activities.Flowchart.Models; |
| | | 3 | | |
| | | 4 | | namespace Elsa.Workflows.Activities.Flowchart.Extensions; |
| | | 5 | | |
| | | 6 | | public static class ActivityExecutionContextExtensions |
| | | 7 | | { |
| | | 8 | | private const string GraphTransientProperty = "FlowGraph"; |
| | | 9 | | |
| | | 10 | | extension(Activities.Flowchart flowchart) |
| | | 11 | | { |
| | | 12 | | public IActivity? GetStartActivity(string? triggerActivityId) |
| | | 13 | | { |
| | 1007 | 14 | | return flowchart.GetTriggerActivity(triggerActivityId) |
| | 1007 | 15 | | ?? flowchart.Start |
| | 1007 | 16 | | ?? flowchart.GetExplicitStartActivity() |
| | 1007 | 17 | | ?? flowchart.GetCanStartWorkflowActivity() |
| | 1007 | 18 | | ?? flowchart.GetRootActivity() |
| | 1007 | 19 | | ?? flowchart.Activities.FirstOrDefault(); |
| | | 20 | | } |
| | | 21 | | |
| | | 22 | | /// <summary> |
| | | 23 | | /// Gets the activity that was triggered by the specified trigger activity ID. |
| | | 24 | | /// </summary> |
| | | 25 | | public IActivity? GetTriggerActivity(string? triggerActivityId) |
| | | 26 | | { |
| | 1007 | 27 | | return triggerActivityId == null |
| | 1007 | 28 | | ? null : |
| | 1826 | 29 | | flowchart.Activities.FirstOrDefault(x => x.Id == triggerActivityId); |
| | | 30 | | } |
| | | 31 | | |
| | | 32 | | /// <summary> |
| | | 33 | | /// Gets the explicit Start activity from the flowchart. |
| | | 34 | | /// </summary> |
| | | 35 | | public IActivity? GetExplicitStartActivity() |
| | | 36 | | { |
| | 892 | 37 | | return flowchart.Activities.FirstOrDefault(x => x is Start); |
| | | 38 | | } |
| | | 39 | | |
| | | 40 | | /// <summary> |
| | | 41 | | /// Gets the first activity marked as "Can Start Workflow". |
| | | 42 | | /// </summary> |
| | | 43 | | public IActivity? GetCanStartWorkflowActivity() |
| | | 44 | | { |
| | 892 | 45 | | return flowchart.Activities.FirstOrDefault(x => x.GetCanStartWorkflow()); |
| | | 46 | | } |
| | | 47 | | } |
| | | 48 | | |
| | | 49 | | extension(ActivityExecutionContext context) |
| | | 50 | | { |
| | | 51 | | /// <summary> |
| | | 52 | | /// Checks if there is any pending work for the flowchart. |
| | | 53 | | /// </summary> |
| | | 54 | | public bool HasPendingWork() |
| | | 55 | | { |
| | 976 | 56 | | return context.HasRunningActivityInstances() |
| | 976 | 57 | | || context.HasScheduledWork() |
| | 976 | 58 | | || context.HasUnconsumedTokens() |
| | 976 | 59 | | || context.HasFaultedChildren(); |
| | | 60 | | } |
| | | 61 | | |
| | | 62 | | public FlowGraph GetFlowGraph() |
| | | 63 | | { |
| | | 64 | | // Store in TransientProperties so FlowChart is not persisted in WorkflowState |
| | 1007 | 65 | | var flowchart = (Activities.Flowchart)context.Activity; |
| | 1007 | 66 | | var startActivity = flowchart.GetStartActivity(context.WorkflowExecutionContext.TriggerActivityId); |
| | 1662 | 67 | | return context.TransientProperties.GetOrAdd(GraphTransientProperty, () => new FlowGraph(flowchart.Connection |
| | | 68 | | } |
| | | 69 | | |
| | | 70 | | public async Task CancelInboundAncestorsAsync(IActivity activity) |
| | | 71 | | { |
| | 31 | 72 | | if(context.Activity is not Activities.Flowchart) |
| | 0 | 73 | | throw new InvalidOperationException("Activity context is not a flowchart."); |
| | | 74 | | |
| | 31 | 75 | | var flowGraph = context.GetFlowGraph(); |
| | 31 | 76 | | var ancestorActivities = flowGraph.GetAncestorActivities(activity); |
| | 244 | 77 | | var inboundActivityExecutionContexts = context.WorkflowExecutionContext.ActivityExecutionContexts.Where(x => |
| | | 78 | | |
| | | 79 | | // Cancel each ancestor activity. |
| | 320 | 80 | | foreach (var activityExecutionContext in inboundActivityExecutionContexts) |
| | | 81 | | { |
| | 129 | 82 | | await activityExecutionContext.CancelActivityAsync(); |
| | | 83 | | } |
| | 31 | 84 | | } |
| | | 85 | | |
| | | 86 | | /// <summary> |
| | | 87 | | /// Checks if the flowchart has any unconsumed tokens. |
| | | 88 | | /// </summary> |
| | | 89 | | public bool HasUnconsumedTokens() |
| | | 90 | | { |
| | 655 | 91 | | var flowchart = (Activities.Flowchart)context.Activity; |
| | 896 | 92 | | return flowchart.GetTokenList(context).Any(x => x is { Consumed: false, Blocked: false }); |
| | | 93 | | } |
| | | 94 | | |
| | | 95 | | /// <summary> |
| | | 96 | | /// Checks if the flowchart has any running activity instances. |
| | | 97 | | /// </summary> |
| | | 98 | | public bool HasRunningActivityInstances() |
| | | 99 | | { |
| | 976 | 100 | | var flowchart = (Activities.Flowchart)context.Activity; |
| | 3103 | 101 | | var activityIds = flowchart.Activities.Select(x => x.Id).ToList(); |
| | 976 | 102 | | var children = context.Children; |
| | 4022 | 103 | | return children.Where(x => activityIds.Contains(x.Activity.Id)).Any(x => x.Status == ActivityStatus.Running) |
| | | 104 | | } |
| | | 105 | | |
| | | 106 | | /// <summary> |
| | | 107 | | /// Checks if the flowchart has any scheduled work items. |
| | | 108 | | /// </summary> |
| | | 109 | | public bool HasScheduledWork() |
| | | 110 | | { |
| | 976 | 111 | | var workflowExecutionContext = context.WorkflowExecutionContext; |
| | | 112 | | |
| | 976 | 113 | | return workflowExecutionContext.Scheduler.List().Any(workItem => |
| | 976 | 114 | | { |
| | 321 | 115 | | var ownerInstanceId = workItem.Owner?.Id; |
| | 976 | 116 | | |
| | 321 | 117 | | if (ownerInstanceId == null) |
| | 0 | 118 | | return false; |
| | 976 | 119 | | |
| | 321 | 120 | | if (ownerInstanceId == context.Id) |
| | 321 | 121 | | return true; |
| | 976 | 122 | | |
| | 0 | 123 | | var ownerContext = context.WorkflowExecutionContext.ActivityExecutionContexts.First(x => x.Id == ownerIn |
| | 0 | 124 | | var ancestors = ownerContext.GetAncestors().ToList(); |
| | 976 | 125 | | |
| | 0 | 126 | | return ancestors.Any(x => x == context); |
| | 976 | 127 | | }); |
| | | 128 | | } |
| | | 129 | | |
| | | 130 | | public bool HasFaultedChildren() |
| | | 131 | | { |
| | 1631 | 132 | | return context.Children.Any(x => x.Status == ActivityStatus.Faulted); |
| | | 133 | | } |
| | | 134 | | } |
| | | 135 | | } |