如何将对象的两个字段收集到同一个列表中？

Question

我有一个商品对象，它有两个属性：firstCategoryId 和 secondCategoryId。我有一个商品清单，我想获取所有类别 ID（包括 firstCategoryId 和 secondCategoryId）。

我目前的解决方案是：

List<Integer> categoryIdList = goodsList.stream().map(g->g.getFirstCategoryId()).collect(toList());
categoryIdList.addAll(goodsList.stream().map(g->g.getSecondCategoryId()).collect(toList()));

有没有更方便的方式可以在一个语句中获取所有 categoryId？

Answer 1

您可以使用 flatMap 使用单个 Stream 管道来完成此操作：

List<Integer> cats = goodsList.stream()
                              .flatMap(c->Stream.of(c.getFirstCategoryID(),c.getSecondCategoryID()))
                              .collect(Collectors.toList());