WorkflowNodeTraversal.java

package com.smartcodeltd.jenkinsci.plugins.buildmonitor.pipeline;

import org.jenkinsci.plugins.workflow.cps.nodes.StepStartNode;
import org.jenkinsci.plugins.workflow.graph.FlowNode;
import org.jenkinsci.plugins.workflow.support.steps.StageStep;

import java.util.Collection;

public class WorkflowNodeTraversal extends BreadthFirstNodeTraversal<FlowNode> {
    @Override
    protected boolean isStageStep(FlowNode node) {
        return node instanceof StepStartNode
                && ((StepStartNode) node).getDescriptor().isSubTypeOf(StageStep.class);
    }

    @Override
    protected String getDisplayName(FlowNode node) {
        return node.getDisplayName();
    }

    @Override
    protected Collection<FlowNode> getParents(FlowNode node) {
        return node.getParents();
    }
}