Java修炼终极指南:179 将自定义代码流式传输到 map

假设我们有以下遗留类：

public class Post {
   
  private final int id;
  private final String title;
  private final String tags;
  public Post(int id, String title, String tags) {
    this.id = id;
    this.title = title;
    this.tags = tags;
  }
  ...
  public static List<String> allTags(Post post) {
       
    return Arrays.asList(post.getTags().split("#"));
  }
}

所以，我们有一个类，它形成了一些博客帖子。每个帖子有几个属性，包括它的标签。每个帖子的标签实际上由一个由井号（#）分隔的标签字符串表示。每当我们需要给定帖子的标签列表时，我们可以调用 allTags() 辅助方法。例如，这里是帖子列表及其标签：

List<Post> posts = List.of(
  new Post(1, "Running jOOQ", "#database #sql #rdbms"),
  new Post(2, "I/O files in Java", "#io #storage #rdbms"),
  new Post(3, "Hibernate Course", "#jpa #database #rdbms"),
  new Post(4, "Hooking Java Sockets", "#io #network"),
  new Post(5, "Analysing JDBC transactions", "#jdbc #rdbms")
);

我们的目标是从这个列表中提取一个 Map<String, List<Integer>>，其中包含每个标签（键）的帖子（值）列表。例如，对于标签 #database，我们有文章 1 和 3；对于标签 #rdbms，我们有文章 1、2、3 和 5，以此类推。

在函数式编程中完成此任务可以通过 flatMap() 和 groupingBy() 实现。这两个在《Java Coding Problems, First Edition》中都有详细说明。简而言之，flatMap() 适用于展平嵌套的 Stream<Stream<R>> 模型，而 groupingBy() 是一个收集器，用于根据某些逻辑/属性将数据分组到一个映射中。我们需要 flatMap()，因为我们有 List<Post>，对于每个 Post，通过 allTags() 嵌套了一个 List<String>（所以，如果我们简单地调用 stream()，那么我们会得到一个 Stream<Stream<R>>）。展平后，我们将每个标签包装在 Map.Entry<String, Integer> 中。最后，我们按标签将这些条目分组到一个 Map 中，如下所示：

Map<String, List<Integer>> result = posts.stream()
  .flatMap(post -> Post.allTags(post).stream()
  .map(t -> entry(t, post.getId())))
  .collect(groupingBy(Entry::getKey,
              mapping(Entry::getValue, toList())));

但是，根据上一个问题，我们知道从 JDK 16 开始，我们可以使用 mapMulti()。所以，我们可以将前面的代码片段重写如下：

Map<String, List<Integer>> resultMulti = posts.stream()
  .<Map.Entry<String, Integer>>mapMulti((post, consumer) -> {
      for (String tag : Post.allTags(post)) {
             consumer.accept(entry(tag, post.getId()));
      }
  })
  .collect(groupingBy(Entry::getKey,
              mapping(Entry::getValue, toList())));

这次，我们节省了 map() 中间操作和中间流。