Java 8 集合流 - 将列表转换为集合，转换结果 [重复]

Question

使用以下方法成像对象：

class A { List<B> getIds(){...} }

现在我有一个集合 A 作为输入； 我想从中获取一组唯一的 ID，通常你会选择：

Set<B> ids = new HashSet<>();
for(A a : input){
  ids.addAll(a.getIds());
}

有没有办法使用流 API 在一行中做同样的事情，如下所示

Set<List<B>> set = input.stream().map((a) -> a.getIds()).collect(Collectors.toSet());

但是制作扁平的B组

Answer 1

你必须使用flatMap

input.stream()
    .map(a -> a.getIds())
    .flatMap(ids -> ids.stream())
    .collect(Collectors.toSet());

这将产生平面集。