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
相关产品推荐
相关产品推荐

