You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

Java 8 Stream收集为Map<String,Integer>的实现问题求助

解决Java 8 Streams收集到Map的问题

嘿,我来帮你搞定这个Map转换的问题!你之前的代码之所以没达到预期,是因为toMap里的键值映射写错了——你把方法引用mapMethodNameConvention本身当成了键,而不是映射后的方法名字符串,这肯定得不到你想要的键值对结果。

根据你的需求,我们需要把三个来源的字符串作为键,其中来自getFieldsMappersMetadata的值为2,另外两个来源的值为1,下面给你两种可行的方案:

方案一:合并流后统一收集(推荐)

这种方式更符合Java Streams的函数式风格,通过合并三个流并带上对应的数值,最后一次性收集到Map中,还能处理重复键的情况:

Map<String, Integer> methodNameMap = Stream.concat(
        // 处理fieldsMappersMetadata,每个元素映射为<方法名, 2>的键值对
        typesMapperGenerationInput.getFieldsMappersMetadata().stream()
                .map(mapMethodNameConvention)
                .map(name -> new AbstractMap.SimpleEntry<>(name, 2)),
        // 合并另外两个流,元素映射为<方法名, 1>的键值对
        Stream.concat(
                typesMapperGenerationInput.getFieldsValueProducersMetadata().stream()
                        .map(produceMethodNameConvention)
                        .map(name -> new AbstractMap.SimpleEntry<>(name, 1)),
                typesMapperGenerationInput.getConstantMappingsMetadata().stream()
                        .map(setConstantValueMethodNameConvention)
                        .map(name -> new AbstractMap.SimpleEntry<>(name, 1))
        )
)
// 收集到Map,遇到重复键时保留较大的值(确保来自fieldsMappers的2不会被覆盖)
.collect(Collectors.toMap(
        AbstractMap.SimpleEntry::getKey,
        AbstractMap.SimpleEntry::getValue,
        (existing, newVal) -> Math.max(existing, newVal)
));

说明:

  • 我们把每个流的元素转换成包含目标键值对的SimpleEntry对象,方便后续统一处理
  • 使用Stream.concat合并三个流,避免多次遍历
  • toMap的第三个参数是重复键的处理逻辑:如果同一个方法名来自多个来源,我们保留值更大的那个(也就是优先保留来自fieldsMappersMetadata的2)

方案二:分步填充Map(更直观)

如果你觉得合并流有点复杂,也可以先创建空Map,然后分三次填充,利用putIfAbsent保证来自fieldsMappersMetadata的2不会被覆盖:

Map<String, Integer> methodNameMap = new HashMap<>();

// 先处理fieldsMappersMetadata,直接put值为2的键值对
typesMapperGenerationInput.getFieldsMappersMetadata().stream()
        .map(mapMethodNameConvention)
        .forEach(name -> methodNameMap.put(name, 2));

// 处理另外两个来源,用putIfAbsent确保只有当键不存在时才设置为1
typesMapperGenerationInput.getFieldsValueProducersMetadata().stream()
        .map(produceMethodNameConvention)
        .forEach(name -> methodNameMap.putIfAbsent(name, 1));

typesMapperGenerationInput.getConstantMappingsMetadata().stream()
        .map(setConstantValueMethodNameConvention)
        .forEach(name -> methodNameMap.putIfAbsent(name, 1));

说明:

  • 先填充值为2的键值对,再用putIfAbsent处理值为1的,这样如果键已经存在(来自第一个流),就不会被覆盖
  • 这种写法更直观,适合对Streams合并不太熟悉的场景

内容的提问来源于stack exchange,提问作者Tali

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.05.11 08:00:58