java - 使用Stream API聚合嵌套列表

假设我有一个Shelf类，每个Shelf具有多个Book。

public class Shelf{
   private String shelfCode;
   private ArrayList<Book> books; //add getters, setters etc.
}

public class Book{
  private String title;
}

现在，通过某种方法，我有一个List的Shelf，每个都包含一些书。如何使用stream将所有书籍收集到此列表？

List<Shelf> shelves = new ArrayList<Shelf>();

Shelf s1 = new Shelf();
s1.add(new Book("book1"));
s1.add(new Book("book2"));

Shelf s2 = new Shelf();
s1.add(new Book("book3"));
s1.add(new Book("book4"));

shelves.add(s1);
shelves.add(s2);

List<Book> booksInLibrary = //??

我在想类似

List<Book> booksInLibrary =
      shelves.stream()
             .map(s -> s.getBooks())
             .forEach(booksInLibrary.addall(books));

但它似乎不起作用，抛出编译错误。

最佳答案

您可以为此使用flatMap

shelves.stream()
       .flatMap(s -> s.getBooks().stream())
       .collect(Collectors.toList());

流式传输过程非常简单:s -> s.getBooks().stream()为每个书架上的每本书提供一个流，flatMap展平所有内容，collect(Collectors.toList())将结果存储在列表中。