如何使用Java 8中的方法引用进行地图合并？

Question

我有以下两种调用集合操作的形式，都返回相同的结果，但我仍然不能完全依赖于方法引用，需要一个lambda。

<R> R collect(Supplier<R> supplier,
          BiConsumer<R,? super T> accumulator,
          BiConsumer<R,R> combiner)

为此，考虑以下由100个随机数组成的流

List<Double> dataList = new Random().doubles().limit(100).boxed()
            .collect(Collectors.toList());

1） 以下示例使用纯lambda

Map<Boolean, Integer> partition = dataList.stream()
            .collect(() -> new ConcurrentHashMap<Boolean, Integer>(),
(map, x) ->
{
    map.merge(x < 0.5 ? Boolean.TRUE : Boolean.FALSE, 1, Integer::sum);
}, (map, map2) ->
{
    map2.putAll(map);
});

2） 以下尝试使用方法引用，但第二个参数仍然需要lambda

Map<Boolean, Integer> partition2 = dataList.stream()
            .collect(ConcurrentHashMap<Boolean, Integer>::new, 
(map, x) ->
{
    map.merge(x < 0.5 ? Boolean.TRUE : Boolean.FALSE, 1, Integer::sum);
}, Map::putAll);

在本例中，如何重写java 8中collect方法的第二个参数以使用方法引用而不是lambda？

System.out.println(partition.toString());
System.out.println(partition2.toString());
{false=55, true=45}
{false=55, true=45}

Answer 1

如果您有一个现有的方法可以精确地执行预期的操作，那么方法引用是一个方便的工具。如果需要修改或附加操作，则方法引用没有特殊的语法来支持这一点，除非您将lambda表达式视为该语法。

当然，您可以在类中创建一个新方法来完成所需的任务，并创建对它的方法引用，当代码的复杂性增加时，这是正确的方法，因为这样，它将得到一个有意义的名称并成为可测试的。但对于简单的代码段，可以使用lambda表达式，这只是用于相同结果的更简单语法。从技术上讲，这没有什么区别，只是保存lambda表达式体的编译器生成的方法将被标记为“合成”。

在您的示例中，您甚至不能使用Map::putAll作为合并函数，因为这会覆盖第一个map的所有现有映射，而不是合并值。

正确的实现看起来像

Map<Boolean, Integer> partition2 = dataList.stream()
    .collect(HashMap::new, 
             (map, x) -> map.merge(x < 0.5, 1, Integer::sum),
             (m1, m2) -> m2.forEach((k, v) -> m1.merge(k, v, Integer::sum)));

但您不需要自己实现它。Collector类中已经提供了适当的内置收集器：

Map<Boolean, Long> partition2 = dataList.stream()
    .collect(Collectors.partitioningBy(x -> x < 0.5, Collectors.counting()));