何用Java8 Stream API进行数据抽取与收集

Posted 2020-11-27 逃离沙漠

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了何用Java8 Stream API进行数据抽取与收集相关的知识，希望对你有一定的参考价值。

原稿于3.2日发布，然而事情并没完，我发现必须得补充一个常见的坑，为了防止大家采坑，我在更新了本文的后两段。

上一篇中我们通过一个实例看到了Java8 Stream API 相较于传统的的Java 集合操作的简洁与优势，本篇我们依然借助于一个实际的例子来看看Java8 Stream API 如何抽取及收集数据。
备注：上一篇内容：如何用Java8 Stream API找到心仪的女朋友

目标 & 背景

我们以“处理订单数据”为例，假设我们的应用是一个分布式应用，有"订单应用"，"物流应用"，"商品应用”等都是独立的服务。本次我们的目的需要展示订单列表完整数据：

1.查询订单列表。

2.批量查询物流信息。

3.将物流信息填充到订单主信息中。

假设我们定义了一个订单类，具有几个关键的属性：订单号，状态，订单价，快递信息。如下所示：

class Order{
    String orderSeq;
    String status;
    double totalPrice;
    String expressInfo;
    // 省略get,set及hashCode等方法
}

我们定义了一个快递信息类，几个关键的属性：订单号，物流公司，物流单号，物流状态。如下所示：

class ExpressInfo{
    String orderSeq;
    String expressName;
    String expressNo;
    String createTime;
    String statusInfo;
    // 省略get,set及hashCode等方法
}

Java7 实现

获取订单列表 & 抽取订单号

   List<Order> orderList = getOrderList();
    // 抽取 订单号
    List<String> orderSeqList = new ArrayList<>();
    for (Order order : orderList) {
        orderSeqList.add(order.getOrderSeq());
    }

这里我们获取了订单列表orderList,此时expressInfo里边是没有数据的。这里抽取单号依然是Java传统的写法。

批量查询快递信息 & 组装订单-快递信息 map

由于我们是通过调用远程服务来获取快递信息，为了减少网络通信次数，我们采取批量查询的方式。这也是为什么，上一步中我们要抽取订单号
下面我们来获取快递信息

// 调用远程服务，
List<ExpressInfo> expressInfos = RpcGetExpressInfoBatch(orderSeqList);
// 组装 订单-快递 关系map
Map<String,String> orderExpressMap = new HashMap<>();
for(ExpressInfo e: expressInfos){
    orderExpressMap.put(e.getOrderSeq(),e.getStatusInfo());
}

这里组装map，也依然是Java7常用的写法。

组合数据,将快递信息填充进订单

for(Order order:orderList){
    String expressInfo = orderExpressMap.get(order.getOrderSeq());
    order.setExpressInfo(expressInfo);
}

至此，我们使用Java7 的写法，完成了开篇设定的目标。下面我们看Java8的写法

Java8 实现

获取订单列表 & 抽取订单号

// 获取列表
List<Order> orderList = getOrderList();
// 抽取单号
List<String> orderSeqs = orderList.stream()
        .map(Order::getOrderSeq)
        .collect(Collectors.toList());

这里我们使用了stream.map,在map()中，我们的写法是Order::getOrderSeq表示调用Order对象的getOrderSeq()方法来抽取订单号。
这里的::叫“方法引用”，是Java8中的新写法。
在map()后面紧跟的是collect收集器，他将抽取的数据toList(),于是我们得到了最终的List。

批量查询快递信息 & 组装订单-快递信息 map

下面我们仍然是通过远程调用来获取快递信息，然后使用Java8的语法建立一个订单-快递关联信息的map。

List<ExpressInfo> expressInfos = RpcGetExpressInfoBatch(orderSeqList);
Map<String,String> orderExpressMap =expressInfos.stream()
        .collect(Collectors.toMap(ExpressInfo::getOrderSeq,ExpressInfo::getStatusInfo));

这里代码比Java7的要少吧，且一目了然，这里用strean().collect来收集数据，收集成什么形式呢？看名知意，Collectors.toMap收集成Map,收集什么数据呢？toMap()中，写了ExpressInfo::getOrderSeq及ExpressInfo::getStatusInfo，即：抽取orderSeq作为key，statusInfo作为value。

通过这里的Collectors.toMap收集器，我们很方便的获得了订单-物流关系数据map。

组合数据,将快递信息填充进订单

经过上面的两步，我们得到了符合我们要求的数据，现在我们需要将快递信息填充进订单，代码如下：

orderList.stream().forEach(o -> o.setExpressInfo(orderExpressMap.get(o.getOrderSeq())));

你没看错，就只有这么一行。

补充说明： key 重复的处理

上面的代码中，我们使用如下代码来从list中收集Map:

Collectors.toMap(ExpressInfo::getOrderSeq,ExpressInfo::getStatusInfo))

这里如果expressInfos这个list中有重复数据，那么orderSeq就会有重复的，这种情况下就会报一个错：

java.lang.IllegalStateException: Duplicate key
想模拟这个错误很简单，往expressInfos中add两条一样的数据，运行上面的抽取map代码就会报错。
为了解决这个问题，我们需要这样写：

List<ExpressInfo> expressInfos = RpcGetExpressInfoBatch(orderSeqList);
 Map<String,String> orderExpressMap =expressInfos.stream()
          .collect(Collectors.toMap(ExpressInfo::getOrderSeq,ExpressInfo::getStatusInfo,(v1,v2)->v1));

这里，我们主要的变动是这一句：

Collectors.toMap(ExpressInfo::getOrderSeq,ExpressInfo::getStatusInfo,(v1,v2)->v1)

我们在在Collectors.toMap中增加了一个lumda表达式：

(v1,v2)->v1

意思是当有重复数据v1,v2出现的时候，我们选择v1。当然，你可以根据你的情况选择v2。
选择v1,或者v2，需要根据你的业务来权衡。比如list中的数据按照时间先后排序的，我们取最新的就选v2，反之取v1。

总结

本节，我们使用Java8 Stream API，完成了数据的抽取和收集，使用了map(),和collect()来完成数据的抽取和收集，并了解了两种收取方式toList和toMap。同时我们也知道应对list中有重复数据导致报错的问题。所以，以后如果有人问你"Java8 stream 如何获取对象的某个属性list啊？"，“java8 stream 如何获取指定数据组装成map啊？”，你就可以把本文中的方法告诉他了。

除此之外，Java8 Streap API 还有分组等功能，后面再说。你也可以关注我的公众号，第一时间收到推送。

alt 逃离沙漠公众号