溫馨提示×

溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊×
其他方式登錄
點(diǎn)擊 登錄注冊 即表示同意《億速云用戶服務(wù)條款》

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

發(fā)布時間：2020-09-26 00:31:56 來源：腳本之家 閱讀：173 作者：風(fēng)塵博客 欄目：編程語言

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é)

  • merger()方法使用起來在一定程度上減少了代碼量，使得代碼更加簡潔。同時，通過打印的方法耗時可以看出，merge()方法效率更高。
  • Map.merge()的出現(xiàn)，和ConcurrentHashMap的結(jié)合，完美處理那些自動執(zhí)行插入或者更新操作的單線程安全的邏輯.

三、總結(jié)

3.1 示例源碼

Github 示例代碼

總結(jié)

以上所述是小編給大家介紹的JDK 1.8 中的 Map.merge(),希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復(fù)大家的。在此也非常感謝大家對億速云網(wǎng)站的支持！
如果你覺得本文對你有幫助，歡迎轉(zhuǎn)載，煩請注明出處，謝謝！

向AI問一下細(xì)節(jié)

免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點(diǎn)不代表本網(wǎng)站立場，如果涉及侵權(quán)請聯(lián)系站長郵箱：is@yisu.com進(jìn)行舉報，并提供相關(guān)證據(jù)，一經(jīng)查實(shí)，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手