使用 java 8 流将键值对转换为按键对象映射分组的最快方法

Question

型号：

public class AgencyMapping {
    private Integer agencyId;
    private String scoreKey;
}

public class AgencyInfo {
    private Integer agencyId;
    private Set<String> scoreKeys;
}

我的代码：

List<AgencyMapping> agencyMappings;
 Map<Integer, AgencyInfo> agencyInfoByAgencyId = agencyMappings.stream()
            .collect(groupingBy(AgencyMapping::getAgencyId,
                    collectingAndThen(toSet(), e -> e.stream().map(AgencyMapping::getScoreKey).collect(toSet()))))
            .entrySet().stream().map(e -> new AgencyInfo(e.getKey(), e.getValue()))
            .collect(Collectors.toMap(AgencyInfo::getAgencyId, identity()));

有没有办法获得相同的结果并使用更简单、更快的代码？

Answer 1

您可以通过调用mapping(AgencyMapping::getScoreKey, toSet()) 来简化对collectingAndThen(toSet(), e -> e.stream().map(AgencyMapping::getScoreKey).collect(toSet())))) 的调用。

Map<Integer, AgencyInfo> resultSet = agencyMappings.stream()
                .collect(groupingBy(AgencyMapping::getAgencyId,
                        mapping(AgencyMapping::getScoreKey, toSet())))
                .entrySet()
                .stream()
                .map(e -> new AgencyInfo(e.getKey(), e.getValue()))
                .collect(toMap(AgencyInfo::getAgencyId, identity()));

使用toMap 收集器的另一种查看方式：

Map<Integer, AgencyInfo> resultSet = agencyMappings.stream()
                .collect(toMap(AgencyMapping::getAgencyId, // key extractor
                        e -> new HashSet<>(singleton(e.getScoreKey())), // value extractor
                        (left, right) -> { // a merge function, used to resolve collisions between values associated with the same key
                            left.addAll(right);
                            return left;
                        }))
                .entrySet()
                .stream()
                .map(e -> new AgencyInfo(e.getKey(), e.getValue()))
                .collect(toMap(AgencyInfo::getAgencyId, identity()));

后一个例子可以说比前一个更复杂。尽管如此，您的方法与使用mapping 相比，与上面提到的collectingAndThen 相比，几乎是一种不同的方法。

除此之外，我没有看到您可以使用显示的代码简化任何其他内容。

至于 faster 代码，如果您建议您当前的方法在性能上很慢，那么您可能需要阅读here 的答案，这些答案谈到了您应该考虑何时考虑并行化。

Answer 2

您正在收集到一个中间映射，然后流式传输此映射的条目以创建 AgencyInfo 实例，这些实例最终被收集到另一个映射。

您可以使用Collectors.toMap 直接收集到地图，而不是所有这些，将每个AgencyMapping 对象映射到所需的AgencyInfo 并根据需要合并scoreKeys：

Map<Integer, AgencyInfo> agencyInfoByAgencyId = agencyMappings.stream()
    .collect(Collectors.toMap(
            AgencyMapping::getAgencyId,
            mapping -> new AgencyInfo(
                    mapping.getAgencyId(), 
                    new HashSet<>(Set.of(mapping.getScoreKey()))),
            (left, right) -> {
                left.getScoreKeys().addAll(right.getScoreKeys());
                return left;
            }));

这是通过将流的AgencyMapping 元素按AgencyMapping::getAgencyId 分组来实现的，但将AgencyInfo 对象存储在映射中。我们通过手动映射每个原始AgencyMapping 对象来获得这些AgencyInfo 实例。最后，我们通过 merge 函数 合并地图中已经存在的AgencyInfo 实例，该函数向左折叠 scoreKeys 从一个AgencyInfo 到另一个。

我正在使用 Java 9 的 Set.of 创建一个单例集。如果您没有 Java 9，可以将其替换为 Collections.singleton。