天天看点

何用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。

下面我们仍然是通过远程调用来获取快递信息,然后使用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 还有分组 等功能,后面再说。你也可以关注我的公众号,第一时间收到推送。

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

Java8系列- 如何用Java8 Stream API找到心仪的女朋友

SpringMVC是怎么工作的,SpringMVC的工作原理

Mybatis Mapper接口是如何找到实现类的-源码分析

小程序云开发:菜鸟也能全栈做产品

CORS详解,CORS原理分析

工作6年,失业19天

Docker & k8s 系列一:快速上手docker

Docker & k8s 系列二:本机k8s环境搭建

Docker & k8s 系列三:在k8s中部署单个服务实例

感谢您的阅读,欢迎您评论交流。PS:博客园已停更,后续请通过邮件或者公众号联系