Java 8并行流并发分组

时间:2017-12-16 18:20:21

标签: java java-8 java-stream

假设我有一个班级

Class Person {
  String name;
  String uid;
  String phone;
}

我正在尝试按类的所有字段进行分组。如何在JAVA 8中使用并行流来转换

List<Person> into Map<String,Set<Person>>

其中地图的关键字是类中每个字段的值。 JAVA 8以下示例组由单个字段组成,如何才能将一个类的所有字段组成一个Map?

ConcurrentMap<Person.Sex, List<Person>> byGender =
roster
    .parallelStream()
    .collect(
        Collectors.groupingByConcurrent(Person::getGender));

2 个答案:

答案 0 :(得分:3)

您可以链接您的分组收藏家,这将为您提供多级地图。但是,如果您想分组超过2个字段,这并不理想。

更好的选择是覆盖equals类中的hashcodePerson方法,以定义两个给定对象的相等性,在这种情况下,这些对象将是所有所述字段。然后,您可以按Person进行分组,即groupingByConcurrent(Function.identity()),在这种情况下,您最终会得到:

ConcurrentMap<Person, List<Person>> resultSet = ....

示例:

class Person {
    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;

        Person person = (Person) o;

        if (name != null ? !name.equals(person.name) : person.name != null) return false;
        if (uid != null ? !uid.equals(person.uid) : person.uid != null) return false;
        return phone != null ? phone.equals(person.phone) : person.phone == null;
    }

    @Override
    public int hashCode() {
        int result = name != null ? name.hashCode() : 0;
        result = 31 * result + (uid != null ? uid.hashCode() : 0);
        result = 31 * result + (phone != null ? phone.hashCode() : 0);
        return result;
    }

    private String name;
    private String uid; // these should be private, don't expose
    private String phone;

   // getters where necessary
   // setters where necessary
}

然后:

ConcurrentMap<Person, List<Person>> resultSet = list.parallelStream()
                .collect(Collectors.groupingByConcurrent(Function.identity()));

答案 1 :(得分:2)

您可以使用of中的Collector静态工厂方法来执行此操作:

Map<String, Set<Person>> groupBy = persons.parallelStream()
    .collect(Collector.of(
        ConcurrentHashMap::new,
        ( map, person ) -> {
            map.computeIfAbsent(person.name, k -> new HashSet<>()).add(person);
            map.computeIfAbsent(person.uid, k -> new HashSet<>()).add(person);
            map.computeIfAbsent(person.phone, k -> new HashSet<>()).add(person);
        },
        ( a, b ) -> {
            b.forEach(( key, set ) -> a.computeIfAbsent(key, k -> new HashSet<>()).addAll(set));
            return a;
        }
    ));

正如Holger在评论中所建议的那样,以下方法可以优先于上述方法:

Map<String, Set<Person>> groupBy = persons.parallelStream()
     .collect(HashMap::new, (m, p) -> { 
         m.computeIfAbsent(p.name, k -> new HashSet<>()).add(p); 
         m.computeIfAbsent(p.uid, k -> new HashSet<>()).add(p); 
         m.computeIfAbsent(p.phone, k -> new HashSet<>()).add(p); 
     }, (a, b) -> b.forEach((key, set) -> {
         a.computeIfAbsent(key, k -> new HashSet<>()).addAll(set));
     });

它使用重载的collect方法,该方法与我上面建议的语句相同。