Java:如何使用 Stream API 按条件对实体进行分组？

Question

我有一个用户列表。

class User {
    String name;
}

我需要按名称对用户进行分组。

List<Set<User>> groupedBy;

分组条件是名称相似度，是一些Comparator

Comparator<User> p = new Comparator<User>() {
    // some comparison logic
};

如何使用 Java 8 流按相似名称对用户进行分组？

我尝试使用分区，但它的 API 与我需要的不同。

Map<Boolean, List<User>> collected = users.stream().collect(Collectors.partitioningBy(new Predicate<User>() {
    @Override
    public boolean test(User user) {
        // ...
    }
}, Collectors.toList()));

Answer 1

通常，如果您想通过简单的属性进行分组，您可以使用

Map<String,Set<User>> m=users.stream()
    .collect(Collectors.groupingBy(User::getName, Collectors.toSet()));

但是，如果您有一个复杂的逻辑，表示为现有的Comparator，例如

Comparator<User> comparator=Comparator.comparing(u -> u.getName(), Collator.getInstance());

你可以使用

Map<User,Set<User>> m=users.stream().collect(
    Collectors.groupingBy(u -> u, () -> new TreeMap<>(comparator), Collectors.toSet()));

根据比较器中封装的逻辑进行分组。

在这两种情况下，您都会获得一个Map，可以使用它转换为您想要的列表

List<Set<User>> groupedBy = new ArrayList<>(m.values());

您还可以将最后一步表示为Collector的一部分，例如

List<Set<User>> groupedBy=users.stream().collect(
    Collectors.collectingAndThen(
        Collectors.groupingBy(u -> u, () -> new TreeMap<>(comparator), Collectors.toSet()),
        m -> new ArrayList<Set<User>>(m.values())));