Java 流 : Grouping, 求和和计数

Question

我是流的新手，但对它的可能性非常感兴趣。

我正在尝试编写一个同时进行分组、计数和求和的流。涉及的数据实际上非常简单，但事实证明编写我需要的流式处理语句具有挑战性，而且我在 Google 搜索中没有看到任何真正有用的东西。

首先，让我描述一下我的数据，然后我将向您展示我是如何解决三分之二问题的。或许你可以告诉我如何填补缺失的部分。

数据是一家销售音乐会门票的公司的门票销售额。每笔销售都包含代理代码、订单号、订单日期和售出的门票数量。因此，它看起来像这样:

AgencyCode  OrderNumber OrderDate  TicketsSold
----------  ----------- ---------  -----------
TW          111111  2016-03-01          4
TW          111112  2016-03-01          2
CP          201000  2016-03-01          3
TW          111113  2016-03-01          8
CP          201001  2016-03-02          2
EL          300001  2016-03-01          4
AS          400000  2016-03-02          2

我试图从这些数据中得到的是一个摘要，显示每个代理代码的订单总数和同一代理代码售出的门票总数。因此，我想为这组特定数据获取的值是:

AgencyCode  Orders   TicketsSold
TW               3            14
CP               2             5
EL               1             4
AS               1             2

我已经完成了分组工作，还有售出的门票数量。这只是我试图获得的订单的计数。

以下是我如何获得代理商售出的门票:

 Map<String, Integer> salesByAgency
     = ticketOrders.stream()
         .collect(Collectors.groupingBy(TicketSale::getAgencyCode,
                 Collectors.summingInt(TicketSale::getTicketsSold)));

TicketSale 是持有单个门票订单的类。我的集合 ticketOrders 是一个包含一堆 TicketSale 记录的 LinkedHashSet。

如何调整我必须获得的每个代理代码的订单数量？

Answer 1

你可以使用

Map<String, Integer> orders = ticketOrders
  .stream()
  .collect(Collectors.groupingBy(TicketSale::getAgencyCode,
                                 Collectors.summingInt(x -> 1)));

或

Map<String, Long> orders = ticketOrders
  .stream()
  .collect(Collectors.groupingBy(TicketSale::getAgencyCode,
                                 Collectors.counting()));

按代理商获取订单数量。

如果您想同时对计数和订单进行分组，您必须定义自己的收集器，例如

Map<String, int[]> grouped = ticketOrders
  .stream()
  .collect(Collectors.groupingBy(TicketSale::getAgencyCode,
                                 Collector.of(
                                     () -> new int[2],
                                     (a, t) -> { a[0] += 1; a[1] += t.getTicketsSold(); },
                                     (a, b) -> { a[0] += b[0]; a[1] += b[1]; return a; })));

但是，这种 lambdaism 可能很难理解。 :-)

[编辑]在这种情况下，收集器由三部分组成。第一行是创建新结果容器的供应商，在本例中是一个包含两个元素的数组:一个用于计数，一个用于总和。第二行是累加器；它将数据添加到现有的结果容器中。第三行是用于将两个结果容器合并为一个的组合器。

有关详细说明，您可以一如既往地查阅 Java API documentation .