使用Java Stream修改嵌套StepsDto的指定子步骤状态并返回父列表

我定义了如下DTO：

public class StepsDto {
    
    private String step_name;
    private String status;
    private List<StepsDto> sub_steps;
    private String previousStep;
    private String nextStep;
}

需要用Java Stream实现：当某个步骤及其子步骤、孙步骤的名称匹配指定条件时，修改这些匹配步骤的状态，最终返回完整的父StepsDto列表。之前尝试用flatMap，但它返回的是子列表而非原主列表。比如存在父步骤P1、子步骤C1、孙步骤CC1，希望修改CC1的状态后返回包含P1的父列表。

因为要保留原列表的层级结构，不能用flatMap扁平化数据，而是要通过递归遍历每个层级的步骤，找到匹配项后修改状态，原列表引用不变，直接返回即可。

实现思路 #

• 编写递归方法，遍历单个StepsDto及其所有子步骤
• 对每个步骤判断名称是否符合条件，匹配则更新状态
• 对原父列表的每个元素调用该递归方法，最后返回原列表

代码示例 #

import java.util.List;
import java.util.function.Predicate;

public class StepProcessor {

    // 处理父步骤列表，修改符合条件的步骤状态
    public List<StepsDto> updateMatchingSteps(List<StepsDto> parentSteps,
                                              Predicate<String> nameMatcher,
                                              String newStatus) {
        parentSteps.forEach(step -> traverseAndUpdate(step, nameMatcher, newStatus));
        return parentSteps;
    }

    // 递归遍历并更新单个步骤及其子步骤
    private void traverseAndUpdate(StepsDto step, Predicate<String> nameMatcher, String newStatus) {
        // 检查当前步骤是否匹配，匹配则修改状态
        if (nameMatcher.test(step.getStep_name())) {
            step.setStatus(newStatus);
        }
        // 递归处理子步骤（避免空指针）
        if (step.getSub_steps() != null && !step.getSub_steps().isEmpty()) {
            step.getSub_steps().forEach(subStep -> traverseAndUpdate(subStep, nameMatcher, newStatus));
        }
    }
}

使用示例 #

比如要修改名称为CC1的步骤状态为COMPLETED：

// 假设已初始化好包含P1的父步骤列表
List<StepsDto> parentSteps = getParentSteps();

StepProcessor processor = new StepProcessor();
List<StepsDto> updatedList = processor.updateMatchingSteps(
    parentSteps,
    stepName -> "CC1".equals(stepName),
    "COMPLETED"
);
// updatedList就是修改后的原父列表，层级结构保持不变，CC1的状态已更新

关键说明 #

• 该方法基于对象引用修改，不会改变原列表的结构，直接返回原父列表即可
• 递归逻辑会覆盖所有层级的子步骤，确保所有匹配的节点都被处理
• 使用Predicate传递匹配条件，可灵活适配不同的名称匹配规则（比如模糊匹配、多名称匹配等）

内容的提问来源于stack exchange，提问作者Gaurav Matta