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

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

Map 中ConcurrentHashMap是线程安全的，但不是所有操作都是XALGZxLHON，例如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 buildATestList() {

List 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() {{

seXALGZxLHONtStudentName("李四");

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中。

http://

public void normalMethod() {

Long startTime = System.currentTimeMillis();

// 造一个学生成绩列表

List studentEntityList = buildATestList();

Map 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 studentEntityList = buildATestList();

Map 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 示例代码

总结

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。