Collectors.reducing 到列表

Question

考虑这个类：

@Data
@AllArgsConstructor
@NoArgsConstructor
class User {
    String name;
    String languages;
}

我有一个List<User>，我想减少语言。输入：

List<User> list = new ArrayList<>();
list.add(new User("sam", "java"));
list.add(new User("sam", "js"));
list.add(new User("apollo", "html"));

预期输出：

[User(name=apollo, languages=html), User(name=sam, languages=java, js)]

我可以使用以下代码实现这一点：

List<User> l = list.stream()
    .collect(Collectors.groupingBy(
         u -> u.name,
         Collectors.reducing((u1, u2) -> 
             new User(u1.name, u1.languages + ", " + u2.languages))))
    .values()
    .stream()
    .filter(user -> user.get() != null)
    .map(user -> user.get())
    .collect(Collectors.toList());

System.out.println(l);

但是我不想创建两个流，可以使用单个流来实现吗？

Answer 1

您可以使用Collectors.toMap()：

List<User> l = new ArrayList<> (list.stream()
    .collect(Collectors.toMap(u -> u.name,
                              u -> new User (u.name,u.languages),
                              (u1, u2) -> new User(u1.name, u1.languages + ", " + u2.languages)))
    .values());

Answer 2

但我不想创建两个流，可以使用单个流来实现吗？

当然这是可能的，您有两个选择，并且在这两个Collect.collectingAndThen选项中您都必须使用从创建的Map.

您需要首先使用以下 lambda 表达式：

BinaryOperator<User> binaryOperator = (u1, u2) -> 
    new User(u1.getName(), String.join(", ", u1.getLanguages(), u2.getLanguages()));

解决方案：

1. Collectors.toMap从中Collectors.collectingAndThen提取值Map<String, User>：

   List<User> l = list.stream()                                       // Stream<User>
       .collect(Collectors.collectingAndThen(
           Collectors.toMap(                                          // Map
               User::getName,                                         // ..name as key
               user -> new User(user.getName(), user.getLanguages()), // ..value as user
               binaryOperator),                                       // ..combined
           map -> new ArrayList<>(map.values())));                    // extracted value
   

2. Collectors.groupingBy从 中提取Collectors.collectingAndThen值：Map<String, Optional<User>>Collectors.reducing

   List<User> ll = list.stream()                     // Stream<User>
       .collect(Collectors.collectingAndThen(
           Collectors.groupingBy(                    // Map<String, Optional<User>>
               User::getName,                        // ... name as key
               Collectors.reducing(binaryOperator)), // ... reduced value
           map -> map.values().stream()              // extracted value
               .filter(Optional::isPresent)          // ... where is Optional present
               .map(Optional::get)                   // ... get it
               .collect(Collectors.toList())));      // ... create a List<User>
   

注意我假设您使用java-8，使用java-11或更高版本可能有不同的解决方案。