如何在 java8 中使用 Lambda 获得相同的结果

Question

for (GroupEntity groupEntity : groups) {
        // find roles belongs to group
        Collection<RoleEntity> groupRoles = groupEntity.getRoles();
        for (RoleEntity roleEntity : groupRoles) {
            if (roleEntity.getType() == RoleType.MODULE) {
                ModuleEntity module = roleEntity.getModule();
                if (!modules.contains(module)) {
                    // prevent duplicate module
                    modules.add(module);
                }
            } else if (roleEntity.getType() == RoleType.OUTLINE) {
                OutlineEntity outline = roleEntity.getOutline();
                //prevent duplicate outline
                if (!outlines.contains(outline)) {
                    outlines.add(outline);
                }
            }
        }
    }

我想用 Lambda 得到同样的结果。我试过了，但结果却是List<List<RoleEntity>>。

    List<List<RoleEntity>> roles= groups.stream().map(g->g.getRoles().stream()
            .filter(distinctByKey(RoleEntity::getId))
            .collect(Collectors.toList())
            .stream().filter(r->RoleType.MODULE.equals(r.getType()))
            .collect(Collectors.toList())
            ).collect(Collectors.toList());

我知道 lambda 对 coder 来说是一个了不起的想法。但是学习它似乎并不那么容易。希望有人可以帮助我解决问题。谢谢！ :XD

Answer 1

你需要使用groupingBy

    Map<RoleType, Set<Object>> roleTypeSetMap = groups.stream()
            .flatMap(g -> g.getRoles().stream())
            .collect(Collectors.groupingBy(RoleEntity::getType, Collectors.mapping(r -> (r.getType() == RoleType.MODULE) ? r.getModule() : r.getOutline(), Collectors.toSet())));

如果ModuleEntity 和OutlineEntity 有一个共同的父级（比如说Entity），我们可以将Set<Object> 更改为Set<Entity>