平面图嵌套集合

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);我还检查了一个示例，该示例显示了如何使用递归函数来展平集合。但是，在此示例中.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

如果被遍历stream的对象是Collection.

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);

  }

}