源码解析JDK 1.8 中的 Map.merge()
Map中ConcurrentHashMap是线程安全的,但不是所有操作都是,例如get()之后再put()就不是了,这时使用merge()确保没有更新会丢失。
因为Map.merge()意味着我们可以原子地执行插入或更新操作,它是线程安全的。
一、源码解析
defaultVmerge(Kkey,Vvalue,BiFunctionremappingFunction){ Objects.requireNonNull(remappingFunction); Objects.requireNonNull(value); VoldValue=get(key); VnewValue=(oldValue==null)?value: remappingFunction.apply(oldValue,value); if(newValue==null){ remove(key); }else{ put(key,newValue); } returnnewValue; }
该方法接收三个参数,一个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 publicclassStudentEntity{ /** *学生姓名 */ privateStringstudentName; /** *学科 */ privateStringsubject; /** *分数 */ privateIntegerscore; }
学生成绩数据
privateListbuildATestList(){ List studentEntityList=newArrayList<>(); StudentEntitystudentEntity1=newStudentEntity(){{ setStudentName("张三"); setSubject("语文"); setScore(60); }}; StudentEntitystudentEntity2=newStudentEntity(){{ setStudentName("张三"); setSubject("数学"); setScore(70); }}; StudentEntitystudentEntity3=newStudentEntity(){{ setStudentName("张三"); setSubject("英语"); setScore(80); }}; StudentEntitystudentEntity4=newStudentEntity(){{ setStudentName("李四"); setSubject("语文"); setScore(85); }}; StudentEntitystudentEntity5=newStudentEntity(){{ setStudentName("李四"); setSubject("数学"); setScore(75); }}; StudentEntitystudentEntity6=newStudentEntity(){{ setStudentName("李四"); setSubject("英语"); setScore(65); }}; StudentEntitystudentEntity7=newStudentEntity(){{ setStudentName("王五"); setSubject("语文"); setScore(80); }}; StudentEntitystudentEntity8=newStudentEntity(){{ setStudentName("王五"); setSubject("数学"); setScore(85); }}; StudentEntitystudentEntity9=newStudentEntity(){{ 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); returnstudentEntityList; }
2.2一般方案
思路:用Map的一组key/value存储一个学生的总成绩(学生姓名作为key,总成绩为value)
Map中不存在指定的key时,将传入的value设置为key的值;
当key存在值时,取出存在的值与当前值相加,然后放入Map中。
publicvoidnormalMethod(){ LongstartTime=System.currentTimeMillis(); //造一个学生成绩列表 ListstudentEntityList=buildATestList(); Map studentScore=newHashMap<>(); 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.3Map.merge()
很明显,这里需要采用remappingFunction的合并方式。
publicvoidmergeMethod(){ LongstartTime=System.currentTimeMillis(); //造一个学生成绩列表 ListstudentEntityList=buildATestList(); Map studentScore=newHashMap<>(); studentEntityList.forEach(studentEntity->studentScore.merge( studentEntity.getStudentName(), studentEntity.getScore(), Integer::sum)); log.info("各个学生成绩:{},耗时:{}ms",studentScore,System.currentTimeMillis()-startTime); }
2.4测试及小结
测试方法
@Test publicvoidtestAll(){ //一般写法 normalMethod(); //merge()方法 mergeMethod(); }
测试结果
00:21:28.305[main]INFOcn.van.jdk.eight.map.merge.MapOfMergeTest-各个学生成绩:{李四=225,张三=210,王五=255},耗时:75ms
00:21:28.310[main]INFOcn.van.jdk.eight.map.merge.MapOfMergeTest-各个学生成绩:{李四=225,张三=210,王五=255},耗时:2ms
结果小结
- merger()方法使用起来在一定程度上减少了代码量,使得代码更加简洁。同时,通过打印的方法耗时可以看出,merge()方法效率更高。
- Map.merge()的出现,和ConcurrentHashMap的结合,完美处理那些自动执行插入或者更新操作的单线程安全的逻辑.
三、总结
3.1示例源码
Github示例代码
总结
以上所述是小编给大家介绍的JDK1.8中的Map.merge(),希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对毛票票网站的支持!
如果你觉得本文对你有帮助,欢迎转载,烦请注明出处,谢谢!
声明:本文内容来源于网络,版权归原作者所有,内容由互联网用户自发贡献自行上传,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任。如果您发现有涉嫌版权的内容,欢迎发送邮件至:czq8825#qq.com(发邮件时,请将#更换为@)进行举报,并提供相关证据,一经查实,本站将立刻删除涉嫌侵权内容。