fix: Fix for missing steps in the UI (#12203)

Signed-off-by: Alan Clucas <alan@clucas.org>
argoproj · Nov 15, 2023 · 206c901 · 206c901
1 parent 18c5925
commit 206c901
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/workflow/controller/dag.go b/workflow/controller/dag.go
@@ -351,6 +351,11 @@ func (woc *wfOperationCtx) executeDAG(ctx context.Context, nodeName string, tmpl
 		return node, err
 	}
 	if outputs != nil {
+		node, err = woc.wf.GetNodeByName(nodeName)
+		if err != nil {
+			woc.log.Errorf("unable to get node by name for %s", nodeName)
+			return nil, err
+		}
 		node.Outputs = outputs
 		woc.wf.Status.Nodes.Set(node.ID, *node)
 	}

diff --git a/workflow/controller/steps.go b/workflow/controller/steps.go
@@ -172,6 +172,10 @@ func (woc *wfOperationCtx) executeSteps(ctx context.Context, nodeName string, tm
 	}
 
 	if outputs != nil {
+		node, err := woc.wf.GetNodeByName(nodeName)
+		if err != nil {
+			return nil, err
+		}
 		node.Outputs = outputs
 		woc.addOutputsToGlobalScope(node.Outputs)
 		woc.wf.Status.Nodes.Set(node.ID, *node)