| | | 1 | | using Elsa.Extensions; |
| | | 2 | | using Elsa.Workflows.Activities; |
| | | 3 | | using Elsa.Workflows.Models; |
| | | 4 | | using Humanizer; |
| | | 5 | | using Microsoft.Extensions.Logging; |
| | | 6 | | |
| | | 7 | | namespace Elsa.Workflows; |
| | | 8 | | |
| | | 9 | | /// <inheritdoc /> |
| | 1021 | 10 | | public class IdentityGraphService(IActivityVisitor activityVisitor, IActivityRegistryLookupService activityRegistryLooku |
| | | 11 | | { |
| | | 12 | | /// <inheritdoc /> |
| | | 13 | | public async Task AssignIdentitiesAsync(Workflow workflow, CancellationToken cancellationToken = default) |
| | | 14 | | { |
| | 2756 | 15 | | await AssignIdentitiesAsync((IActivity)workflow, cancellationToken); |
| | 2756 | 16 | | } |
| | | 17 | | |
| | | 18 | | /// <inheritdoc /> |
| | | 19 | | public async Task AssignIdentitiesAsync(IActivity root, CancellationToken cancellationToken = default) |
| | | 20 | | { |
| | 2756 | 21 | | var graph = await activityVisitor.VisitAsync(root, cancellationToken); |
| | 2756 | 22 | | await AssignIdentitiesAsync(graph); |
| | 2756 | 23 | | } |
| | | 24 | | |
| | | 25 | | /// <inheritdoc /> |
| | 2756 | 26 | | public Task AssignIdentitiesAsync(ActivityNode root) => AssignIdentitiesAsync(root.Flatten().ToList()); |
| | | 27 | | |
| | | 28 | | /// <inheritdoc /> |
| | | 29 | | public async Task AssignIdentitiesAsync(ICollection<ActivityNode> flattenedList) |
| | | 30 | | { |
| | 9705 | 31 | | var identityCounters = new Dictionary<string, int>(); |
| | | 32 | | |
| | 102028 | 33 | | foreach (var node in flattenedList) |
| | | 34 | | { |
| | 41309 | 35 | | node.Activity.Id = CreateId(node, identityCounters, flattenedList); |
| | 41309 | 36 | | node.Activity.NodeId = node.NodeId; |
| | 41309 | 37 | | await AssignInputOutputsAsync(node.Activity); |
| | | 38 | | |
| | 41309 | 39 | | if (node.Activity is IVariableContainer variableContainer) |
| | 21006 | 40 | | AssignVariables(variableContainer); |
| | 41309 | 41 | | } |
| | 9705 | 42 | | } |
| | | 43 | | |
| | | 44 | | /// <inheritdoc /> |
| | | 45 | | public async Task AssignInputOutputsAsync(IActivity activity) |
| | | 46 | | { |
| | 41309 | 47 | | var activityDescriptor = await activityRegistryLookup.FindAsync(activity.Type, activity.Version); |
| | | 48 | | |
| | 41309 | 49 | | if (activityDescriptor == null!) |
| | | 50 | | { |
| | 644 | 51 | | logger.LogWarning("Activity descriptor not found for activity type {ActivityType}. Skipping identity assignm |
| | 644 | 52 | | return; |
| | | 53 | | } |
| | | 54 | | |
| | 40665 | 55 | | var inputDictionary = activityDescriptor.GetWrappedInputProperties(activity); |
| | | 56 | | |
| | 181834 | 57 | | foreach (var (inputName, input) in inputDictionary) |
| | | 58 | | { |
| | 50252 | 59 | | var blockReference = input?.MemoryBlockReference(); |
| | | 60 | | |
| | 50252 | 61 | | if (blockReference == null!) |
| | | 62 | | continue; |
| | | 63 | | |
| | 32879 | 64 | | if (string.IsNullOrEmpty(blockReference.Id)) |
| | 17724 | 65 | | blockReference.Id = $"{activity.Id}:input-{inputName.Humanize().Kebaberize()}"; |
| | | 66 | | } |
| | | 67 | | |
| | 40665 | 68 | | var outputs = activity.GetOutputs(); |
| | | 69 | | |
| | 99474 | 70 | | foreach (var output in outputs) |
| | | 71 | | { |
| | 9072 | 72 | | var blockReference = output.Value.MemoryBlockReference(); |
| | | 73 | | |
| | 9072 | 74 | | if (blockReference == null!) |
| | | 75 | | continue; |
| | | 76 | | |
| | 9072 | 77 | | if (string.IsNullOrEmpty(blockReference.Id)) |
| | 3842 | 78 | | blockReference.Id = $"{activity.Id}:output-{output.Name.Humanize().Kebaberize()}"; |
| | | 79 | | } |
| | 41309 | 80 | | } |
| | | 81 | | |
| | | 82 | | /// <inheritdoc /> |
| | | 83 | | public void AssignVariables(IVariableContainer activity) |
| | | 84 | | { |
| | 21006 | 85 | | var variables = activity.Variables; |
| | 21006 | 86 | | var seed = 0; |
| | | 87 | | |
| | 47238 | 88 | | foreach (var variable in variables) |
| | 2613 | 89 | | variable.Id = variable.Id != null! ? variable.Id : $"{activity.Id}:variable-{++seed}"; |
| | 21006 | 90 | | } |
| | | 91 | | |
| | | 92 | | private string CreateId(ActivityNode activityNode, IDictionary<string, int> identityCounters, ICollection<ActivityNo |
| | | 93 | | { |
| | 41309 | 94 | | if (!string.IsNullOrWhiteSpace(activityNode.Activity.Id)) |
| | 22866 | 95 | | return activityNode.Activity.Id; |
| | | 96 | | |
| | | 97 | | while (true) |
| | | 98 | | { |
| | 19378 | 99 | | var fullTypeName = activityNode.Activity.Type; |
| | 19378 | 100 | | var shortTypeName = fullTypeName.Split('.').Last(); |
| | 19378 | 101 | | var index = GetNextIndexFor(shortTypeName, identityCounters); |
| | 19378 | 102 | | var name = $"{shortTypeName}{index + 1}"; |
| | | 103 | | |
| | 112513 | 104 | | if (allNodes.All(x => x.Activity.Id != name)) |
| | 18443 | 105 | | return name; |
| | | 106 | | } |
| | | 107 | | } |
| | | 108 | | |
| | | 109 | | private int GetNextIndexFor(string activityType, IDictionary<string, int> identityCounters) |
| | | 110 | | { |
| | 19378 | 111 | | if (!identityCounters.TryGetValue(activityType, out var index)) |
| | | 112 | | { |
| | 17075 | 113 | | identityCounters[activityType] = index; |
| | | 114 | | } |
| | | 115 | | else |
| | | 116 | | { |
| | 2303 | 117 | | index = identityCounters[activityType] + 1; |
| | 2303 | 118 | | identityCounters[activityType] = index; |
| | | 119 | | } |
| | | 120 | | |
| | 19378 | 121 | | return index; |
| | | 122 | | } |
| | | 123 | | } |