平面图嵌套集合

Question

我有一个对象列表，其中一些可以是集合。我想得到一个普通对象流。

List<Object> objects = List.of(1, 2, "SomeString", List.of(3, 4, 5, 6), 
    7, List.of("a", "b", "c"),
    List.of(8, List.of(9, List.of(10))));

我想要一个包含元素的流。

1, 2, "SomeString", 3, 4, 5, 6, 7, "a", "b", "c", 8, 9, 10

我试过了

Function<Object, Stream<Object>> mbjectToStreamMapper = null; //define it. I have not figured it out yet!
objects.stream().flatMap(ObjectToStreamMapper).forEach(System.out::println);

我还检查了一个 example，它展示了如何使用递归函数来展平集合。但是，在此示例中，.collect(Collectors.toList()); 用于保留中间结果。 Collectors.toList() 是一个终端操作，它会立即开始处理流。我想要一个流，稍后我可以对其进行迭代。

更新

我同意评论，拥有一个由不同性质的对象组成的流是一个糟糕的想法。为了简单起见，我只是写了这个问题。在现实生活中，可能是我监听不同的事件，并处理来自传入流的一些业务对象，其中一些可以发送对象流，其他 - 只是单个对象。

Answer 1

class Loop {
    private static Stream<Object> flat(Object o) {
        return o instanceof Collection ?
                ((Collection) o).stream().flatMap(Loop::flat) : Stream.of(o);
    }

    public static void main(String[] args) {
        List<Object> objects = List.of(1, 2, "SomeString", List.of( 3, 4, 5, 6),
                7, List.of("a", "b", "c"), List.of(8, List.of(9, List.of(10))));

        List<Object> flat = flat(objects).collect(Collectors.toList());

        System.out.println(flat);
    }
}

请注意 List.of(null) 抛出 NPE。

Answer 2

如果遍历的对象是Collection的实例，我们可以递归得到嵌套的stream。

public static void main(String args[]) {
       List<Object> objects = List.of(1, 2, "SomeString", List.of(3, 4, 5, 6),
            7, List.of("a", "b", "c"),
            List.of(8, List.of(9, List.of(10))));
       List<Object> list = objects.stream().flatMap(c -> getNestedStream(c)).collect(Collectors.toList());
}

public static Stream<Object> getNestedStream(Object obj) {
    if(obj instanceof Collection){
        return ((Collection)obj).stream().flatMap((coll) -> getNestedStream(coll));
    }
    return Stream.of(obj);
}

Answer 3

注意，可以在字段中定义递归方法：

public class Test {
  static Function<Object,Stream<?>> flat=
    s->s instanceof Collection ? ((Collection<?>)s).stream().flatMap(Test.flat) : Stream.of(s);
  public static void main(String[] args) {
    objects.stream().flatMap(flat).forEach(System.out::print);
  }
}

平面图嵌套集合

Flatmap nested collection

java

java-stream

flatmap

更新