收集Set字段的所有值

Question

我有一个集合,其字段类型为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>>但我希望将所有值聚合到一个集合.

Answer 1

您需要使用flatMap而不是map:

Set<String> teacherIds = 
    batches.stream()
           .flatMap(b -> b.getTeacherIds().stream())
           .collect(Collectors.toSet());

请注意,对于空集合,过滤是多余的 - 流式空集合只会导致空流,这不会影响最终结果.但是,如果getTeacherIds()可以返回null,你仍然需要处理它.使用filter(Objects::nonNull)就足够了,并且可以节省对Apache Commons的依赖.