收集器的自定义收集器.groupingBy无法按预期工作

Question

考虑简单的类Foo：

public class Foo {

    public Float v1;
    public Float v2;
    public String name;

    public Foo(String name, Float v1, Float v2) {
        this.name = name;
        this.v1 = v1;
        this.v2 = v2;
    }

    public String getName() {
        return name;
    }
}

现在，我有一个Foos的集合，我想通过Foo::getName将它们分组。我写了一个自定义收集器来做到这一点，但它似乎没有按预期工作。更确切地说，combiner()永远不会被召唤。为什么？

public class Main {

    public static void main(String[] args) {

        List<Foo> foos = new ArrayList<>();
        foos.add(new Foo("blue", 2f, 2f));
        foos.add(new Foo("blue", 2f, 3f));
        foos.add(new Foo("green", 3f, 4f));

        Map<String, Float> fooGroups = foos.stream().collect(Collectors.groupingBy(Foo::getName, new FooCollector()));
        System.out.println(fooGroups);
    }

    private static class FooCollector implements Collector<Foo, Float, Float> {

        @Override
        public Supplier<Float> supplier() {
            return () -> new Float(0);
        }

        @Override
        public BiConsumer<Float, Foo> accumulator() {
            return (v, foo) -> v += foo.v1 * foo.v2;
        }

        @Override
        public BinaryOperator<Float> combiner() {
            return (v1, v2) -> v1 + v2;
        }

        @Override
        public Function<Float, Float> finisher() {
            return Function.identity();
        }

        @Override
        public Set<Characteristics> characteristics() {
            Set<Characteristics> characteristics = new TreeSet<>();
            return characteristics;
        }
    }
}

Answer 1

首先，如果您不使用多个线程（并行流），则不需要调用组合器函数。调用组合器以在流的块上组合操作的结果。这里没有并行性，因此不需要调用组合器。

由于您的累加器功能，您将获得零值。表达方式

v += foo.v1 * foo.v2;

将用新的v对象取代Float。原始累加器对象未被修改;它仍然是0f。此外，与其他数字包装类型（和Float）一样，String是不可变的，无法更改。

你需要一些其他类型的可变的累加器对象。

class FloatAcc {
    private Float total;
    public FloatAcc(Float initial) {
        total = initial;
    }
    public void accumulate(Float item) {
        total += item;
    }
    public Float get() {
        return total;
    }
}

然后你可以修改自定义Collector以使用FloatAcc。供应一个新的FloatAcc，在accumulate函数中调用accumulator等。

class FooCollector implements Collector<Foo, FloatAcc, Float> {
    @Override
    public Supplier<FloatAcc> supplier() {
        return () -> new FloatAcc(0f);
    }
    @Override
    public BiConsumer<FloatAcc, Foo> accumulator() {
        return (v, foo) -> v.accumulate(foo.v1 * foo.v2);
    }
    @Override
    public BinaryOperator<FloatAcc> combiner() {
        return (v1, v2) -> {
            v1.accumulate(v2.get());
            return v1;
        };
    }
    @Override
    public Function<FloatAcc, Float> finisher() {
        return FloatAcc::get;
    }
    @Override
    public Set<Characteristics> characteristics() {
        Set<Characteristics> characteristics = new TreeSet<>();
        return characteristics;
    }
}

通过这些更改，我得到了您的期望：

{green=12.0, blue=10.0}

Answer 2

你有一个解释，为什么当前收集器不能从rgettman工作。

有必要检查一下创建自定义收集器的辅助方法。例如，整个收集器可以更简洁地定义为：

reducing(0.f, v -> v.v1 * v.v2, (a, b) -> a + b)

并不总是可以使用这些方法;但简洁（并且，大概是经过充分测试的）应该使它们成为可能的首选。