如何在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对于编码器来说是一个了不起的想法.但是学习它似乎并不那么容易.希望有人可以帮我解决问题.谢谢!: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>