| | | 1 | | using Elsa.Workflows.Management; |
| | | 2 | | using Elsa.Workflows.Management.Entities; |
| | | 3 | | using Elsa.Workflows.Management.Exceptions; |
| | | 4 | | using Elsa.Workflows.Management.Filters; |
| | | 5 | | using Elsa.Workflows.Management.Mappers; |
| | | 6 | | using Elsa.Workflows.Management.Options; |
| | | 7 | | using Elsa.Workflows.Activities; |
| | | 8 | | using Elsa.Workflows.ActivationValidators; |
| | | 9 | | using Elsa.Workflows.Models; |
| | | 10 | | using Elsa.Workflows.Options; |
| | | 11 | | using Elsa.Workflows.Runtime.Exceptions; |
| | | 12 | | using Elsa.Workflows.Runtime.Messages; |
| | | 13 | | using Elsa.Workflows.State; |
| | | 14 | | using Microsoft.Extensions.Logging; |
| | | 15 | | |
| | | 16 | | namespace Elsa.Workflows.Runtime; |
| | | 17 | | |
| | | 18 | | /// <summary> |
| | | 19 | | /// Represents a client for executing and managing local workflows. |
| | | 20 | | /// </summary> |
| | 242 | 21 | | public class LocalWorkflowClient( |
| | 242 | 22 | | string workflowInstanceId, |
| | 242 | 23 | | IWorkflowInstanceManager workflowInstanceManager, |
| | 242 | 24 | | IWorkflowDefinitionService workflowDefinitionService, |
| | 242 | 25 | | IWorkflowRunner workflowRunner, |
| | 242 | 26 | | IWorkflowCanceler workflowCanceler, |
| | 242 | 27 | | IWorkflowActivationGate workflowActivationGate, |
| | 242 | 28 | | WorkflowStateMapper workflowStateMapper, |
| | 242 | 29 | | ILogger<LocalWorkflowClient> logger) : IWorkflowClient |
| | | 30 | | { |
| | | 31 | | /// <summary> |
| | | 32 | | /// Retained for source compatibility. Hosts should prefer the constructor resolved by DI, |
| | | 33 | | /// which supplies the configured activation gate. |
| | | 34 | | /// </summary> |
| | | 35 | | public LocalWorkflowClient( |
| | | 36 | | string workflowInstanceId, |
| | | 37 | | IWorkflowInstanceManager workflowInstanceManager, |
| | | 38 | | IWorkflowDefinitionService workflowDefinitionService, |
| | | 39 | | IWorkflowRunner workflowRunner, |
| | | 40 | | IWorkflowCanceler workflowCanceler, |
| | | 41 | | WorkflowStateMapper workflowStateMapper, |
| | | 42 | | ILogger<LocalWorkflowClient> logger) |
| | 0 | 43 | | : this(workflowInstanceId, workflowInstanceManager, workflowDefinitionService, workflowRunner, workflowCanceler, |
| | | 44 | | { |
| | 0 | 45 | | } |
| | | 46 | | |
| | | 47 | | /// <inheritdoc /> |
| | 558 | 48 | | public string WorkflowInstanceId => workflowInstanceId; |
| | | 49 | | |
| | | 50 | | /// <inheritdoc /> |
| | | 51 | | public async Task<CreateWorkflowInstanceResponse> CreateInstanceAsync(CreateWorkflowInstanceRequest request, Cancell |
| | | 52 | | { |
| | 44 | 53 | | var workflowDefinitionHandle = request.WorkflowDefinitionHandle; |
| | 44 | 54 | | var workflowGraph = await GetWorkflowGraphAsync(workflowDefinitionHandle, cancellationToken); |
| | | 55 | | |
| | 42 | 56 | | await using var lease = await workflowActivationGate.EvaluateAsync(workflowGraph.Workflow, request.CorrelationId |
| | 42 | 57 | | if (!lease.CanStart) |
| | | 58 | | { |
| | 1 | 59 | | LogActivationDenial(workflowGraph.Workflow, request.CorrelationId); |
| | 1 | 60 | | return new() |
| | 1 | 61 | | { |
| | 1 | 62 | | CannotStart = true |
| | 1 | 63 | | }; |
| | | 64 | | } |
| | | 65 | | |
| | 41 | 66 | | var effectiveCancellationToken = lease.GetEffectiveCancellationToken(cancellationToken); |
| | 41 | 67 | | effectiveCancellationToken.ThrowIfCancellationRequested(); |
| | 40 | 68 | | var workflowInstance = CreateWorkflowInstance(workflowGraph.Workflow, request); |
| | 40 | 69 | | effectiveCancellationToken.ThrowIfCancellationRequested(); |
| | 40 | 70 | | await workflowInstanceManager.SaveAsync(workflowInstance, effectiveCancellationToken); |
| | 38 | 71 | | return new(); |
| | 39 | 72 | | } |
| | | 73 | | |
| | | 74 | | /// <inheritdoc /> |
| | | 75 | | public async Task<RunWorkflowInstanceResponse> RunInstanceAsync(RunWorkflowInstanceRequest request, CancellationToke |
| | | 76 | | { |
| | 87 | 77 | | var workflowInstance = await GetWorkflowInstanceAsync(cancellationToken); |
| | 87 | 78 | | return await RunInstanceAsync(workflowInstance, request, cancellationToken); |
| | 87 | 79 | | } |
| | | 80 | | |
| | | 81 | | /// <inheritdoc /> |
| | | 82 | | public async Task<RunWorkflowInstanceResponse> CreateAndRunInstanceAsync(CreateAndRunWorkflowInstanceRequest request |
| | | 83 | | { |
| | 141 | 84 | | var workflowDefinitionHandle = request.WorkflowDefinitionHandle; |
| | 141 | 85 | | var workflowGraph = await GetWorkflowGraphAsync(workflowDefinitionHandle, cancellationToken); |
| | | 86 | | |
| | 139 | 87 | | await using var lease = await workflowActivationGate.EvaluateAsync(workflowGraph.Workflow, request.CorrelationId |
| | 136 | 88 | | if (!lease.CanStart) |
| | | 89 | | { |
| | 8 | 90 | | LogActivationDenial(workflowGraph.Workflow, request.CorrelationId); |
| | 8 | 91 | | return new() |
| | 8 | 92 | | { |
| | 8 | 93 | | CannotStart = true |
| | 8 | 94 | | }; |
| | | 95 | | } |
| | | 96 | | |
| | 128 | 97 | | var effectiveCancellationToken = lease.GetEffectiveCancellationToken(cancellationToken); |
| | 128 | 98 | | effectiveCancellationToken.ThrowIfCancellationRequested(); |
| | 127 | 99 | | var workflowInstance = CreateWorkflowInstance(workflowGraph.Workflow, new CreateWorkflowInstanceRequest |
| | 127 | 100 | | { |
| | 127 | 101 | | WorkflowDefinitionHandle = workflowDefinitionHandle, |
| | 127 | 102 | | CorrelationId = request.CorrelationId, |
| | 127 | 103 | | Name = request.Name, |
| | 127 | 104 | | ParentId = request.ParentId, |
| | 127 | 105 | | Input = request.Input, |
| | 127 | 106 | | Properties = request.Properties |
| | 127 | 107 | | }); |
| | 127 | 108 | | effectiveCancellationToken.ThrowIfCancellationRequested(); |
| | | 109 | | |
| | | 110 | | // Do not durably publish a Running/Pending row before execution. If the run is |
| | | 111 | | // interrupted before WorkflowRunner commits its result, that row would occupy the |
| | | 112 | | // activation scope despite the failed activation. |
| | 127 | 113 | | return await RunInstanceAsync(workflowInstance, new() |
| | 127 | 114 | | { |
| | 127 | 115 | | Input = request.Input, |
| | 127 | 116 | | Variables = request.Variables, |
| | 127 | 117 | | Properties = request.Properties, |
| | 127 | 118 | | TriggerActivityId = request.TriggerActivityId, |
| | 127 | 119 | | ActivityHandle = request.ActivityHandle, |
| | 127 | 120 | | SchedulingActivityExecutionId = request.SchedulingActivityExecutionId, |
| | 127 | 121 | | SchedulingWorkflowInstanceId = request.SchedulingWorkflowInstanceId, |
| | 127 | 122 | | SchedulingCallStackDepth = request.SchedulingCallStackDepth, |
| | 127 | 123 | | IncludeWorkflowOutput = request.IncludeWorkflowOutput |
| | 127 | 124 | | }, effectiveCancellationToken); |
| | 133 | 125 | | } |
| | | 126 | | |
| | | 127 | | private void LogActivationDenial(Workflow workflow, string? correlationId) |
| | | 128 | | { |
| | 9 | 129 | | var identity = workflow.Identity; |
| | 9 | 130 | | logger.LogWarning( |
| | 9 | 131 | | "Workflow activation strategy {ActivationStrategyType} disallowed creating an instance for definition {Workf |
| | 9 | 132 | | workflow.Options.ActivationStrategyType?.FullName, |
| | 9 | 133 | | identity.DefinitionId, |
| | 9 | 134 | | identity.Version, |
| | 9 | 135 | | identity.Id, |
| | 9 | 136 | | !string.IsNullOrWhiteSpace(correlationId)); |
| | 9 | 137 | | } |
| | | 138 | | |
| | | 139 | | /// <inheritdoc /> |
| | | 140 | | public async Task CancelAsync(CancellationToken cancellationToken = default) |
| | | 141 | | { |
| | 2 | 142 | | var workflowInstance = await GetWorkflowInstanceAsync(cancellationToken); |
| | 2 | 143 | | await CancelAsync(workflowInstance, cancellationToken); |
| | 2 | 144 | | } |
| | | 145 | | |
| | | 146 | | private async Task CancelAsync(WorkflowInstance workflowInstance, CancellationToken cancellationToken) |
| | | 147 | | { |
| | 12 | 148 | | if (workflowInstance.Status != WorkflowStatus.Running) return; |
| | 6 | 149 | | var workflowGraph = await GetWorkflowGraphAsync(workflowInstance, cancellationToken); |
| | 6 | 150 | | var workflowState = await workflowCanceler.CancelWorkflowAsync(workflowGraph, workflowInstance.WorkflowState, ca |
| | 6 | 151 | | await workflowInstanceManager.SaveAsync(workflowState, cancellationToken); |
| | 9 | 152 | | } |
| | | 153 | | |
| | | 154 | | /// <inheritdoc /> |
| | | 155 | | public async Task<WorkflowState> ExportStateAsync(CancellationToken cancellationToken = default) |
| | | 156 | | { |
| | 71 | 157 | | var workflowInstance = await GetWorkflowInstanceAsync(cancellationToken); |
| | 71 | 158 | | return workflowInstance.WorkflowState; |
| | 71 | 159 | | } |
| | | 160 | | |
| | | 161 | | /// <inheritdoc /> |
| | | 162 | | public async Task ImportStateAsync(WorkflowState workflowState, CancellationToken cancellationToken = default) |
| | | 163 | | { |
| | 3 | 164 | | var workflowInstance = workflowStateMapper.Map(workflowState)!; |
| | 3 | 165 | | await workflowInstanceManager.SaveAsync(workflowInstance, cancellationToken); |
| | 3 | 166 | | } |
| | | 167 | | |
| | | 168 | | public Task<bool> InstanceExistsAsync(CancellationToken cancellationToken = default) |
| | | 169 | | { |
| | 0 | 170 | | return workflowInstanceManager.ExistsAsync(workflowInstanceId, cancellationToken); |
| | | 171 | | } |
| | | 172 | | |
| | | 173 | | /// <inheritdoc /> |
| | | 174 | | public async Task<bool> DeleteAsync(CancellationToken cancellationToken = default) |
| | | 175 | | { |
| | | 176 | | // Load the workflow instance (single DB call) |
| | 8 | 177 | | var workflowInstance = await TryGetWorkflowInstanceAsync(cancellationToken); |
| | 8 | 178 | | if (workflowInstance == null) |
| | 1 | 179 | | return false; |
| | | 180 | | |
| | 7 | 181 | | await CancelAsync(workflowInstance, cancellationToken); |
| | | 182 | | |
| | | 183 | | // Delete the workflow instance |
| | 7 | 184 | | var filter = new WorkflowInstanceFilter { Id = workflowInstanceId }; |
| | 7 | 185 | | await workflowInstanceManager.DeleteAsync(filter, cancellationToken); |
| | 7 | 186 | | return true; |
| | 8 | 187 | | } |
| | | 188 | | |
| | | 189 | | public async Task<RunWorkflowInstanceResponse> RunInstanceAsync(WorkflowInstance workflowInstance, RunWorkflowInstan |
| | | 190 | | { |
| | 214 | 191 | | cancellationToken.ThrowIfCancellationRequested(); |
| | 214 | 192 | | var workflowState = workflowInstance.WorkflowState; |
| | | 193 | | |
| | 214 | 194 | | if (workflowInstance.Status != WorkflowStatus.Running) |
| | | 195 | | { |
| | 0 | 196 | | logger.LogWarning("Attempt to resume workflow {WorkflowInstanceId} that is not in the Running state. The act |
| | 0 | 197 | | return new() |
| | 0 | 198 | | { |
| | 0 | 199 | | WorkflowInstanceId = WorkflowInstanceId, |
| | 0 | 200 | | Status = workflowInstance.Status, |
| | 0 | 201 | | SubStatus = workflowInstance.SubStatus |
| | 0 | 202 | | }; |
| | | 203 | | } |
| | | 204 | | |
| | 214 | 205 | | var runWorkflowOptions = new RunWorkflowOptions |
| | 214 | 206 | | { |
| | 214 | 207 | | Input = request.Input, |
| | 214 | 208 | | Variables = request.Variables, |
| | 214 | 209 | | Properties = request.Properties, |
| | 214 | 210 | | BookmarkId = request.BookmarkId, |
| | 214 | 211 | | TriggerActivityId = request.TriggerActivityId, |
| | 214 | 212 | | ActivityHandle = request.ActivityHandle, |
| | 214 | 213 | | SchedulingActivityExecutionId = request.SchedulingActivityExecutionId, |
| | 214 | 214 | | SchedulingWorkflowInstanceId = request.SchedulingWorkflowInstanceId, |
| | 214 | 215 | | SchedulingCallStackDepth = request.SchedulingCallStackDepth |
| | 214 | 216 | | }; |
| | | 217 | | |
| | 214 | 218 | | var workflowGraph = await GetWorkflowGraphAsync(workflowInstance, cancellationToken); |
| | 214 | 219 | | cancellationToken.ThrowIfCancellationRequested(); |
| | 214 | 220 | | var workflowResult = await workflowRunner.RunAsync(workflowGraph, workflowState, runWorkflowOptions, cancellatio |
| | | 221 | | |
| | 212 | 222 | | workflowState = workflowResult.WorkflowState; |
| | | 223 | | |
| | 212 | 224 | | return new() |
| | 212 | 225 | | { |
| | 212 | 226 | | WorkflowInstanceId = WorkflowInstanceId, |
| | 212 | 227 | | Status = workflowState.Status, |
| | 212 | 228 | | SubStatus = workflowState.SubStatus, |
| | 212 | 229 | | Incidents = workflowState.Incidents, |
| | 212 | 230 | | Output = request.IncludeWorkflowOutput ? new Dictionary<string, object>(workflowState.Output) : null, |
| | 212 | 231 | | Bookmarks = workflowState.Bookmarks |
| | 212 | 232 | | }; |
| | 212 | 233 | | } |
| | | 234 | | |
| | | 235 | | public async Task<WorkflowInstance> CreateInstanceInternalAsync(CreateWorkflowInstanceRequest request, CancellationT |
| | | 236 | | { |
| | 0 | 237 | | var workflowDefinitionHandle = request.WorkflowDefinitionHandle; |
| | 0 | 238 | | var workflowGraph = await GetWorkflowGraphAsync(workflowDefinitionHandle, cancellationToken); |
| | | 239 | | |
| | 0 | 240 | | return CreateWorkflowInstance(workflowGraph.Workflow, request); |
| | 0 | 241 | | } |
| | | 242 | | |
| | | 243 | | private WorkflowInstance CreateWorkflowInstance(Workflow workflow, CreateWorkflowInstanceRequest request) |
| | | 244 | | { |
| | 167 | 245 | | var options = new WorkflowInstanceOptions |
| | 167 | 246 | | { |
| | 167 | 247 | | WorkflowInstanceId = WorkflowInstanceId, |
| | 167 | 248 | | CorrelationId = request.CorrelationId, |
| | 167 | 249 | | Name = request.Name, |
| | 167 | 250 | | ParentWorkflowInstanceId = request.ParentId, |
| | 167 | 251 | | Input = request.Input, |
| | 167 | 252 | | Properties = request.Properties |
| | 167 | 253 | | }; |
| | | 254 | | |
| | 167 | 255 | | return workflowInstanceManager.CreateWorkflowInstance(workflow, options); |
| | | 256 | | } |
| | | 257 | | |
| | | 258 | | private async Task<WorkflowInstance> GetWorkflowInstanceAsync(CancellationToken cancellationToken) |
| | | 259 | | { |
| | 160 | 260 | | var workflowInstance = await TryGetWorkflowInstanceAsync(cancellationToken); |
| | 160 | 261 | | if (workflowInstance == null) throw new WorkflowInstanceNotFoundException("Workflow instance not found.", Workfl |
| | 160 | 262 | | return workflowInstance; |
| | 160 | 263 | | } |
| | | 264 | | |
| | | 265 | | private Task<WorkflowInstance?> TryGetWorkflowInstanceAsync(CancellationToken cancellationToken) |
| | | 266 | | { |
| | 168 | 267 | | return workflowInstanceManager.FindByIdAsync(WorkflowInstanceId, cancellationToken); |
| | | 268 | | } |
| | | 269 | | |
| | | 270 | | private async Task<WorkflowGraph> GetWorkflowGraphAsync(WorkflowInstance workflowInstance, CancellationToken cancell |
| | | 271 | | { |
| | 220 | 272 | | var handle = WorkflowDefinitionHandle.ByDefinitionVersionId(workflowInstance.DefinitionVersionId); |
| | 220 | 273 | | return await GetWorkflowGraphAsync(handle, cancellationToken); |
| | 220 | 274 | | } |
| | | 275 | | |
| | | 276 | | private async Task<WorkflowGraph> GetWorkflowGraphAsync(WorkflowDefinitionHandle definitionHandle, CancellationToken |
| | | 277 | | { |
| | 405 | 278 | | var result = await workflowDefinitionService.TryFindWorkflowGraphAsync(definitionHandle, cancellationToken); |
| | 407 | 279 | | if (!result.WorkflowDefinitionExists) throw new WorkflowDefinitionNotFoundException("Workflow definition not fou |
| | 405 | 280 | | if (!result.WorkflowGraphExists) throw new WorkflowMaterializerNotFoundException(result.WorkflowDefinition!.Mate |
| | 401 | 281 | | return result.WorkflowGraph!; |
| | 401 | 282 | | } |
| | | 283 | | |
| | | 284 | | private sealed class CompatibilityActivationGate : IWorkflowActivationGate |
| | | 285 | | { |
| | | 286 | | public Task<WorkflowActivationLease> EvaluateAsync(Workflow workflow, string? correlationId, CancellationToken c |
| | | 287 | | { |
| | 0 | 288 | | if (workflow.Options.ActivationStrategyType != null && workflow.Options.ActivationStrategyType != typeof(All |
| | 0 | 289 | | throw new InvalidOperationException("The compatibility LocalWorkflowClient constructor cannot enforce a |
| | | 290 | | |
| | 0 | 291 | | return Task.FromResult(new WorkflowActivationLease(true, null, cancellationToken)); |
| | | 292 | | } |
| | | 293 | | } |
| | | 294 | | } |