如何在java 8流中使用一对多映射？

Question

在Stream map函数中，我们可以将一个对象转换为另一个对象，因此我们可以将一个包含3个A类元素的Stream转换为另一个包含B类型的3个元素的Stream 。

如何转换类型的3个元素A Stream类型为6个以上的元素B Stream取决于条件？

在代码方面。

我们可以做的

Stream<B> converted = original.map( a -> new B(a) );

但我们怎么能这样做呢？

Steam<B> converted = original.map( a -> { 
    if(a.split()){
        return [ new B(a), new B(a) ];
    }else return new B(a);
});

我无法找到并理解如何做到这一点。 谢谢你。

Answer 1

您可以使用flatMap将原始Stream每个元素映射到某种类型的元素Stream 。

Steam<B> converted = original.flatMap( a -> { 
    if(a.split()){
        return Stream.of(new B(a), new B(a));
    } else {
        return Stream.of(new B(a));
    }
});

要么

Steam<B> converted = original.flatMap(a -> a.split() ? 
                                      Stream.of(new B(a), new B(a)) : 
                                      Stream.of(new B(a)));