嵌套集合lambda迭代

时间:2017-05-15 14:09:38

标签: java lambda java-8

假设我有一个包含集合的对象,所述集合中的每个元素都包含一个集合,每个集合都包含一个集合。

我想迭代最深的对象并将相同的代码应用于它。

这种必要的方式是微不足道的,但有没有办法让这一切变得简单?

以下是代码今天的样子:

My object o;
SecretType computedThingy = 78;
for (FirstLevelOfCollection coll : o.getList()) {
  for (SecondLevelOfCollection colColl : coll.getSet()) {
    for (MyCoolTinyObjects mcto : colColl.getFoo()) {
      mcto.setSecretValue(computedThingy);
    }
  }
}

我可以看到如何从最深的循环中创建一个lambda:

colColl.getFoo().stream().forEach(x -> x.setSecretValue(computedThingy)

但我可以做更多吗?

2 个答案:

答案 0 :(得分:5)

flatMap可用于此目的。你在这里得到的是迭代各种最深集合的所有元素,就像它们是一个集合一样:

o.getList().stream()
    .flatMap(c1 -> c1.getSet().stream())
    .flatMap(c2 -> c2.getFoo().stream())
    .forEach(x -> x.setSecretValue(computedThingy));

答案 1 :(得分:3)

flatMap救援,带有嵌套String

集合的简单示例

另见: Java 8 Streams FlatMap method example

Turn a List of Lists into a List Using Lambdas

    Set<List<List<String>>> outerMostSet = new HashSet<>();
    List<List<String>> middleList = new ArrayList<>();
    List<String> innerMostList = new ArrayList<>();
    innerMostList.add("foo");
    innerMostList.add("bar");
    middleList.add(innerMostList);

    List<String> anotherInnerMostList = new ArrayList<>();
    anotherInnerMostList.add("another foo");

    middleList.add(anotherInnerMostList);
    outerMostSet.add(middleList);

    outerMostSet.stream()
                .flatMap(mid -> mid.stream())
                .flatMap(inner -> inner.stream())
                .forEach(System.out::println);

可生产

foo 
bar 
another foo