源码解析JDK 1.8 中的 Map.merge()

Map 中ConcurrentHashMap是线程安全的,但不是所有操作都是,例如get()之后再put()就不是了,这时使用merge()确保没有更新会丢失。

因为Map.merge()意味着我们可以原子地执行插入或更新操作,它是线程安全的。

一、源码解析

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;
}

该方法接收三个参数,一个 key 值,一个 value,一个 remappingFunction 。如果给定的key不存在,它就变成了put(key, value);但是,如果key已经存在一些值,我们 remappingFunction 可以选择合并的方式:

  • 只返回新值即可覆盖旧值: (old, new) -> new;
  • 只需返回旧值即可保留旧值:(old, new) -> old;
  • 合并两者,例如:(old, new) -> old + new;
  • 删除旧值:(old, new) -> null。

二、使用场景

merge()方法在统计时用的场景比较多,例如:有一个学生成绩对象的列表,对象包含学生姓名、科目、科目分数三个属性,求得每个学生的总成绩。

2.1 准备数据

学生对象StudentEntity.java

@Data
public class StudentEntity {
 /**
  * 学生姓名
  */
 private String studentName;
 /**
  * 学科
  */
 private String subject;
 /**
  * 分数
  */
 private Integer score;
}

学生成绩数据

private List<StudentEntity> buildATestList() {
 List<StudentEntity> studentEntityList = new ArrayList<>();
 StudentEntity studentEntity1 = new StudentEntity() {{
  setStudentName("张三");
  setSubject("语文");
  setScore(60);
 }};
 StudentEntity studentEntity2 = new StudentEntity() {{
  setStudentName("张三");
  setSubject("数学");
  setScore(70);
 }};
 StudentEntity studentEntity3 = new StudentEntity() {{
  setStudentName("张三");
  setSubject("英语");
  setScore(80);
 }};
 StudentEntity studentEntity4 = new StudentEntity() {{
  setStudentName("李四");
  setSubject("语文");
  setScore(85);
 }};
 StudentEntity studentEntity5 = new StudentEntity() {{
  setStudentName("李四");
  setSubject("数学");
  setScore(75);
 }};
 StudentEntity studentEntity6 = new StudentEntity() {{
  setStudentName("李四");
  setSubject("英语");
  setScore(65);
 }};
 StudentEntity studentEntity7 = new StudentEntity() {{
  setStudentName("王五");
  setSubject("语文");
  setScore(80);
 }};
 StudentEntity studentEntity8 = new StudentEntity() {{
  setStudentName("王五");
  setSubject("数学");
  setScore(85);
 }};
 StudentEntity studentEntity9 = new StudentEntity() {{
  setStudentName("王五");
  setSubject("英语");
  setScore(90);
 }};

 studentEntityList.add(studentEntity1);
 studentEntityList.add(studentEntity2);
 studentEntityList.add(studentEntity3);
 studentEntityList.add(studentEntity4);
 studentEntityList.add(studentEntity5);
 studentEntityList.add(studentEntity6);
 studentEntityList.add(studentEntity7);
 studentEntityList.add(studentEntity8);
 studentEntityList.add(studentEntity9);

 return studentEntityList;
}

2.2 一般方案

思路:用Map的一组key/value存储一个学生的总成绩(学生姓名作为key,总成绩为value)

Map中不存在指定的key时,将传入的value设置为key的值;

当key存在值时,取出存在的值与当前值相加,然后放入Map中。

public void normalMethod() {
 Long startTime = System.currentTimeMillis();
 // 造一个学生成绩列表
 List<StudentEntity> studentEntityList = buildATestList();

 Map<String, Integer> studentScore = new HashMap<>();
 studentEntityList.forEach(studentEntity -> {
  if (studentScore.containsKey(studentEntity.getStudentName())) {
   studentScore.put(studentEntity.getStudentName(),
     studentScore.get(studentEntity.getStudentName()) + studentEntity.getScore());
  } else {
   studentScore.put(studentEntity.getStudentName(), studentEntity.getScore());
  }
 });
 log.info("各个学生成绩:{},耗时:{}ms",studentScore, System.currentTimeMillis() - startTime);
}

2.3 Map.merge()

很明显,这里需要采用remappingFunction的合并方式。

public void mergeMethod() {
 Long startTime = System.currentTimeMillis();
 // 造一个学生成绩列表
 List<StudentEntity> studentEntityList = buildATestList();
 Map<String, Integer> studentScore = new HashMap<>();
 studentEntityList.forEach(studentEntity -> studentScore.merge(
   studentEntity.getStudentName(),
   studentEntity.getScore(),
   Integer::sum));
 log.info("各个学生成绩:{},耗时:{}ms",studentScore, System.currentTimeMillis() - startTime);
}

2.4 测试及小结

测试方法

@Test
public void testAll() {
 // 一般写法
 normalMethod();
 // merge()方法
 mergeMethod();
}

测试结果

00:21:28.305 [main] INFO cn.van.jdk.eight.map.merge.MapOfMergeTest - 各个学生成绩:{李四=225, 张三=210, 王五=255},耗时:75ms
00:21:28.310 [main] INFO cn.van.jdk.eight.map.merge.MapOfMergeTest - 各个学生成绩:{李四=225, 张三=210, 王五=255},耗时:2ms

结果小结

  • merger()方法使用起来在一定程度上减少了代码量,使得代码更加简洁。同时,通过打印的方法耗时可以看出,merge()方法效率更高。
  • Map.merge()的出现,和ConcurrentHashMap的结合,完美处理那些自动执行插入或者更新操作的单线程安全的逻辑.

三、总结

3.1 示例源码

Github 示例代码

总结

以上所述是小编给大家介绍的JDK 1.8 中的 Map.merge(),希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对我们网站的支持!
如果你觉得本文对你有帮助,欢迎转载,烦请注明出处,谢谢!

(0)

相关推荐

  • 源码解析JDK 1.8 中的 Map.merge()

    Map 中ConcurrentHashMap是线程安全的,但不是所有操作都是,例如get()之后再put()就不是了,这时使用merge()确保没有更新会丢失. 因为Map.merge()意味着我们可以原子地执行插入或更新操作,它是线程安全的. 一.源码解析 default V merge(K key, V value, BiFunction<? super V, ? super V, ? extends V> remappingFunction) { Objects.requireNonNu

  • Android 中 SwipeLayout一个展示条目底层菜单的侧滑控件源码解析

    由于项目上的需要侧滑条目展示收藏按钮,记得之前代码家有写过一个厉害的开源控件 AndroidSwipeLayout 本来准备直接拿来使用,但是看过 issue 发现现在有不少使用者反应有不少的 bug ,而且代码家现在貌似也不进行维护了.故自己实现了一个所要效果的一个控件.因为只是实现我需要的效果,所以大家也能看到,代码里有不少地方我是写死的.希望对大家有些帮助.而且暂时也不需要 AndroidSwipeLayout 大而全的功能,算是变相给自己做的项目精简代码了. 完整示例代码请看:GitHu

  • java中break和continue源码解析

    在自己学习java语言的过程中,很容易把break和continue的用法混淆.为了便于以后快速查阅及温习,在此特留学习笔记一份. 简述 在任何迭代语句的主体部分,都可以用break和continue控制循环的流程.其中,break用于强行退出循环,不执行循环中剩余的语句.而continue则停止执行当前迭代,然后退回循环起始处,开始下一次迭代. 源码 下面这个程序向大家展示了break和continue在for和while循环中的例子: package com.mufeng.thefourth

  • 浅谈Vuejs中nextTick()异步更新队列源码解析

    vue官网关于此解释说明如下: vue2.0里面的深入响应式原理的异步更新队列 官网说明如下: 只要观察到数据变化,Vue 将开启一个队列,并缓冲在同一事件循环中发生的所有数据改变.如果同一个 watcher 被多次触发,只会一次推入到队列中.这种在缓冲时去除重复数据对于避免不必要的计算和 DOM 操作上非常重要.然后,在下一个的事件循环"tick"中,Vue 刷新队列并执行实际(已去重的)工作.Vue 在内部尝试对异步队列使用原生的 Promise.then 和 MutationOb

  • Python中getpass模块无回显输入源码解析

    本文主要讨论了python中getpass模块的相关内容,具体如下. getpass模块 昨天跟学弟吹牛b安利Python标准库官方文档的时候偶然发现了这个模块.仔细一看内容挺少的,只有两个主要api,就花了点时间阅读了一下源码,感觉挺实用的,在这安利给大家. getpass.getpass(prompt='Password: ', stream=None) 调用该函数可以在命令行窗口里面无回显输入密码.参数prompt代表提示字符串,默认是'Password: '.在Unix系统中,strea

  • .Net Core中ObjectPool的使用与源码解析

    一.对象池 运用对象池化技术可以显著地提升性能,尤其是当对象的初始化过程代价较大或者频率较高.下面是ObjectPool源码中涉及的几个类.当你看过.Net Core源码很多时,你会发现,微软的开发很多都是这种模式,通过Policy构建Provider,通过Provider创建最终的类. 二.使用 这个组件的目的主要是将对象保存到对象池,用的时候直接去取,不需要重新创建,实现对象的重复利用.但是有个问题,假如对象池中开始没有对象或者取得数量大于对象池中的数量怎么办?在对象池中对象的数量不足时,此

  • C++11中的智能指针shared_ptr、weak_ptr源码解析

    目录 1.前言 2.源码准备 3.智能指针概念 4.源码解析 4.1.shared_ptr解析 4.1.1.shared_ptr 4.1.2.__shared_ptr 4.1.3.__shared_count 4.1.4._Sp_counted_base 4.1.5._Sp_counted_ptr 4.1.6.shared_ptr总结 4.2.weak_ptr解析 4.2.1.weak_ptr 4.2.2.__weak_ptr 4.2.3.__weak_count 4.2.4.回过头看weak_

  • java中CopyOnWriteArrayList源码解析

    目录 简介 继承体系 源码解析 属性 构造方法 add(Ee)方法 add(intindex,Eelement)方法 addIfAbsent(Ee)方法 get(intindex) remove(intindex)方法 size()方法 提问 总结 简介 CopyOnWriteArrayList是ArrayList的线程安全版本,内部也是通过数组实现,每次对数组的修改都完全拷贝一份新的数组来修改,修改完了再替换掉老数组,这样保证了只阻塞写操作,不阻塞读操作,实现读写分离. 继承体系 public

  • 源码解析python中randint函数的效率缺陷

    目录 一.前言 二.对randint()运行效率的测试 三.从源码分析randint()的缺陷 random.random() random.randint() 四.更快的生成随机整数的方法 random.random() 直接使用 getrandbits() 使用 Numpy.random 一.前言 前几天,在写一个与差分隐私相关的简单程序时,我发现了一些奇怪的东西:相对于其他的随机数生成函数,Python的random.randint()函数感觉很慢. 由于 randint() 是 Pyth

  • Vue中$nextTick实现源码解析

    目录 正文 先看一个简单的问题 内部实现 先看第一块: 再看第二块: 然后是第三块: 最后是第四块: 正文 先看一个简单的问题 <template> <div @click="handleClick" ref="div">{{ text }}</div </template> <script> export default { data() { return { text: 'old' } }, methods:

随机推荐