Java8实现分组求和
由于的美国的制裁,加速了国内很多公司产品构成去美化进程,而在软件行业的数据库层面,越来越多的公司进行去Oracle化。本人所在公司有自研内存数据库,聚合函数的支持初期没那么完善,有时一部分逻辑要从数据库转化到java实现,今天举一个java实现分组求和来代替数据库聚合函数sum()+group by的栗子
分组求和
举一个生产过程的引用例子,代码如下:
/**
* @Description //TODO 根据客户列表查欠费信息
* @Date: 2021/7/15
**/
public List<UnPayBillDTO> getUnPayInfo(List<Long> contractNoList,String yearMonth){
List<UnPayBillDTO> unPayBillDTOList = new ArrayList<>(); // 结果集
for (long contractNo : contractNoList) {
ActUnpayoweInfo actUnpayoweInfoParam = new ActUnpayoweInfo();
actUnpayoweInfoParam.setContractNo(contractNo);
List<ActUnpayoweInfo> actUnpayoweInfos = actUnpayoweInfoMapper.select(actUnpayoweInfoParam); // 该账户所有欠费信息
/* 分组 */
Map<String,List<ActUnpayoweInfo>> afterGroupMap
= actUnpayoweInfos.stream().
filter(o -> o.getNaturalMonth().equals(yearMonth)). // 过滤掉当月欠费
collect(Collectors.groupingBy(o -> o.getContractNo() + "_" + o.getNaturalMonth() + "_" + o.getAcctItemCode()));
afterGroupMap.forEach(
(group, groupInfo) -> {
/* 获取分组字段 */
String[] tmpInfo = group.split("_");
long tmpContractNo = Long.parseLong(tmpInfo[0]);
String tmpNaturalMonth = tmpInfo[1];
String tmpAcctItemCode = tmpInfo[2];
/* 求和 */
long totalMoney = groupInfo.stream().mapToLong(this::calculate).sum();
UnPayBillDTO unPayBillDTO = new UnPayBillDTO();
unPayBillDTO.setContractNo(tmpContractNo);
unPayBillDTO.setNaturalMonht(tmpNaturalMonth);
unPayBillDTO.setAcctItemCode(tmpAcctItemCode);
unPayBillDTO.setTotalMoney(totalMoney);
unPayBillDTOList.add(unPayBillDTO);
}
);
}
return unPayBillDTOList;
}
分组
分组是先利用stream进行遍历,然后以group by涉及的字段进行拼接,作为Collector进行分组的条件,最终返回一个以分组条件拼接的字符串作为key值,分好组的各个对应list作为value的map;
Map<String,List<ActUnpayoweInfo>> afterGroupMap
= actUnpayoweInfos.stream().
filter(o -> o.getNaturalMonth().equals(yearMonth)). // 过滤掉当月欠费
collect(Collectors.groupingBy(o -> o.getContractNo() + "_" + o.getNaturalMonth() + "_" + o.getAcctItemCode()));
求和
对每一组进行处理,从key值获取分组条件信息,对value值里面的多条数据相关字段进行计算后累加求和
/* 获取分组字段 */
String[] tmpInfo = group.split("_");
long tmpContractNo = Long.parseLong(tmpInfo[0]);
String tmpNaturalMonth = tmpInfo[1];
String tmpAcctItemCode = tmpInfo[2];
/* 求和 */
long totalMoney = groupInfo.stream().mapToLong(this::calculate).sum();
this::calculate
是涉及业务的一些计算,这里不做展开
public long calculate(ActUnpayoweInfo actUnpayoweInfo){
return actUnpayoweInfo.getShouldPay() - actUnpayoweInfo.getFavourFee() - actUnpayoweInfo.getPayedLater() - actUnpayoweInfo.getPayedPrepay();
}