首页手记别再这么写代码了，这几个方法不香吗？

别再这么写代码了，这几个方法不香吗？

标签：

Java

JDK8 应该是 Java 中最坚挺一个版本，这个版本新增很多特性，让我们开发起来多了很多便利。

不过最近 Review 项目代码的时候发现，虽然很多项目工程已经使用了 JDK8，但是工程代码却很少使用到 JDK8 新特性、新方法。

如果单从代码正确性上来说，老方式写法写当然没有什么问题，那唯一的缺点其实就是代码行数比较多，比较繁琐。

那同样的需求，使用 JDK8 新方法，其实几行代码就可以搞定，这样代码就会变得非常简洁。

今天就以三个比较常见的场景为例，教你几招，使用 JDK8 Map新增的方法简化代码开发。

下面就来看看这次即将用到 Map几个新方法：

> 欢迎关注我的公众号：小黑十一点半，获得日常干货推送。

预防空指针问题

日常开发中我们通常会从 Map获取元素，然后进行相关的业务处理，示例代码如下：

Map map = new HashMap();
map.put("公号", "小黑十一点半");
map.put("主理人", "楼下小黑哥");
// 可能存在 NPE 问题
System.out.println(map.get("支付").toUpperCase());

如果就像示例代码直接处理，一旦 Map中相应元素不存在，那么我们就会碰到空指针问题。

为了解决这个问题，通常我们可以先判断一下元素是否为 null，如果不为 null，再做相应的业务处理。

// 第一种 if 判空
String value = map.get("支付");
if (!Objects.isNull(value)) {
    System.out.println(value.toUpperCase());
}

这种方式唯一劣势就是代码处理上比较繁琐，不是很简洁。

所以针对这种情况，其实可以使用条件运算符，设置一个默认空值，从而避免后续处理发生空指针。

// 第一种 if 判空
String value = map.get("支付");
// 第二种 条件运算符
value = Objects.isNull(value) ? "" : value;

这种方式比较简洁，所以日常开发中我比较喜欢用这种方式。

> ps: 这里的前提，空字符串对于业务没有特殊意义。如果存在特殊意义，那就不能使用这种方式了。

那如果使用 JDK8 ，其实就很方便了，我们就可以使用 Map#getOrDefault直接代替条件运算符。

// 等同于条件运算符的效果： Objects.isNull(value) ? "" : value;
String value = map.getOrDefault("支付","");

借助 Map#getOrDefault 一行代码直接搞定，就是这么简单。

如果你还在使用 JDK8 之前的版本，没办法使用这个方法。没关系，我们可以借助 Apache Common-Lang3 提供的工具类 MapUtils 避免空指针。

// Apache MapUtils
String value = MapUtils.getString(map, "支付", "");

MapUtils这个工具类相对于Map#getOrDefault有一个好处，针对传入 Map为 null 的情况，可以设置默认值。

假设我们是从 POJO对象获取 Map 参数，这个时候为了防止空指针，我们就需要提前做一个空指针的判断。

不过如果使用 MapUtils，那我们就不需要判断是否为 null，方法内部已经封装这个逻辑。

MapUtils.getString(pojo.getMap(),"支付", "");

巧用 computeIfAbsent

日常开发中，我们会碰到这类场景，需要一个键需要映射到多个值，这个时候我们可以使用 Map>这个结构。

此时添加元素的时候，我们需要做一些判断，当内部元素不存在时候主动创建一个集合对象，示例代码如下：

Map&gt; map = new HashMap();

List classify = map.get("java框架");
if (Objects.isNull(classify)) {
    classify = new ArrayList&lt;&gt;();
    classify.add("Spring");
    map.put("java框架", classify);
} else {
    classify.add("Spring");
}

上面的代码比较繁琐，到了 JDK8，Map新增一个 computeIfAbsent方法：

default V computeIfAbsent(K key,
        Function&lt;? super K, ? extends V&gt; mappingFunction) {

如果 Map中 key 对应的 value 不存在，则会将 mappingFunction 计算产生的值作为保存为该 key 的 value，并且返回该值。否则不作任何计算，将会直接返回 key 对应的 value。

利用这个特性，我们可以直接使用 Map#computeIfAbsent一行代码完成上面的场景，示例代码如下：

map.computeIfAbsent("java框架", key -&gt; new ArrayList&lt;&gt;()).add("Spring");

那其实 Map 中还有一个方法 putIfAbsent，功能跟 computeIfAbsent比较类似。

那刚开始使用的时候，误以为可以使用 putIfAbsent完成上面的需求：

// ERROR:会有 NPE 问题
map.putIfAbsent("java框架", new ArrayList&lt;&gt;()).add("Spring");

那其实这是错误的，当 Map 中 key 对应 value 不存在的时候，putIfAbsent将会直接返回 null。

而 computeIfAbsent将会返回 mappingFunction计算之后的值，像上面的场景直接返回就是 new ArrayList。

这一点需要注意一下，切勿用错方法，导致空指针。

最后针对上面这种一个键需要映射到多个值，其实还有一个更优秀的解决办法，使用 Google Guava 提供的新集合类型 Multiset，以此快速完成一个键需要映射到多个值的场景。

示例代码如下：

ArrayListMultimap multiset= ArrayListMultimap.create();
multiset.put("java框架","Spring");
multiset.put("java框架","Mybatis");
// java框架---&gt;Spring,Mybatis

单词统计

假设有如下需求，我们需要统计一段文字中相关单词出现的次数。那实现方式其实很简单，使用 Map存储相关单词的次数即可，示例代码如下：

Map countMap = new HashMap();
Integer count = countMap.get("java");
if (Objects.isNull(count)) {
    countMap.put("java", 1);
} else {
    countMap.put("java", count++);
}

这类代码是不是很熟悉？同样比较繁琐。

接下来我们可以使用 JDK8 Map 新增方法进行改造，这次使用上面用过的 getOrDefault 再加 put 方法快速解决，示例代码如下：

// getOrDefault
Integer count = countMap.getOrDefault("java",0);
countMap.put("java", count + 1);

那其实我们还有一种办法，这次我们使用 Map#merge这个新方法，一句代码完成上述需求，示例代码如下：

countMap.merge("java", 1, Integer::sum);

说真的，刚看到 merge这个方法的时候还是有点懵，尤其后面直接使用 lambda 函数，让人不是很好理解。

这里先将lambda 函数还原成正常类，给大家着重解释一下这个方法：

countMap.merge("java", 1, new BiFunction() {
    @Override
    public Integer apply(Integer oldValue, Integer newValue) {
        return Integer.sum(oldValue,newValue);
    }
});