平面图嵌套集合

时间:2019-02-03 15:45:52

标签: java java-stream flatmap

我有一个对象列表,其中一些可以是集合。我想要一堆普通的对象。

lapply(list1,  function(x)  as.data.frame(rbind(as.matrix(x)[-5, ], 
              c('replaced',  as.character(x$names1[5])))))

我想获得包含元素的流。

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

我还检查了example,它显示了如何使用拉平集合的递归函数。但是,在此示例中,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());是一种终端操作,它将立即开始处理流。我想要一个流,以后可以对其进行迭代。


更新

我同意评论,拥有由不同性质的对象组成的信息流是一个糟糕的主意。我只是为了简化而写了这个问题。在现实生活中,可能是我听了不同的事件,并从传入流中处理了一些业务对象,其中一些可以发送对象流,其他的-仅是单个对象。

3 个答案:

答案 0 :(得分:16)

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。

答案 1 :(得分:5)

如果要遍历的对象是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);
}

答案 2 :(得分:2)

注意,可以在字段中定义递归方法:

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