Fix null ID reference bug in function definition provider

This commit is contained in:
Sipke Schoorstra 2023-06-10 15:30:47 +02:00
parent 6e03f61818
commit 8d33084d4e

View file

@ -14,11 +14,13 @@ internal class ActivityOutputFunctionsDefinitionProvider : FunctionDefinitionPro
{
private readonly IActivityVisitor _activityVisitor;
private readonly IActivityRegistry _activityRegistry;
private readonly IIdentityGraphService _identityGraphService;
public ActivityOutputFunctionsDefinitionProvider(IActivityVisitor activityVisitor, IActivityRegistry activityRegistry)
public ActivityOutputFunctionsDefinitionProvider(IActivityVisitor activityVisitor, IActivityRegistry activityRegistry, IIdentityGraphService identityGraphService)
{
_activityVisitor = activityVisitor;
_activityRegistry = activityRegistry;
_identityGraphService = identityGraphService;
}
protected override async ValueTask<IEnumerable<FunctionDefinition>> GetFunctionDefinitionsAsync(TypeDefinitionContext context)
@ -27,6 +29,8 @@ internal class ActivityOutputFunctionsDefinitionProvider : FunctionDefinitionPro
var nodes = (await _activityVisitor.VisitAsync(context.Workflow.Root, context.CancellationToken)).Flatten().Distinct().ToList();
var activitiesWithOutputs = nodes.GetActivitiesWithOutputs(_activityRegistry);
var definitions = new List<FunctionDefinition>();
_identityGraphService.AssignIdentities(nodes);
foreach (var (activity, activityDescriptor) in activitiesWithOutputs)
{