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

时间:2021-05-19

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

@Datapublic 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 测试及小结

测试方法

@Testpublic 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(),希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对网站的支持!
如果你觉得本文对你有帮助,欢迎转载,烦请注明出处,谢谢!

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章