Java8中Map新增的getOrDefault(),putIfAbsent()和computeIfAbsent()三个方法

先说一下,在 Java8 中 computeIfAbsent 方法有一个严重的死循环 bug,Java9 中已经修复了,这个后面我们单独列举文章来详细说它是如何产生以及如何被修复的。

实际上从 Java 8 开始,Map 提供了 computeIfAbsent() 方法。为了讲明白它的用法,我们先来看一个代码场景。

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

这段代码看起来很多人都写过,所以 Java8 推出了 computeIfAbsent 方法,用来简化整个代码块,使其只需要一行代码就可以搞定。

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

其中变量 k 是 Map 的 key。

是不是很方便?但是除此之外,Map 还有两个方法:getOrDefault() 和 putIfAbsent(),这三个方法都接受 Key 和一个“默认值”作为参数,且返回一个 Value。如果不小心把它们搞混用错了,可能会带来大问题。下面分别介绍下。

computeIfAbsent

V computeIfAbsent(K, Function<? super K, ? extends V>)

这个方法有两个参数,Key 和一个根据 Key 来产生 Value 的 Function;然后返回一个 Value。

这个方法会检查 Map 中的 Key,如果发现 Key 不存在或者对应的值是 null,则调用 Function 来产生一个值,然后将其放入 Map,最后返回这个值;否则的话返回 Map 已经存在的值。

补充:在 ConcurrentHashMap 中,这个方法保证线程安全,且多线程并发执行的情况下,Function 参数只会被调用一次。所以这个方法完全可以代替 DCL(Double Check Lock)写法。

getOrDefault

V getOrDefault(Object, V)

这个方法同样检查 Map 中的 Key,如果发现 Key 不存在或者对应的值是 null,则返回第二个参数即默认值。要注意,这个默认值不会放入 Map。所以如果你这样写:

Map<String, List<String>> map = new HashMap<>();
map.getOrDefault("codedq", new ArrayList<>()).add("");

执行完之后 map 仍然是空的!

putIfAbsent

V putIfAbsent(K, V)

这个方法的逻辑完全不同,注意它不是一个 get() 方法,而是 put() 方法的变种!这个方法的逻辑是,如果 Key 不存在或者对应的值是 null,则将 Value 设置进去,然后返回 null;否则只返回 Map 当中对应的值,而不做其他操作。

所以显而易见,在最开始的例子中,如果将 computeIfAbsent() 替换成其他两个方法都是错的。

目前根据我在 codereview 以及面试老程序员的过程中,发现很多人不知道 Map 中新增了方法,也从来没用过。还是建议大家在空闲时间多关注一些 Java 不同版本的新特性,不然在编程的道路上,你会吃亏的!

Java8中Map新增的getOrDefault(),putIfAbsent()和computeIfAbsent()三个方法

: » Java8中Map新增的getOrDefault(),putIfAbsent()和computeIfAbsent()三个方法

原创文章,作者:wure,如若转载,请注明出处:https://blog.ytso.com/252249.html

(0)
上一篇 2022年5月4日
下一篇 2022年5月4日

相关推荐

发表回复

登录后才能评论