我有一个集合,其字段类型为Set,带有一些值.我需要创建一个收集所有这些值的新集.
我想知道使用lambda表达式是否可行.
以下是代码行：
Set<String> teacherId = batches.stream()
.filter(b -> !CollectionUtils.isEmpty(b.getTeacherIds()))
.map(b -> b.getTeacherIds())
.collect(Collectors.toSet());
问题是post map操作,它包含一组字符串.因此,收集操作返回Set< Set< String>>但我希望将所有值汇总到一个集合.
解决方法
您需要使用
flatMap而不是map：
Set<String> teacherIds =
batches.stream()
.flatMap(b -> b.getTeacherIds().stream())
.collect(Collectors.toSet());
请注意,对于空集合,过滤是多余的 – 流式空集合只会导致空流,这不会影响最终结果.但是,如果getTeacherIds()可以返回null,那么你仍然需要处理它.使用filter(Objects :: nonNull)就足够了,并且可以节省对Apache Commons的依赖.
