java - Guava 多重映射到响应模型的映射结果

Question

在我的 Web 服务应用程序中，以下模型作为统计信息的 UI 合约，其中对于给定日期，我需要提供有多少个 accepted 的摘要。 , rejected等等:

@Data
public class StatusResponse {
    private LocalDate processedDate;
    private int accepted;
    private int rejected;
    private int failed;
    private int other;
}

我的数据库查询返回的摘要结果如下:

date        status              groupStatus     count

2020-01-01  ACC                 Accepted        3
2020-01-22  RJCT                Rejected        47
2020-01-22  NM                  Other           1
2020-01-23  NIY                 Failed          55

我有一个枚举 Status :

public enum Status{

    ACCEPTED("Accepted"),
    REJECTED("Rejected"),
    OTHER("Other"),
    FAILED("Failed");

    private final String text;

    Status (final String text) {
        this.text = text;
    }
}

在我的存储库类中，我查询并将状态摘要检索到 Guava MultiMap 中。对象，其中键是日期，值是 EnumMap状态及其计数:

public Multimap<LocalDate, EnumMap<Status, Integer>> statusSummary(final Request search) {
    return this.namedParameterJdbcTemplate.query(this.QUERY,
            new MapSqlParameterSource()
                    .addValue("processedDate", search.getProcessedDate()),
            rs -> {
                final Multimap<LocalDate, EnumMap<Status, Integer>> statusMap = ArrayListMultimap.create();
                while (rs.next()) {
                    final EnumMap<Status, Integer> groupedStatusCount = new EnumMap<>(Status.class);
                    groupedStatusCount.put(fromValue(rs.getString("groupStatus")), rs.getInt("count"));
                    statusMap.put(rs.getDate("date").toLocalDate(), groupedStatusCount);
                }
                return statusMap;
            }
    );
}

因此键(processedDate)可以具有与值相同的状态。例如， map 可以包含:

key --> 2020-01-23
   val1 --> OTHER 23
   val2 --> ACCEPTED 2
   val3 --> OTHER 4

我的问题以及我正在努力完成的事情:

如何映射 statusMap 的结果至StatusResponse并提供每个日期的摘要。所以StatusResponse上面的内容是这样的:

{
    "processedDate": [
        2020,
        1,
        23
    ],
    "accepted": 2,
    "rejected": 0,
    "failed": 0,
    "other": 27,
    "date": "2020-01-23T00:00:00"
}

我正在使用 Java 8，并且需要一些帮助，如何通过流操作并使用 groupBy 来实现此目的功能性风格？

此外，我需要进一步分组，将今天之前的所有内容分组到今天，将今天之后的所有内容分组到今天 + 1。

谢谢

Answer 1

您可以尝试以下方法。摘要可以存储在 Map 中，并使用 Jackson 的 JsonAnyGetter 注释在根级别展开。这将从响应中删除汇总字段并将内部值置于根级别。

@Data
@Builder
public class StatusResponse {
    private LocalDate processedDate;

    @JsonAnyGetter
    private Map<Status, Integer> summary;
}

现在，让我们将 MultiMap 转换为所需的格式

List<StatusResponse> responseList = data.entries().stream()
  .map(entry -> {
    Map<Status, Integer> summary = entry.getValue().entries().stream()
      .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue, (a, b) -> b));

    return StatusResponse.builder()
      .processedDate(entry.getKey())
      .summary(summary)
      .build();
  }).collect(Collectors.toList());

现在，我们需要将 LocaleDate 转换为正确的格式。我们可以使用jackson-datatype-jsr310。要转换为“yyyy-mm-dd”这样的格式，我们只需将 JavaTimeModule 注册到 ObjectMapper 即可。

ObjectMapper mapper = new ObjectMapper();
mapper.registerModule(new JavaTimeModule());

输出会是这样的。

[{
   "processedDate":"2020-01-23",
    "FAILED":3,
    "ACCEPTED":2
}]

注意:如果数据库返回所有状态，即使值为零，它也会包含在响应中。因此，要让所有状态字段都得到响应，只需使它们全部成为多重映射的一部分即可。

您可以进一步探索自定义序列化器来微调日期格式。 here