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 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<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) 那样是单行的，也不是容易并行化的，但是您也不需要“一堆” if 来实现这一点（并且您还避免创建临时地图条目实例）。