如何通过更新重复值来合并两个集合

Question

我有这门课：

class A{
 int count=0;


 //equals and hashCode implemented
}

创建两个集合：

Collection<A>  collection1;//something
Collection<A>  collection2;//something

连接集合：

Stream.concat(collection1.stream(), collection2.stream())
                //do something here???
                .distinct()
                .sorted(new Comparator<A>() {
                  @Override
                  public int compare(A o1, A o2) {

                     return new Integer(o1.count).compareTo(new Integer(o2.count));
                  }
                })
                .collect(Collectors.toList());

我想合并这两个列表，如果有重复，我想更新整数count; 上面的代码完全摆脱了重复

例如， 假设我们在 collection1 和 collection2 中都有一个对象，它们相等 (o1.equals(o2) == true) 。一个有count=10，另一个有count=20 我想拥有一组不同的对象，该对象有count=30

有办法吗？

Answer 1

你不能用你当前的方法做到这一点，你可以使用toMap收集器：

Stream.concat(collection1.stream(), collection2.stream())
       .collect(Collectors.toMap(Function.identity(),Function.identity(),
       (left, right) -> {
          left.setCount(left.getCount() + right.getCount());
          return left;
       })).values()
         .stream()
         .sorted(Comparator.comparingInt(A::getCount))
         .collect(Collectors.toList());

这使用Stream.concat 将collection1 和collection2 合并为一个Stream<A>。

然后我们调用toMap 收集器，其中第一个Function.identity() 选择源的元素作为键，即A，第二个Function.identity() 也选择源的元素作为值；再次A，所以此时，您可以将数据可视化为Map<A, A>。

函数(left, right) -> { ... }被称为合并函数，如果两个给定的对象相等，我们就说取第一个元素的计数并取第二个元素的计数，将它们加在一起并赋值新值返回到我们要保留的元素之一，然后丢弃另一个。

因为您已经覆盖了equals 和hashcode，所以我们不需要显式调用它，而是会隐式调用equals 方法 确定两个给定对象是否相等。

一旦我们将元素收集到Map<A, A>，然后我们调用values() 方法来检索Collection<A>，这使我们能够使用stream() 方法从它创建一个流，并进一步启用我们使用 sorted 方法对流元素进行排序，最后使用toList 收集器将其收集到List<A>。