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

Question

我有一个Person对象的集合：

public class Person {

  String name;

  ChildrenListHolder childrenListHolder;
}

public class ChildrenListHolder {
   List<Children> children;
}

public class Children {
   String childrensName;
}

（实体结构由第三方提供。）

现在，我需要一个Map<String,List<Person>> childrensName - > person-list

例如（简化）：

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和if来做到这一点。 但是我如何使用流和收集器来做到这一点。 我尝试了很多方法，但是我无法得到预期的结果。 TIA。

Answer 1

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

Answer 2

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

我知道你要求一个流/收集器解决方案，但无论如何使用Map#computeIfAbsent的嵌套for循环也可以Map#computeIfAbsent工作：

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）那样易于并行化，但是你也不需要“束”来实现它（并且你也避免创建临时的映射条目实例）。