Java 8使用流和收集器映射到集合的子列表条目

时间:2016-03-28 20:54:50

标签: java java-8 java-stream collectors

我有一组Person个对象:。

public class Person {

  String name;

  ChildrenListHolder childrenListHolder;
}

public class ChildrenListHolder {
   List<Children> children;
}

public class Children {
   String childrensName;
}

(实体结构由第三方提供。)

现在,我需要一个Map<String,List<Person>> childrensName - &gt;人列表

例如(简化):

Person father: {name: "John", childrensListHolder -> {"Lisa", "Jimmy"}}
Person mother: {name: "Clara", childrensListHolder -> {"Lisa", "Paul"}}
Person george: {name: "George", childrensListHold -> "Paul"}}

地图,我需要的是

Map<String, List<Person>> map: {"Lisa"  -> {father, mother},
                                "Jimmy" -> {father},
                                "Paul"  -> {mother, george}}

我可以用一堆for for和#ifs来做到这一点。但是我如何使用流和收集器来做到这一点。我尝试了很多方法,但是我无法得到预期的结果。 TIA。

2 个答案:

答案 0 :(得分:8)

给定List<Person> persons,您可以拥有以下内容

Map<String,List<Person>> map =
    persons.stream()
           .flatMap(p -> p.childrenListHolder.children.stream().map(c -> new AbstractMap.SimpleEntry<>(c, p)))
           .collect(Collectors.groupingBy(
             e -> e.getKey().childrensName,
             Collectors.mapping(Map.Entry::getValue, Collectors.toList())
           ));

这正在为人们创造一条流。然后每个人通过一个元组来平面映射,该元组持有孩子和每个孩子的人。最后,我们按子名称分组,并将所有人员收集到一个列表中。

假设有适当的构造函数的示例代码:

public static void main(String[] args) {
    List<Person> persons = Arrays.asList(
        new Person("John", new ChildrenListHolder(Arrays.asList(new Children("Lisa"), new Children("Jimmy")))),
        new Person("Clara", new ChildrenListHolder(Arrays.asList(new Children("Lisa"), new Children("Paul")))),
        new Person("George", new ChildrenListHolder(Arrays.asList(new Children("Paul"))))
    );

    Map<String,List<Person>> map =
        persons.stream()
               .flatMap(p -> p.childrenListHolder.children.stream().map(c -> new AbstractMap.SimpleEntry<>(c, p)))
               .collect(Collectors.groupingBy(
                 e -> e.getKey().childrensName,
                 Collectors.mapping(Map.Entry::getValue, Collectors.toList())
               ));

    System.out.println(map);
}

答案 1 :(得分:5)

  

我可以用一堆for和if来做到这一点。

我知道您要求使用流/收集器解决方案,但无论如何使用Map#computeIfAbsent嵌套for循环也可以正常工作:

Map<String, List<Person>> map = new HashMap<>();
for(Person p : persons) {
    for(Children c : p.childrenListHolder.children) {
        map.computeIfAbsent(c.childrensName, k -> new ArrayList<>()).add(p);
    }
}

这是使用集合中引入的新forEach方法编写的:

Map<String, List<Person>> map = new HashMap<>();
persons.forEach(p -> p.childrenListHolder.children.forEach(c -> map.computeIfAbsent(c.childrensName, k -> new ArrayList<>()).add(p)));

当然,它不是单线程,也不像Tunaki的解决方案(+1)那样易于并行化,但是你也不需要“束”来实现它(并且你也避免创建临时映射条目)实例)。