Question

Java这段代码如何优化？List 对象和foreach有点多

public Collection<String> getTypes(@PathVariable long id) {
List<UserGroup> userGroups = this.userGroupRepository.findAllByUserId(id);
if (userGroups==null){
return null;
}
List<Long> gids = userGroups.stream().map(UserGroup::getGroupId).collect(Collectors.toList());
List<GroupRole> groupRoles = this.groupRoleRepository.findAllByGroupIdIn(gids);
if (groupRoles==null){
return null;
}
List<Long> rids = groupRoles.stream().map(GroupRole::getRoleId).collect(Collectors.toList());
List<Role> roles = this.roleRepository.findAllByIdIn(rids);
List<Long> resIds = new ArrayList<>();
for (Role role : roles) {
Collection<Privilege> ps = role.getPrivileges().values();
for (Privilege p:ps) {
if(PrivilegeType.VALUE.equals(p.getResId())) {
resIds.add(p.getResId());
}
}
}
List<ResValue> resValues = this.resValueRepository.findAllByIdIn(resIds);
Set<String> values = new HashSet<>();
for (ResValue resValue: resValues ) {
values.add(resValue.getValue());
}
return values;
}

Answer 1

1、把这些提炼到一个方法里面来完成
2、一句sql语句能搞定的就用一句sql语句。你写的查询多个表java代码肯定没有mysql的速度快
3、方法模块化，把功能尽量小化，然后根据模块划分和封装