Java按对象属性分组并获取最大值

Question

我有Documents对象要按文档ID分组。分组后，我想获得它们的“最大值”。这就是我目前掌握的：

List<Document> docList = getDocuments(...);
Map<Long, Document> docIdsToLatestDocVersions = docList.stream()
     .collect(Collectors.groupingBy(
           Document::getDocumentId,
           Collectors.reducing(BinaryOperator.maxBy(Comparator.comparing( Function.identity() ))
     ));

文档类：

class Document {
    int documentId;
    int majorVersion;
    int minorVersion;

    @Override
    public int compareTo(Document document) {
        return new CompareToBuilder()
            .append(this.getDocumentId(), document.getDocumentId())
            .append(this.getMajorVersion(), document.getMajorVersion())
            .append(this.getMinorVersion(), document.getMinorVersion())
            .toComparison();
    }
}

重要的是，我已经实现了一个compareTo函数。我不确定在< code>groupingBy子句的< code>reducer参数中放什么。我也试过:

Map<Long, Document> docIdsToLatestDocVersions = docList.stream()
    .collect(Collectors.toMap(
          Document::getDocumentId, Function.identity(),
          BinaryOperator.maxBy(Comparator.comparing(d -> d))));

但无济于事。

Answer 1

在第一个代码段中，Collectors.reduce 将返回“可选”，您需要将结果地图的类型更改为“地图”

Map<Integer, Document> docIdsToLatestDocVersions =
     docList.stream()
            .collect(Collectors.groupingBy(Document::getDocumentId,
                                           Collectors.collectingAndThen(
                                                   Collectors.reducing(BinaryOperator.maxBy(Document::compareTo)),
                                                   Optional::get)));

但是正如您已经提到并尝试过的，无论何时当您在分组后遇到选项时，最好检查一下< code>Collectors.toMap是否是更好的选择。在这种情况下，它确实更具可读性:

Map<Integer, Document> docIdsToLatestDocVersions_second =
     docList.stream()
            .collect(Collectors.toMap(Document::getDocumentId,
                                      Function.identity(),
                                      BinaryOperator.maxBy(Document::compareTo)));