如何使用Java 8流获取每个不同键的第一个值

Question

例如:我有一个包含电子邮件的用户对象列表。我想收集具有不同电子邮件的用户列表(因为具有相同电子邮件的两个用户可能是同一个人，所以我不关心差异)。换句话说，对于每个不同的键(e-mail)，我希望获取使用该键找到的第一个值(user)，而忽略其余的。

这是我想出来的代码:

public List<User> getUsersToInvite(List<User> users) {
    return users
            .stream()
            // group users by their e-mail
            // the resulting map structure: email -> [user]
            .collect(Collectors.groupingBy(User::getEmail, LinkedHashMap::new, Collectors.toList()))
            // browse each map entry
            .entrySet()
            .stream()
            // get the first user object for each entry
            .map(entry -> entry.getValue().get(0))
            // collect such first entries into a list
            .collect(Collectors.toList());
}

有没有更优雅的方式?

Answer 1

您可以使用Collectors.toMap(keyMapper, valueMapper, mergeFunction)，使用电子邮件作为键，用户作为值，并通过始终返回第一个值来忽略键冲突。在Map中，您可以通过调用values()来获取用户。

public List<User> getUsersToInvite(List<User> users) {
    return new ArrayList<>(users.stream()
                                .collect(Collectors.toMap(User::getEmail, 
                                                          Function.identity(), 
                                                          (u1, u2) -> u1))
                                .values());
}