使用Java 8流从另一个列表创建对象列表

时间:2017-10-03 10:55:28

标签: java list java-8 java-stream

我有这三个dto类:

public class OrganismeDTO {
    private Long organismeId;
    private String libelle;
    private List<RoleDTO> rolesDTO;
}

public class RoleDTO {
    private String lib;
    private Long roleId;
    private List<ProfilDTO> profilsDTO;
}

public class UtilisateurRoleDto {
    private String organismeLib;
    private String roleLib;
}

DAO我的函数中会返回UtilisateurRoleDto列表。

我想要的是从OrganismeDTO列表中创建一个UtilisateurRoleDto列表(我可以从我的DAO中获取),例如,如果我有一个UtilisateurRoleDto列表如下:

"Organisme 1", "Role 1"
"Organisme 1", "Role 2"
"Organisme 2", "Role 1"
"Organisme 2", "Role 3"
"Organisme 3", "Role 3"

我希望结果中的OrganismeDTO列表如下(其中每个OrganismeDTO都有RoleDTO列表):

"Organisme 1" : ["Role 1", "Role 2"]
"Organisme 2" : ["Role 1", "Role 3"]
"Organisme 3" : ["Role 3"]

如何使用Java 8流实现此功能?

更新

当我构建RoleDTO我需要通过列表注入RoleDTO.profilsDTO时,我从一个带OrganismeDTO.roleId并返回List<ProfilDTO>的函数中获取。

2 个答案:

答案 0 :(得分:1)

目前我能想到的唯一方法是:

Map<String, List<String>> collect = list.stream()
            .collect(Collectors.groupingBy(UtilisateurRoleDto::getOrganismeLib,
                    Collectors.mapping(UtilisateurRoleDto::getRoleLib, Collectors.toList())));

List<OrganismeDTO> result = collect.entrySet().stream()
        .map(entry -> new OrganismeDTO(entry.getKey(),
                entry.getValue().stream()
                        .map(RoleDTO::new)
                        .collect(toList())))
        .collect(Collectors.toList());

首先,您可以将UtilisateurRoleDtoorganismeLib字段进行分组。之后,您处理获得的结果并将其映射到您想要的实体。

答案 1 :(得分:1)

你可以使用自定义收集(我添加了getter和一些明显的构造函数):

Map<String, OrganismeDTO> map = dt.stream().collect(HashMap::new, (m, t) -> {
    m.computeIfAbsent(t.getOrganismeLib(), x -> new OrganismeDTO(t.getOrganismeLib())).getRolesDTO().add(new RoleDTO(t.getRoleLib()));
}, (m1, m2) -> {
    m2.forEach((k,v) -> {
        OrganismeDTO organismeDTO = m1.get(k);
        if (organismeDTO != null ) {
            organismeDTO.getRolesDTO().addAll(v.getRolesDTO());
        } else {
            m1.put(k, v);
        }
    });
});

然后,如果要从值中创建List,那么剩下的就是。

List<OrganismeDTO> list = map.values().stream().collect(Collectors.toList());