将 Collectors maxBy 与 groupingBy 一起使用时展开可选

Question

我有一个带有String 和int 字段的类。

public class Data {
    private String name;
    private int value;

    private Data(String name, int value) {
        this.name = name;
        this.value = value;
    }

    public String getName() {
        return name;
    }

    public int getValue() {
        return value;
    }
}

我有一个List<Data>，如果我想创建一个地图（分组）来了解每个 name 的 最大值 的 Data 对象，我可以这样做,

Map<String, Optional<Data>> result = list.stream()
            .collect(Collectors.groupingBy(Data::getName,
                    Collectors.maxBy(Comparator.comparing(Data::getValue))));

但是，由于Collectors.maxBy 的合同，RHS 包含在 Optional 中，但我想要 Map<String, Data>。

我可以想到一个选项使用Collectors.collectingAndThen，但我将不得不创建一个不必要的中间映射。

list.stream()
         .collect(Collectors.collectingAndThen(Collectors.groupingBy(Data::getName,
                Collectors.maxBy(Comparator.comparing(Data::getValue))),
                map -> map.entrySet().stream()
                        .collect(Collectors.toMap(Map.Entry::getKey,
                                 e -> e.getValue().get()))));

有没有一种惯用的方式来使用流来实现这一点？

Answer 1

您可以将BinaryOperator.maxBy 与Collectors.toMap 一起使用

Map<String, Data> result = list.stream()
        .collect(Collectors.toMap(Data::getName, Function.identity(),
            BinaryOperator.maxBy(Comparator.comparing(Data::getValue))));