Java流flatMap，其中保留第一级和第二级对象

Question

我想知道如何使用Java Stream API来扁平化其中包含对象和嵌套相同类型对象的集合的结构。

我的意思是，我有一个Component类，该类具有一个List<Component>类型的字段。 我想做的是找到一个整洁的流解决方案，该解决方案与以下代码相同（我需要以所有组件和嵌套子组件的列表结尾）。

List<Component> components = getComponents(id);
List<Component> componentsAndSubcomponents = new ArrayList<>();
for (Component component : components) {
  componentsAndSubcomponents.add(component);
  componentsAndSubcomponents.addAll(component.getSubComponents());
}

Answer 1

您可以将flatMap与Stream串联一起使用：

List<Component> componentsAndSubcomponents =
    components.stream()
              .flatMap(c -> Stream.concat(Stream.of(c),c.getSubComponents().stream()))
              .collect(Collectors.toList());

这将每个映射Component成Stream ，其中包含Component ，随后所有的子组件，并压平所有这些Stream s转换为平坦Stream<Component> ，将被收集到一个List<Component> 。

Answer 2

一个简单的解决方案是动态创建内部流，如下所示：

List<Component> result = components.stream()
    .flatMap(comp -> 
        Stream.concat(Stream.of(comp), comp.getSubComponents().stream()))
    .collect(Collectors.toList());