Java 8集合流 - 将列表转换为Set,转换结果

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());

这将产生平面套装.