"在Java流API中，groupingBy方法的问题"

Question

我需要创建一个名为 getRentals 的方法，该方法给定一个书籍 ID，返回一个 SortedMap。映射中的键是读者 ID，值是一个字符串，格式为 "DD-MM-YYYY DD-MM-YYYY"，这两个日期分别表示书籍租赁的开始和结束日期。
我已经编写了如下代码，但遇到了一些问题，可能是我对 groupingBy 方法的理解不够准确。

public SortedMap getRentals(String bookID) throws LibException {
    Book b = this.idsColl.get(bookID);

    SortedMap res = b.getRentalsList().stream()
            .collect(Collectors.groupingBy(Rental::getReaderId,
                    r -> {
                        String start = r.getStartDate();
                        String end = r.getEndDate();
                        return start + " " + end;
                    }));
    return null;
}

这里的 idsColl 是一个将唯一书籍 ID 与书籍对象关联起来的映射。

Rental 类的属性如下：

private String readerId;
private String bookId;
private String startDate;
private String endDate;

Book 类的属性如下：

private String copy_id;
private String title;
private boolean rented = false;
private LinkedList rentals = new LinkedList<>();

能否请你帮我理解一下我做错了什么？

我期望得到一个映射，其中包含所有租过这本书的读者及其对应的租赁起止日期字符串。但是 Eclipse 给出错误提示：“Collectors 类型中 groupingBy 方法（Function，Collector>）对于参数 Rental::getReaderId 和 (( r) -> {}) 不适用”。

Eclipse 显示的问题可能指向了lambda表达式部分。

Paul LeBeau · Answer

默认情况下，groupingBy 方法返回一个 HashMap 类型的映射。您可能希望使用 Collectors#groupingBy(Function, Supplier, Collector) 版本来返回类似 TreeMap 的结果，其中键为 K，映射为 M，流元素为 T，下游收集器为：

public SortedMap getRentals(String bookID) throws LibException {
    Book b = this.idsColl.get(bookID);
    return b.getRentalsList().stream().collect(Collectors.groupingBy(
            Rental::getReaderId, // 将流中的每个元素（Rental 对象）映射到键（读者 ID）
            TreeMap::new, // 供应器用于创建 TreeMap 类型的映射
            Collectors.mapping( // 下游收集器 - 将元素进一步映射并收集为值列表
                r -> r.getStartDate() + " " + r.getEndDate(), // 将 Rental 对象映射为日期字符串
                Collectors.toList() // 将所有映射后的日期字符串收集到一个列表中
            ));
}

mnel · Answer

groupingBy 方法会始终收集多个元素并将它们组合成映射的值。在这种情况下，使用 toMap 可能更适合您。但不幸的是，您需要使用一个带有合并函数的重载版本：toMap(keyMapper, valueMapper, mergeFunction, mapFactory)。换句话说，如果键函数生成的键不是唯一的，则您无法免费获得默认抛出异常的合并函数。

public SortedMap getRentals(String bookID) throws LibException {
    Book b = this.idsColl.get(bookID);

    return b.getRentalsList()
            .stream()
            .collect(Collectors.toMap(
                Rental::getReaderId,
                r -> r.getStartDate() + " " + r.getEndDate(),
                (u, v) -> {
                    throw new IllegalStateException("found non-unique keys");
                },
                TreeMap::new
            ));
}

由于在提供供应商时不得不编写合并函数令人不悦，因此我编写了 stream-utils 这个工具库，其中包含了一个内部提供此合并函数的收集器：

public SortedMap getRentals(String bookID) throws LibException {
    Book b = this.idsColl.get(bookID);

    return b.getRentalsList()
            .stream()
            .collect(AdditionalCollectors.toMapWithSupplier(
                Rental::getReaderId,
                r -> r.getStartDate() + " " + r.getEndDate(),
                TreeMap::new
            ));
}