快捷導(dǎo)航

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

更新時間：2019年10月10日 10:06:28 作者：風(fēng)塵博客

這篇文章主要介紹了JDK 1.8 之 Map.merge()的相關(guān)知識，本文給大家介紹的非常詳細(xì)，具有一定的參考借鑒價值,需要的朋友可以參考下

Map 中ConcurrentHashMap是線程安全的，但不是所有操作都是，例如get()之后再put()就不是了，這時使用merge()確保沒有更新會丟失。

因?yàn)镸ap.merge()意味著我們可以原子地執(zhí)行插入或更新操作，它是線程安全的。

一、源碼解析

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

該方法接收三個參數(shù)，一個 key 值，一個 value，一個 remappingFunction 。如果給定的key不存在，它就變成了put(key, value)；但是，如果key已經(jīng)存在一些值，我們 remappingFunction 可以選擇合并的方式:

只返回新值即可覆蓋舊值： (old, new) -> new;
只需返回舊值即可保留舊值：(old, new) -> old;
合并兩者，例如：(old, new) -> old + new;
刪除舊值：(old, new) -> null。

二、使用場景

merge()方法在統(tǒng)計時用的場景比較多，例如：有一個學(xué)生成績對象的列表，對象包含學(xué)生姓名、科目、科目分?jǐn)?shù)三個屬性，求得每個學(xué)生的總成績。

2.1 準(zhǔn)備數(shù)據(jù)

學(xué)生對象StudentEntity.java

@Data
public class StudentEntity {
 /**
  * 學(xué)生姓名
  */
 private String studentName;
 /**
  * 學(xué)科
  */
 private String subject;
 /**
  * 分?jǐn)?shù)
  */
 private Integer score;
}

學(xué)生成績數(shù)據(jù)

private List<StudentEntity> buildATestList() {
 List<StudentEntity> studentEntityList = new ArrayList<>();
 StudentEntity studentEntity1 = new StudentEntity() {{
  setStudentName("張三");
  setSubject("語文");
  setScore(60);
 }};
 StudentEntity studentEntity2 = new StudentEntity() {{
  setStudentName("張三");
  setSubject("數(shù)學(xué)");
  setScore(70);
 }};
 StudentEntity studentEntity3 = new StudentEntity() {{
  setStudentName("張三");
  setSubject("英語");
  setScore(80);
 }};
 StudentEntity studentEntity4 = new StudentEntity() {{
  setStudentName("李四");
  setSubject("語文");
  setScore(85);
 }};
 StudentEntity studentEntity5 = new StudentEntity() {{
  setStudentName("李四");
  setSubject("數(shù)學(xué)");
  setScore(75);
 }};
 StudentEntity studentEntity6 = new StudentEntity() {{
  setStudentName("李四");
  setSubject("英語");
  setScore(65);
 }};
 StudentEntity studentEntity7 = new StudentEntity() {{
  setStudentName("王五");
  setSubject("語文");
  setScore(80);
 }};
 StudentEntity studentEntity8 = new StudentEntity() {{
  setStudentName("王五");
  setSubject("數(shù)學(xué)");
  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存儲一個學(xué)生的總成績(學(xué)生姓名作為key,總成績?yōu)関alue)

Map中不存在指定的key時，將傳入的value設(shè)置為key的值；

當(dāng)key存在值時,取出存在的值與當(dāng)前值相加，然后放入Map中。

public void normalMethod() {
 Long startTime = System.currentTimeMillis();
 // 造一個學(xué)生成績列表
 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("各個學(xué)生成績:{},耗時：{}ms",studentScore, System.currentTimeMillis() - startTime);
}

2.3 Map.merge()

很明顯，這里需要采用remappingFunction的合并方式。

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

2.4 測試及小結(jié)

測試方法

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

測試結(jié)果

00:21:28.305 [main] INFO cn.van.jdk.eight.map.merge.MapOfMergeTest - 各個學(xué)生成績:{李四=225, 張三=210, 王五=255},耗時：75ms
00:21:28.310 [main] INFO cn.van.jdk.eight.map.merge.MapOfMergeTest - 各個學(xué)生成績:{李四=225, 張三=210, 王五=255},耗時：2ms

結(jié)果小結(jié)