| | | 1 | | using Elsa.Extensions; |
| | | 2 | | using Elsa.Workflows.CommitStates; |
| | | 3 | | using Elsa.Workflows.Models; |
| | | 4 | | using Elsa.Workflows.Options; |
| | | 5 | | using Elsa.Workflows.Pipelines.WorkflowExecution; |
| | | 6 | | using Microsoft.Extensions.Options; |
| | | 7 | | |
| | | 8 | | namespace Elsa.Workflows.Middleware.Workflows; |
| | | 9 | | |
| | | 10 | | /// <summary> |
| | | 11 | | /// Installs middleware that executes scheduled work items. |
| | | 12 | | /// </summary> |
| | | 13 | | public static class UseActivitySchedulerMiddlewareExtensions |
| | | 14 | | { |
| | | 15 | | /// <summary> |
| | | 16 | | /// Installs middleware that executes scheduled work items. |
| | | 17 | | /// </summary> |
| | 505 | 18 | | public static IWorkflowExecutionPipelineBuilder UseDefaultActivityScheduler(this IWorkflowExecutionPipelineBuilder p |
| | | 19 | | } |
| | | 20 | | |
| | | 21 | | /// <summary> |
| | | 22 | | /// A workflow execution middleware component that executes scheduled work items. |
| | | 23 | | /// </summary> |
| | | 24 | | public class DefaultActivitySchedulerMiddleware(WorkflowMiddlewareDelegate next, IActivityInvoker activityInvoker, IComm |
| | | 25 | | { |
| | | 26 | | /// <inheritdoc /> |
| | | 27 | | public override async ValueTask InvokeAsync(WorkflowExecutionContext context) |
| | | 28 | | { |
| | | 29 | | var scheduler = context.Scheduler; |
| | | 30 | | |
| | | 31 | | context.TransitionTo(WorkflowSubStatus.Executing); |
| | | 32 | | await ConditionallyCommitStateAsync(context, WorkflowLifetimeEvent.WorkflowExecuting); |
| | | 33 | | |
| | | 34 | | while (scheduler.HasAny) |
| | | 35 | | { |
| | | 36 | | // Do not start a workflow if cancellation has been requested. |
| | | 37 | | if (context.CancellationToken.IsCancellationRequested) |
| | | 38 | | break; |
| | | 39 | | |
| | | 40 | | var currentWorkItem = scheduler.Take(); |
| | | 41 | | await ExecuteWorkItemAsync(context, currentWorkItem); |
| | | 42 | | } |
| | | 43 | | |
| | | 44 | | await Next(context); |
| | | 45 | | |
| | | 46 | | if (context.Status == WorkflowStatus.Running) |
| | | 47 | | context.TransitionTo(context.AllActivitiesCompleted() ? WorkflowSubStatus.Finished : WorkflowSubStatus.Suspe |
| | | 48 | | } |
| | | 49 | | |
| | | 50 | | private async Task ExecuteWorkItemAsync(WorkflowExecutionContext context, ActivityWorkItem workItem) |
| | | 51 | | { |
| | | 52 | | var options = new ActivityInvocationOptions |
| | | 53 | | { |
| | | 54 | | Owner = workItem.Owner, |
| | | 55 | | ExistingActivityExecutionContext = workItem.ExistingActivityExecutionContext, |
| | | 56 | | Tag = workItem.Tag, |
| | | 57 | | Variables = workItem.Variables, |
| | | 58 | | Input = workItem.Input |
| | | 59 | | }; |
| | | 60 | | |
| | | 61 | | await activityInvoker.InvokeAsync(context, workItem.Activity, options); |
| | | 62 | | } |
| | | 63 | | |
| | | 64 | | private async Task ConditionallyCommitStateAsync(WorkflowExecutionContext context, WorkflowLifetimeEvent lifetimeEve |
| | | 65 | | { |
| | | 66 | | var strategyName = context.Workflow.Options.CommitStrategyName; |
| | | 67 | | IWorkflowCommitStrategy? strategy = !string.IsNullOrWhiteSpace(strategyName) |
| | | 68 | | ? commitStrategyRegistry.FindWorkflowStrategy(strategyName) |
| | | 69 | | : commitStateOptions.Value.DefaultWorkflowCommitStrategy; |
| | | 70 | | |
| | | 71 | | if (strategy == null) |
| | | 72 | | return; |
| | | 73 | | |
| | | 74 | | var strategyContext = new WorkflowCommitStateStrategyContext(context, lifetimeEvent); |
| | | 75 | | var commitAction = strategy.ShouldCommit(strategyContext); |
| | | 76 | | |
| | | 77 | | if (commitAction is CommitAction.Commit) |
| | | 78 | | await context.CommitAsync(); |
| | | 79 | | } |
| | | 80 | | } |