场景描述:根据实体类相同名称(name)属性,对金额(money)属性进行分组求和
实体类:
@Data
public class DepartmentEntity{
private String name;
private BigDecimal money;
}
代码实现(但是这种不够优雅):
List<DepartmentEntity> ds = new List<DepartmentEntity>();
//假设集合中已有数据………………
//1、先对集合根据名称进行分组,此时 List<DepartmentEntity> 已转为以名称(name)属性为key的 Map<String,List<DepartmentEntity>>
Map<String,BigDecimal> moneySumGroupByName = ds.stream().collect(Collectors.groupingBy(DepartmentEntity::getName))
//2、此时利用 Map 的 entrySet() 方法得到 ,得到 Map.Entry 的 Set 集合
.entrySet()
//3、对 Set 集合进行 toMap,此时新 Map 的 key 继续使用 Map.Entry 的 key ,也就是名称;
//在处理 value 时,因为此时value是已经根据名称分组后的集合,所以这里直接进行集合求和即可
.stream().collect(Collectors.toMap(Map.Entry::getKey,m->m.getValue()
.stream()
//3、过滤非空
.filter(d->Objects.nonNull(d.getMoney()))
//4、stream 对 BigDecimal 的求和
.map(DepartmentEntity::getMoney).reduce(BigDecimal.ZERO,BigDecimal::add)));
优雅篇:
public static void main(String[] args) {
List<YearNum> years = Lists.newArrayList();
Integer startYear = 2021;
for (int i = 0; i < 15; i++) {
startYear = startYear+i;
if(i % 2 == 0){
years.add(new YearNum(2021,BigDecimal.valueOf(i)));
}else{
years.add(new YearNum(2022,BigDecimal.valueOf(i)));
}
}
Map<Integer, BigDecimal> numByYear = years.stream()
.collect(Collectors.groupingBy(YearNum::getYear,
Collectors.reducing(BigDecimal.ZERO, YearNum::getNum, BigDecimal::add)));
System.out.println(numByYear);
}
@Data
@AllArgsConstructor
@NoArgsConstructor
static
class YearNum{
private Integer year;
private BigDecimal num;
}