Java 8 流图自定义函数并将其转换为 Map答案

【问题标题】：Java 8 stream map custom function and convert it to MapJava 8 流图自定义函数并将其转换为 Map
【发布时间】：2020-09-17 16:22:20
【问题描述】：

我有以下对象：

public class Book {

    private Long id;
    private Long bookId;
    private String bookName;
    private String owner;
}

由下表表示：

基本上，一本书可以由多个所有者拥有，即所有者“a”拥有图书 1 和 2。

我有一个基本函数，当传递一个书本对象时，它会在一个列表中给出它的所有者。

private List<String> getBookToOwner(Book book) {
        List<String> a = new ArrayList<>();
        if (book.getOwner() != null && !book.getOwner().isEmpty()) {
            a.addAll(Arrays.asList(book.getOwner().split("/")));
        }
        return a;
}

我想用它来应用到每本书，检索它们的所有者并创建以下地图。

Map<String, List<Long>> ownerToBookMap;

像这样：

如何在此处使用流？

//books is List<Book>
Map<String, List<Long>> ownerToBookMap = books.stream().map( 
// apply the above function to get its owners, flatten it and finally collect it to get the above Map object
// Need some help here..
);

【问题讨论】：

只是一个建议：不要在单个列中包含多个值（在您的情况下为所有者列）。因此，要修复它，我建议您与 Owner 表创建多对多关系。在这种情况下，将创建一个中间映射表。在线查看更多详细信息。

标签： java-8 java-stream collect

【解决方案1】：

您可以从书中获取所有者列表，然后使用 flatMap 将所有者和映射为 bookId 和 owner 对。然后使用groupingBy按owner分组，收集owner的bookId列表。

Map<String, List<Long>> ownerToBookMap = 
     books.stream()
          .flatMap(b -> getBookToOwner(b)
                         .stream()
                         .map(o -> new AbstractMap.SimpleEntry<>(o, b.getBookId())))
          .collect(Collectors.groupingBy(Map.Entry::getKey,
                     Collectors.mapping(Map.Entry::getValue, Collectors.toList())));

【讨论】：

Map.entry() 可以从 Java 9 开始使用，而不是 SimpleEntry

【解决方案2】：

将所有者平面映射为一个，创建条目，键为 single owner，值为 bookId。然后按键（owner）对结构进行分组。最后使用Collectors::mapping 得到bookIds 的List 而不是实际的条目：

List<Book> books = ...

Map<String, List<Long>> booksByOwner = books.stream()
    .flatMap(book -> Arrays.stream(book.getOwner().split("/"))
             .map(owner -> new AbstractMap.SimpleEntry<>(owner, book.getBookId())))
    .collect(Collectors.groupingBy(
             AbstractMap.SimpleEntry::getKey, 
             Collectors.mapping(AbstractMap.SimpleEntry::getValue, Collectors.toList())));

【讨论】：

因此，例如，如果一本书有三个所有者，您可以创建三个独立的 Book 对象，其中每个 Book 恰好有一个所有者。然后您可以执行groupingBy，因为每个Book 只有一个所有者。这不是很贵（就时间和资源而言）？
实际上，短寿命的对象几乎立即被垃圾收集器消耗掉。您可以改用AbstractMap.SimpleEntry。

【解决方案3】：

我使用reduce 而不是map。

Map<String, List<Long>> ownerToBookMap = books.stream().reduce(
    HashMap::new,
    (acc,b) -> {
        getBookToOwner(b).stream().forEach( o -> {
            if (!acc.containsKey(o))
                acc.put(o, new ArrayList<Long>());
            acc.get(o).put(b.bookId);
        });
        return acc;
    }
).get();

【讨论】：

@Abra 对不起，错过了一些东西.. 我刚刚更新了答案
我在尝试您的第一个选项时遇到了一些错误。后来我使用了@Rono 的上述解决方案。