Java Stream 实现根据多字段分组求和

代码块如下：

 List<AgentCheckoutStockDetail> agentCheckoutStockDetails = agentCheckoutStockDetailRepository.selectByExample(ws);
 
Map<String, Long> countMap = agentCheckoutStockDetails.stream()
        .collect(Collectors.groupingBy(o -> o.getDeliveryAgentId() + "_" + o.getProductSku() + "_" + o.getCheckoutTime(),Collectors.summingLong(AgentCheckoutStockDetail::getActualCount)));
        
List<AgentCheckoutStockDetail> countRecords = countMap.keySet().stream().map(key -> {
    String[] arr = key.split("_");
    String deliveryAgentId1 = arr[0];
    String productSku = arr[1];
    String checkOutTime = arr[2];
    AgentCheckoutStockDetail obj = new AgentCheckoutStockDetail();
    obj.setDeliveryAgentId(deliveryAgentId1);
    obj.setProductSku(productSku);
    obj.setCheckoutTime(checkOutTime);
    obj.setActualCount(countMap.get(key).intValue());
    return obj;
}).collect(Collectors.toList());

1、分组核心思想

1.1、Collectors.groupingBy

如下代码：
Collectors.groupingBy(o -> o.getDeliveryAgentId() + "_" + o.getProductSku() + "_" + o.getCheckoutTime(), Collectors.summingLong(AgentCheckoutStockDetail::getActualCount))

首先明确的是 Collectors.groupingBy 是只能支持一个进行分组的参数

这里需要根据字符串的拼接来完成出多字段的分组处理

通过lambda表达式：o -> o.getDeliveryAgentId() + “-” + o.getProductSku() + “-” + o.getCheckoutTime()
这里可以看到原本只能接收一个参数的 groupby参数位置变成了多个字符串拼接

1.2、Collectors.summingLong（当然这里Long值可变）

Collectors.summingLong(AgentCheckoutStockDetail::getActualCount)

根据需要的字段进行求和，前提是根据前面的分组（如上图）

2、分组遍历（真正从分组中拿到数据）

 List<AgentCheckoutStockDetail> countRecords = countMap.keySet().stream().map(key -> {
    String[] arr = key.split("_");
    String deliveryAgentId1 = arr[0];
    String productSku = arr[1];
    String checkOutTime = arr[2];
    AgentCheckoutStockDetail obj = new AgentCheckoutStockDetail();
    obj.setDeliveryAgentId(deliveryAgentId1);
    obj.setProductSku(productSku);
    obj.setCheckoutTime(checkOutTime);
    obj.setActualCount(countMap.get(key).intValue());
    return obj;
}).collect(Collectors.toList());

2.1、拆分关键 key（group by）的那个字符串

    String[] arr = key.split("_");
    String deliveryAgentId1 = arr[0];
    String productSku = arr[1];
    String checkOutTime = arr[2];

2.2、重新创建对象，放入map

AgentCheckoutStockDetail obj = new AgentCheckoutStockDetail();
    obj.setDeliveryAgentId(deliveryAgentId1);
    obj.setProductSku(productSku);
    obj.setCheckoutTime(checkOutTime);
    obj.setActualCount(countMap.get(key).intValue());
    return obj;

这里注意 countMap.get(key).intValue() 根据key可以获取到对应的分组时的求和的值

后面如果需要再次转对象，自行转化即可

最后

以上就是高大小蝴蝶为你收集整理的Java Stream 实现根据多字段分组求和的全部内容，希望文章能够帮你解决Java Stream 实现根据多字段分组求和所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。