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
答案 0 :(得分:7)
您需要使用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>