java - 流收集操作可以是特定的 POJO 而不是 Map 吗？

Question

我想对多个字段进行分组并生成 POJO 输出。

原始的 POJO 是这样的:

@Data
@AllArgsConstructor
@NoArgsConstructor
@EqualsAndHashCode(onlyExplicitlyIncluded = true)
public class DealDTO {
    @EqualsAndHashCode.Include
    private String id;

    private Float amount;
    private Date closeDate;
    private String stage;

    public int getYear() {
        return getLocalDate().getYear();
    }

    public int getMonth() {
        return getLocalDate().getMonthValue();
    }

    private LocalDate getLocalDate() {
        return getCloseDate().toInstant()
            .atZone(ZoneId.systemDefault()).toLocalDate();
    }
}

目标 POJO

@Data
public class GroupByYearMonthStageDTO {
    private int year;
    private int month;
    private String stage;
    private DoubleSummaryStatistics statistics;
}

下面的代码产生预期的分组:

List<DealDTO> list;
var grouping = list.stream().collect(
  groupingBy(DealDTO::getYear,
    groupingBy(DealDTO::getMonth,
      groupingBy(DealDTO::getStage, summarizingDouble(DealDTO::getAmount)))));

但是......分组的类型是:

Map<Integer, Map<Integer, Map<String, DoubleSummaryStatistics>>> 

我想看看收集操作中是否有办法将输出更改为最终分组类型:

List<GroupByYearMonthStageDTO>

Answer 1

作为 groupingBy 的第一个参数，发送创建目标 POJO 的 lambda，并使用 summarizingDouble 收集器作为第二个参数。之后，您必须遍历结果 map 并设置统计信息

Map<GroupByYearMonthStageDTO, DoubleSummaryStatistics> collect = list.stream()
            .collect(groupingBy(d -> new GroupByYearMonthStageDTO(d.getYear(), d.getMonth(), d.getStage()),
                            summarizingDouble(DealDTO::getAmount)));

collect.forEach(GroupByYearMonthStageDTO::setStatistics);
Set<GroupByYearMonthStageDTO> groupByYearMonthStageDTOS = collect.keySet();