| | | 1 | | using Elsa.Mediator.Contracts; |
| | | 2 | | using Elsa.Common.Multitenancy; |
| | | 3 | | using Elsa.Workflows.Management.Activities.WorkflowDefinitionActivity; |
| | | 4 | | using Elsa.Workflows.Management.Contracts; |
| | | 5 | | using Elsa.Workflows.Management.Entities; |
| | | 6 | | using Elsa.Workflows.Management.Notifications; |
| | | 7 | | using JetBrains.Annotations; |
| | | 8 | | |
| | | 9 | | namespace Elsa.Workflows.Management.Handlers.Notifications; |
| | | 10 | | |
| | | 11 | | /// <summary> |
| | | 12 | | /// Refreshes the <see cref="IActivityRegistry"/> for the <see cref="WorkflowDefinitionActivityProvider"/> provider when |
| | | 13 | | /// </summary> |
| | | 14 | | [PublicAPI] |
| | 405 | 15 | | public class RefreshActivityRegistry( |
| | 405 | 16 | | IWorkflowDefinitionActivityRegistryUpdater workflowDefinitionActivityRegistryUpdater, |
| | 405 | 17 | | IWorkflowDefinitionRegistryGenerationStore generationStore, |
| | 405 | 18 | | ITenantAccessor tenantAccessor) : |
| | | 19 | | INotificationHandler<WorkflowDefinitionPublished>, |
| | | 20 | | INotificationHandler<WorkflowDefinitionRetracted>, |
| | | 21 | | INotificationHandler<WorkflowDefinitionVersionRetracted>, |
| | | 22 | | INotificationHandler<WorkflowDefinitionDeleted>, |
| | | 23 | | INotificationHandler<WorkflowDefinitionsDeleted>, |
| | | 24 | | INotificationHandler<WorkflowDefinitionVersionDeleted>, |
| | | 25 | | INotificationHandler<WorkflowDefinitionVersionsDeleted>, |
| | | 26 | | INotificationHandler<WorkflowDefinitionVersionsUpdated> |
| | | 27 | | { |
| | | 28 | | /// <inheritdoc /> |
| | | 29 | | public async Task HandleAsync(WorkflowDefinitionPublished notification, CancellationToken cancellationToken) |
| | | 30 | | { |
| | 8 | 31 | | await UpdateDefinition(notification.WorkflowDefinition.Id, notification.WorkflowDefinition.Options.UsableAsActiv |
| | 8 | 32 | | await generationStore.IncrementAsync(notification.WorkflowDefinition.TenantId, cancellationToken); |
| | 8 | 33 | | } |
| | | 34 | | |
| | | 35 | | /// <inheritdoc /> |
| | | 36 | | public async Task HandleAsync(WorkflowDefinitionRetracted notification, CancellationToken cancellationToken) |
| | | 37 | | { |
| | 1 | 38 | | await UpdateDefinition(notification.WorkflowDefinition.Id, notification.WorkflowDefinition.Options.UsableAsActiv |
| | 1 | 39 | | await generationStore.IncrementAsync(notification.WorkflowDefinition.TenantId, cancellationToken); |
| | 1 | 40 | | } |
| | | 41 | | |
| | | 42 | | /// <inheritdoc /> |
| | | 43 | | public async Task HandleAsync(WorkflowDefinitionVersionRetracted notification, CancellationToken cancellationToken) |
| | | 44 | | { |
| | 4 | 45 | | await UpdateDefinition(notification.WorkflowDefinition.Id, notification.WorkflowDefinition.Options.UsableAsActiv |
| | 4 | 46 | | await generationStore.IncrementAsync(notification.WorkflowDefinition.TenantId, cancellationToken); |
| | 4 | 47 | | } |
| | | 48 | | |
| | | 49 | | /// <inheritdoc /> |
| | | 50 | | public async Task HandleAsync(WorkflowDefinitionDeleted notification, CancellationToken cancellationToken) |
| | | 51 | | { |
| | 6 | 52 | | workflowDefinitionActivityRegistryUpdater.RemoveDefinitionFromRegistry(notification.DefinitionId); |
| | 6 | 53 | | await IncrementTenantAndAgnosticGenerationsAsync(tenantAccessor.TenantId, cancellationToken); |
| | 6 | 54 | | } |
| | | 55 | | |
| | | 56 | | /// <inheritdoc /> |
| | | 57 | | public async Task HandleAsync(WorkflowDefinitionsDeleted notification, CancellationToken cancellationToken) |
| | | 58 | | { |
| | 18 | 59 | | foreach (string id in notification.DefinitionIds) |
| | | 60 | | { |
| | 6 | 61 | | workflowDefinitionActivityRegistryUpdater.RemoveDefinitionFromRegistry(id); |
| | | 62 | | } |
| | | 63 | | |
| | 3 | 64 | | await IncrementTenantAndAgnosticGenerationsAsync(tenantAccessor.TenantId, cancellationToken); |
| | 3 | 65 | | } |
| | | 66 | | |
| | | 67 | | /// <inheritdoc /> |
| | | 68 | | public async Task HandleAsync(WorkflowDefinitionVersionDeleted notification, CancellationToken cancellationToken) |
| | | 69 | | { |
| | 1 | 70 | | workflowDefinitionActivityRegistryUpdater.RemoveDefinitionVersionFromRegistry(notification.WorkflowDefinition.Id |
| | 1 | 71 | | await generationStore.IncrementAsync(notification.WorkflowDefinition.TenantId, cancellationToken); |
| | 1 | 72 | | } |
| | | 73 | | |
| | | 74 | | /// <inheritdoc /> |
| | | 75 | | public async Task HandleAsync(WorkflowDefinitionVersionsDeleted notification, CancellationToken cancellationToken) |
| | | 76 | | { |
| | 6 | 77 | | foreach (string id in notification.Ids) |
| | | 78 | | { |
| | 2 | 79 | | workflowDefinitionActivityRegistryUpdater.RemoveDefinitionVersionFromRegistry(id); |
| | | 80 | | } |
| | | 81 | | |
| | 1 | 82 | | await IncrementTenantAndAgnosticGenerationsAsync(tenantAccessor.TenantId, cancellationToken); |
| | 1 | 83 | | } |
| | | 84 | | |
| | | 85 | | /// <inheritdoc /> |
| | | 86 | | public async Task HandleAsync(WorkflowDefinitionVersionsUpdated notification, CancellationToken cancellationToken) |
| | | 87 | | { |
| | 1 | 88 | | var definitions = notification.WorkflowDefinitions.ToList(); |
| | 8 | 89 | | foreach (var definition in definitions) |
| | | 90 | | { |
| | 3 | 91 | | await UpdateDefinition(definition.Id, definition.Options.UsableAsActivity); |
| | | 92 | | } |
| | | 93 | | |
| | 11 | 94 | | foreach (var tenantId in definitions.Select(x => x.TenantId).Distinct()) |
| | 3 | 95 | | await generationStore.IncrementAsync(tenantId, cancellationToken); |
| | 1 | 96 | | } |
| | | 97 | | |
| | | 98 | | private Task UpdateDefinition(string id, bool? usableAsActivity) |
| | | 99 | | { |
| | | 100 | | // Once a workflow has been published, it should remain in the activity registry unless no longer being marked a |
| | 16 | 101 | | if (usableAsActivity.GetValueOrDefault()) |
| | 8 | 102 | | return workflowDefinitionActivityRegistryUpdater.AddToRegistry(id); |
| | | 103 | | |
| | 8 | 104 | | workflowDefinitionActivityRegistryUpdater.RemoveDefinitionVersionFromRegistry(id); |
| | 8 | 105 | | return Task.CompletedTask; |
| | | 106 | | } |
| | | 107 | | |
| | | 108 | | private async Task IncrementTenantAndAgnosticGenerationsAsync(string? tenantId, CancellationToken cancellationToken) |
| | | 109 | | { |
| | 10 | 110 | | var normalizedTenantId = tenantId.NormalizeTenantId(); |
| | 10 | 111 | | if (normalizedTenantId != Tenant.AgnosticTenantId) |
| | 9 | 112 | | await generationStore.IncrementAsync(Tenant.AgnosticTenantId, cancellationToken); |
| | | 113 | | |
| | 10 | 114 | | await generationStore.IncrementAsync(normalizedTenantId, cancellationToken); |
| | 10 | 115 | | } |
| | | 116 | | } |