| | | 1 | | using Elsa.Authorization; |
| | | 2 | | using Elsa.Abstractions; |
| | | 3 | | using Elsa.Alterations.AlterationTypes; |
| | | 4 | | using Elsa.Alterations.Core.Contracts; |
| | | 5 | | using Elsa.Alterations.Core.Results; |
| | | 6 | | using Elsa.Workflows.Management; |
| | | 7 | | using Elsa.Workflows.Management.Entities; |
| | | 8 | | using Elsa.Workflows.Management.Filters; |
| | | 9 | | using Elsa.Workflows.Runtime; |
| | | 10 | | using Elsa.Workflows.Runtime.Contracts; |
| | | 11 | | using Elsa.Workflows.Runtime.Requests; |
| | | 12 | | using JetBrains.Annotations; |
| | | 13 | | |
| | | 14 | | namespace Elsa.Alterations.Endpoints.Workflows.Retry; |
| | | 15 | | |
| | | 16 | | /// <summary> |
| | | 17 | | /// Retries the specified workflow instances. |
| | | 18 | | /// </summary> |
| | | 19 | | [PublicAPI] |
| | | 20 | | public class Retry : ElsaEndpoint<Request, Response> |
| | | 21 | | { |
| | | 22 | | private readonly IAlterationRunner _alterationRunner; |
| | | 23 | | private readonly IWorkflowDispatcher _workflowDispatcher; |
| | | 24 | | private readonly IWorkflowInstanceStore _workflowInstanceStore; |
| | | 25 | | |
| | | 26 | | /// <inheritdoc /> |
| | 3 | 27 | | public Retry(IAlterationRunner alterationRunner, IWorkflowDispatcher workflowDispatcher, IWorkflowInstanceStore work |
| | | 28 | | { |
| | 3 | 29 | | _alterationRunner = alterationRunner; |
| | 3 | 30 | | _workflowDispatcher = workflowDispatcher; |
| | 3 | 31 | | _workflowInstanceStore = workflowInstanceStore; |
| | 3 | 32 | | } |
| | | 33 | | |
| | | 34 | | /// <inheritdoc /> |
| | | 35 | | public override void Configure() |
| | | 36 | | { |
| | 3 | 37 | | Routes("/alterations/workflows/retry"); |
| | 3 | 38 | | Verbs(FastEndpoints.Http.GET, FastEndpoints.Http.POST); |
| | 3 | 39 | | RequirePermission(Elsa.Alterations.Permissions.AlterationPermissions.Alterations, CoreVerbs.Execute); |
| | 3 | 40 | | } |
| | | 41 | | |
| | | 42 | | /// <inheritdoc /> |
| | | 43 | | public override async Task HandleAsync(Request request, CancellationToken cancellationToken) |
| | | 44 | | { |
| | 0 | 45 | | var allResults = new List<RunAlterationsResult>(); |
| | | 46 | | |
| | | 47 | | // Load each workflow instance. |
| | 0 | 48 | | var workflowInstances = (await _workflowInstanceStore.FindManyAsync(new WorkflowInstanceFilter { Ids = request.W |
| | | 49 | | |
| | 0 | 50 | | foreach (var workflowInstance in workflowInstances) |
| | | 51 | | { |
| | | 52 | | // Setup an alteration plan. |
| | 0 | 53 | | var activityIds = GetActivityIds(request, workflowInstance).ToList(); |
| | 0 | 54 | | var alterations = activityIds.Select(activityId => new ScheduleActivity { ActivityId = activityId }).Cast<IA |
| | | 55 | | |
| | | 56 | | // Run the plan. |
| | 0 | 57 | | var results = await _alterationRunner.RunAsync(request.WorkflowInstanceIds, alterations, cancellationToken); |
| | 0 | 58 | | allResults.AddRange(results); |
| | | 59 | | |
| | | 60 | | // Schedule updated workflow. |
| | 0 | 61 | | await _workflowDispatcher.DispatchAsync(new DispatchWorkflowInstanceRequest(workflowInstance.Id), cancellati |
| | 0 | 62 | | } |
| | | 63 | | |
| | | 64 | | // Write response. |
| | 0 | 65 | | var response = new Response(allResults); |
| | 0 | 66 | | await Send.OkAsync(response, cancellationToken); |
| | 0 | 67 | | } |
| | | 68 | | |
| | | 69 | | private IEnumerable<string> GetActivityIds(Request request, WorkflowInstance workflowInstance) |
| | | 70 | | { |
| | | 71 | | // If activity IDs are explicitly specified, use them. |
| | 0 | 72 | | if (request.ActivityIds?.Any() == true) |
| | 0 | 73 | | return request.ActivityIds; |
| | | 74 | | |
| | | 75 | | // Otherwise, select IDs of all faulted activities. |
| | 0 | 76 | | return workflowInstance.WorkflowState.Incidents.Select(x => x.ActivityId).Distinct().ToList(); |
| | | 77 | | } |
| | | 78 | | } |