feat: display failure node in workflow graph

Signed-off-by: Lyon Lu <lyonlu13@gmail.com>
flyteorg · Feb 23, 2024 · f89be26 · f89be26
1 parent a162fb0
commit f89be26
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/packages/console/src/components/WorkflowGraph/transformerWorkflowToDag.tsx b/packages/console/src/components/WorkflowGraph/transformerWorkflowToDag.tsx
@@ -492,6 +492,28 @@ const parseWorkflow = (
     };
   }
 
+  /* Build failure node and add downstream connection for edges building */
+  const failureNode = context.template.failureNode;
+  if (failureNode && failureNode.id) {
+    parseNode({
+      node: failureNode as CompiledNode,
+      root: root,
+      dynamicToMerge,
+      nodeExecutionsById,
+      staticExecutionIdsMap,
+      workflow,
+    });
+    nodeMap[failureNode.id] = {
+      dNode: root.nodes[root.nodes.length - 1],
+      compiledNode: failureNode as CompiledNode,
+    };
+    if (
+      !context.connections.downstream[startNodeId].ids.includes(failureNode.id)
+    ) {
+      context.connections.downstream[startNodeId].ids.push(failureNode.id);
+    }
+  }
+
   /* Build Edges */
   buildWorkflowEdges(root, context.connections, startNodeId, nodeMap);
   return root;