文章目录
- 前言
- 结论
- 代码示例
- 源码分析
前言
在java8 Stream流中经常有看到这么一种写法Collectors.toMap(Person::getName, Function.identity(),(o1, o2)->o2), 经常会比较好奇o1,o2指的是什么含义,本篇博文主要讲解o1,o2的含义。
结论
这里使用(o1, o2)->o2的写法是因为stream流中使用Collectors.toMap的方式将数据转化为map集合时,若发生key值重复会抛出异常,所以这里指定key值重复时的策略,即当发生key值重复时,使用前面的还是用后面的覆盖更新。
这里的o1,o2可以理解为o1为老的,o2为新的,所以(o1, o2)->o2理解为当key值重复时,用后面的值。
代码示例
需求:将一个Person列表中的姓名属性做为key, Person对象作为value, 姓名字段可能会有重复
public class Test {public static void main(String[] args) throws IOException {List<Person> personList = new ArrayList<>();Person p1 = new Person("张三","男",11);Person p2 = new Person("李四","男",11);Person p3 = new Person("张三","女",11);personList.add(p1);personList.add(p2);personList.add(p3);//key: 姓名 value: Person对象Map<String,Person> personMap = new HashMap<>();personMap = personList.stream().collect(Collectors.toMap(Person::getName, Function.identity(),(o1, o2)->o2));System.out.println(personMap);}
}@Data
@AllArgsConstructor
@NoArgsConstructor
class Person{String name;String sex;int age;
}
执行结果:
{李四=Person(name=李四, sex=男, age=11), 张三=Person(name=张三, sex=女, age=11)}
结论:当指定key值重复策略时,(o1, o2)->o2 值的是用后面的值覆盖新的值
源码分析
- keyMapper:Key 的映射函数
- valueMapper:Value 的映射函数
- mergeFunction:当 Key 冲突时,调用的合并方法
public static <T, K, U>Collector<T, ?, Map<K,U>> toMap(Function<? super T, ? extends K> keyMapper,Function<? super T, ? extends U> valueMapper,BinaryOperator<U> mergeFunction) {return toMap(keyMapper, valueMapper, mergeFunction, HashMap::new);}
default V merge(K key, V value,BiFunction<? super V, ? super V, ? extends V> remappingFunction) {Objects.requireNonNull(remappingFunction);Objects.requireNonNull(value);V oldValue = get(key);V newValue = (oldValue == null) ? value :remappingFunction.apply(oldValue, value);if (newValue == null) {remove(key);} else {put(key, newValue);}return newValue;}
可以看到,在源码中mergeFunction参数也是前面的为oldValue,后面的是新的值