我如何使用Stream API合并然后区分集合？[副本]

Question

class MyObject {
  String foo;
  MyObject( String foo ) {
    this.foo = foo;
  }
  public String getFoo() { return foo; }
}


Collection<MyObject> listA = Arrays.asList("a", "b", "c").stream().map(MyObject::new)
        .collect(Collectors.toList());

Collection<MyObject> listB = Arrays.asList("b", "d").stream().map(MyObject::new)
        .collect(Collectors.toList());


// magic

我如何合并和消除列表的重复，以便得到的列表应该是包含“a”，“b”，“c”，“d”的MyObjects？

注意：这是对我们实际需要进行重复数据删除的方法的简化，这些方法实际上是hibernate加载的实体的复杂DTO，但是这个示例应该充分展示了目标。

Answer 1

JDK开发人员讨论了这样的特性（参见JDK-8072723)，并且可能包含在Java-9中（尽管没有保证）。我开发StreamEx库已经有了这样的特性，所以您可以使用它：

List<MyObject> distinct = StreamEx.of(listA).append(listB)
                                  .distinct(MyObject::getFoo).toList();

Streamex类是一个增强的Stream，它与JDK流完全兼容，但有许多附加操作，包括distinct（函数），它允许您为distinct操作指定键提取器。在内部，它非常类似于@fge提出的解决方案。

您还可以考虑编写自定义收集器,该收集器将结合获取不同的对象和将它们存储到列表中：

public static <T> Collector<T, ?, List<T>> distinctBy(Function<? super T, ?> mapper) {
    return Collector.<T, Map<Object, T>, List<T>> of(LinkedHashMap::new,
        (map, t) -> map.putIfAbsent(mapper.apply(t), t), (m1, m2) -> {
            for(Entry<Object, T> e : m2.entrySet()) {
                m1.putIfAbsent(e.getKey(), e.getValue());
            }
            return m1;
        }, map -> new ArrayList<>(map.values()));
}

这个收集器将结果中间地收集到map ，其中Key是提取的键，Element是相应的流元素。为了确保在所有重复的元素中保留第一个元素，使用LinkedHashMap。最后，您只需要将此映射的values()转储到列表中。所以现在你可以写：

List<MyObject> distinct = Stream.concat(listA.stream(), listB.stream())
                                .collect(distinctBy(MyObject::getFoo));

如果您不关心结果集合是否为list，甚至可以删除new arrayList<>()步骤（只需使用map::values作为完成器）。如果您不关心顺序，还可以进行更多的简化：

public static <T> Collector<T, ?, Collection<T>> distinctBy(Function<? super T, ?> mapper) {
    return Collector.<T, Map<Object, T>, Collection<T>> of(HashMap::new,
            (map, t) -> map.put(mapper.apply(t), t), 
            (m1, m2) -> { m1.putAll(m2); return m1; }, 
            Map::values);
}

StreamEx库中也提供了这样的收集器（保留顺序并返回列表)。