平面图嵌套集合

java java-stream flatmap

2022-09-02 21:22:33

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

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

更新

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

答案 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);
    }
}

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

答案 2

如果正在遍历的对象是的实例，我们可以递归地获取嵌套。streamCollection

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