Map 的 getOrDefault()，putIfAbsent() 和 computeIfAbse

李義發(fā)布于2019-08-16 15:20 / 1952人閱讀

摘要：如果不小心把它們搞混用錯(cuò)了，可能會(huì)帶來(lái)大問(wèn)題。這個(gè)方法會(huì)檢查中的，如果發(fā)現(xiàn)不存在或者對(duì)應(yīng)的值是，則調(diào)用來(lái)產(chǎn)生一個(gè)值，然后將其放入，最后返回這個(gè)值否則的話返回已經(jīng)存在的值。要注意，這個(gè)默認(rèn)值不會(huì)放入。

假設(shè)我們定義下面一個(gè) Map:

Map> map = new HashMap<>();

如果我們要放一個(gè)元素進(jìn)去，很多人會(huì)這么寫(xiě)：

List list = map.get("list1");
if (list == null) {
  list = new ArrayList<>();
  map.put("list1", list);
}
list.add("A");

實(shí)際上從 Java 8 開(kāi)始，Map 提供了 computeIfAbsent() 方法，我們可以寫(xiě)成一行即可：

map.computeIfAbsent("list1", k -> new ArrayList<>()).add("A");

其中變量 k 是 Map 的 key。

是不是很方便？但是除此之外，Map 還有兩個(gè)方法：getOrDefault() 和 putIfAbsent()，這三個(gè)方法都接受 Key 和一個(gè)“默認(rèn)值”作為參數(shù)，且返回一個(gè) Value。如果不小心把它們搞混用錯(cuò)了，可能會(huì)帶來(lái)大問(wèn)題。下面分別介紹下。

? V computeIfAbsent(K, Function)

這個(gè)方法有兩個(gè)參數(shù)，Key 和一個(gè)根據(jù) Key 來(lái)產(chǎn)生 Value 的 Function；然后返回一個(gè) Value。
這個(gè)方法會(huì)檢查 Map 中的 Key，如果發(fā)現(xiàn) Key 不存在或者對(duì)應(yīng)的值是 null，則調(diào)用 Function 來(lái)產(chǎn)生一個(gè)值，然后將其放入 Map，最后返回這個(gè)值；否則的話返回 Map 已經(jīng)存在的值。

? V getOrDefault(Object, V)

這個(gè)方法同樣檢查 Map 中的 Key，如果發(fā)現(xiàn) Key 不存在或者對(duì)應(yīng)的值是 null，則返回第二個(gè)參數(shù)即默認(rèn)值。要注意，這個(gè)默認(rèn)值不會(huì)放入 Map。所以如果你這樣寫(xiě)：

Map> map = new HashMap<>();
map.getOrDefault("list1", new ArrayList<>()).add("A");

執(zhí)行完之后 map 仍然是空的！

? V putIfAbsent(K, V)

這個(gè)方法的邏輯完全不同，注意它不是一個(gè) get() 方法，而是 put() 方法的變種！這個(gè)方法的邏輯是，如果 Key 不存在或者對(duì)應(yīng)的值是 null，則將 Value 設(shè)置進(jìn)去，然后返回 null；否則只返回 Map 當(dāng)中對(duì)應(yīng)的值，而不做其他操作。

所以顯而易見(jiàn)，在最開(kāi)始的例子中，如果將 computeIfAbsent() 替換成其他兩個(gè)方法都是錯(cuò)的。