| | | 1 | | using System.Diagnostics.CodeAnalysis; |
| | | 2 | | using Elsa.Common; |
| | | 3 | | using Elsa.Common.Multitenancy; |
| | | 4 | | using Elsa.Mediator.Contracts; |
| | | 5 | | using Elsa.Workflows.Runtime.Models; |
| | | 6 | | using Elsa.Workflows.Runtime.Notifications; |
| | | 7 | | using Elsa.Workflows.Runtime.Options; |
| | | 8 | | using Elsa.Workflows.Runtime.Requests; |
| | | 9 | | using Elsa.Workflows.Runtime.Responses; |
| | | 10 | | using Microsoft.Extensions.DependencyInjection; |
| | | 11 | | using Microsoft.Extensions.Options; |
| | | 12 | | |
| | | 13 | | namespace Elsa.Workflows.Runtime; |
| | | 14 | | |
| | | 15 | | /// <summary> |
| | | 16 | | /// Writes workflow dispatch commands to a durable outbox while a workflow execution is in progress. |
| | | 17 | | /// </summary> |
| | 472 | 18 | | public class TransactionalWorkflowDispatcher( |
| | 472 | 19 | | IWorkflowDispatcher decoratedService, |
| | 472 | 20 | | IServiceProvider serviceProvider, |
| | 472 | 21 | | IWorkflowDispatchOutboxAccessor outboxAccessor, |
| | 472 | 22 | | INotificationSender notificationSender, |
| | 472 | 23 | | IIdentityGenerator identityGenerator, |
| | 472 | 24 | | IOptions<WorkflowDispatcherOptions> options, |
| | 472 | 25 | | ITenantAccessor? tenantAccessor = null) : IWorkflowDispatcher |
| | | 26 | | { |
| | | 27 | | /// <inheritdoc /> |
| | | 28 | | public async Task<DispatchWorkflowResponse> DispatchAsync(DispatchWorkflowDefinitionRequest request, DispatchWorkflo |
| | | 29 | | { |
| | 27 | 30 | | if (!TryGetWorkflowExecutionContext(out var context)) |
| | 27 | 31 | | return await decoratedService.DispatchAsync(request, dispatchOptions, cancellationToken); |
| | | 32 | | |
| | 0 | 33 | | await notificationSender.SendAsync(new WorkflowDefinitionDispatching(request), cancellationToken); |
| | 0 | 34 | | var generatedInstanceId = string.IsNullOrWhiteSpace(request.InstanceId) ? identityGenerator.GenerateId() : null; |
| | | 35 | | |
| | 0 | 36 | | await GetOutbox().EnqueueAsync(context, new() |
| | 0 | 37 | | { |
| | 0 | 38 | | TenantId = tenantAccessor?.Tenant?.Id, |
| | 0 | 39 | | Kind = WorkflowDispatchOutboxItemKind.WorkflowDefinition, |
| | 0 | 40 | | WorkflowDefinitionCommand = WorkflowDispatchCommandFactory.CreateCommand(request, generatedInstanceId) |
| | 0 | 41 | | }, cancellationToken); |
| | | 42 | | |
| | 0 | 43 | | var response = DispatchWorkflowResponse.Success(); |
| | 0 | 44 | | await notificationSender.SendAsync(new WorkflowDefinitionDispatched(request, response), cancellationToken); |
| | 0 | 45 | | return response; |
| | 27 | 46 | | } |
| | | 47 | | |
| | | 48 | | /// <inheritdoc /> |
| | | 49 | | public async Task<DispatchWorkflowResponse> DispatchAsync(DispatchWorkflowInstanceRequest request, DispatchWorkflowO |
| | | 50 | | { |
| | 1 | 51 | | if (!TryGetWorkflowExecutionContext(out var context)) |
| | 1 | 52 | | return await decoratedService.DispatchAsync(request, dispatchOptions, cancellationToken); |
| | | 53 | | |
| | 0 | 54 | | await notificationSender.SendAsync(new WorkflowInstanceDispatching(request), cancellationToken); |
| | | 55 | | |
| | 0 | 56 | | await GetOutbox().EnqueueAsync(context, new() |
| | 0 | 57 | | { |
| | 0 | 58 | | TenantId = tenantAccessor?.Tenant?.Id, |
| | 0 | 59 | | Kind = WorkflowDispatchOutboxItemKind.WorkflowInstance, |
| | 0 | 60 | | WorkflowInstanceCommand = WorkflowDispatchCommandFactory.CreateCommand(request) |
| | 0 | 61 | | }, cancellationToken); |
| | | 62 | | |
| | 0 | 63 | | var response = DispatchWorkflowResponse.Success(); |
| | 0 | 64 | | await notificationSender.SendAsync(new WorkflowInstanceDispatched(request, response), cancellationToken); |
| | 0 | 65 | | return response; |
| | 1 | 66 | | } |
| | | 67 | | |
| | | 68 | | /// <inheritdoc /> |
| | | 69 | | public async Task<DispatchWorkflowResponse> DispatchAsync(DispatchTriggerWorkflowsRequest request, DispatchWorkflowO |
| | | 70 | | { |
| | 0 | 71 | | if (!TryGetWorkflowExecutionContext(out var context)) |
| | 0 | 72 | | return await decoratedService.DispatchAsync(request, dispatchOptions, cancellationToken); |
| | | 73 | | |
| | 0 | 74 | | await GetOutbox().EnqueueAsync(context, new() |
| | 0 | 75 | | { |
| | 0 | 76 | | TenantId = tenantAccessor?.Tenant?.Id, |
| | 0 | 77 | | Kind = WorkflowDispatchOutboxItemKind.TriggerWorkflows, |
| | 0 | 78 | | TriggerWorkflowsCommand = WorkflowDispatchCommandFactory.CreateCommand(request) |
| | 0 | 79 | | }, cancellationToken); |
| | | 80 | | |
| | 0 | 81 | | return DispatchWorkflowResponse.Success(); |
| | 0 | 82 | | } |
| | | 83 | | |
| | | 84 | | /// <inheritdoc /> |
| | | 85 | | public async Task<DispatchWorkflowResponse> DispatchAsync(DispatchResumeWorkflowsRequest request, DispatchWorkflowOp |
| | | 86 | | { |
| | 0 | 87 | | if (!TryGetWorkflowExecutionContext(out var context)) |
| | 0 | 88 | | return await decoratedService.DispatchAsync(request, dispatchOptions, cancellationToken); |
| | | 89 | | |
| | 0 | 90 | | await GetOutbox().EnqueueAsync(context, new() |
| | 0 | 91 | | { |
| | 0 | 92 | | TenantId = tenantAccessor?.Tenant?.Id, |
| | 0 | 93 | | Kind = WorkflowDispatchOutboxItemKind.ResumeWorkflows, |
| | 0 | 94 | | ResumeWorkflowsCommand = WorkflowDispatchCommandFactory.CreateCommand(request) |
| | 0 | 95 | | }, cancellationToken); |
| | | 96 | | |
| | 0 | 97 | | return DispatchWorkflowResponse.Success(); |
| | 0 | 98 | | } |
| | | 99 | | |
| | | 100 | | private bool TryGetWorkflowExecutionContext([NotNullWhen(true)] out WorkflowExecutionContext? context) |
| | | 101 | | { |
| | 28 | 102 | | var currentContext = outboxAccessor.WorkflowExecutionContext; |
| | 28 | 103 | | if (!options.Value.UseTransactionalOutbox || currentContext == null) |
| | | 104 | | { |
| | 28 | 105 | | context = null; |
| | 28 | 106 | | return false; |
| | | 107 | | } |
| | | 108 | | |
| | 0 | 109 | | context = currentContext; |
| | 0 | 110 | | return true; |
| | | 111 | | } |
| | | 112 | | |
| | 0 | 113 | | private IWorkflowDispatchOutbox GetOutbox() => serviceProvider.GetRequiredService<IWorkflowDispatchOutbox>(); |
| | | 114 | | } |