将LocalDates映射的内容分组为日期范围映射

Question

我有一张按天索引的工作班次图map

"2020-03-26": [
    {
      "id": 4,
      "startTime": "21:00:00"
    },
    {
      "id": 5,
      "startTime": "09:00:00",
    }
  ],
  "2020-03-27": [
    {
      "id": 4,
      "startTime": "22:00:00"
    },
    {
      "id": 5,
      "startTime": "10:00:00",
    }
  ],
  ...
]

可以看到，班次有一个id和一个开始时间。

但将地图按天索引是一种浪费：变化碰巧是一致的，除了一件事。开始时间是LocalTime，因此班次的唯一区别是，在应用时区后，当DST翻转发生时，产生的班次仅在小时内不同。

我想在收集器中执行某种groupBy，以获得类似的功能：

Range.of(2020-01-01 ... 2020-03-26) [
    {
      "id": 4,
      "startTime": "21:00:00"
    },
    {
      "id": 5,
      "startTime": "09:00:00",
    }
],
Range.of(2020-03-27 ... 2020-10-30) [
    {
      "id": 4,
      "startTime": "22:00:00"
    },
    {
      "id": 5,
      "startTime": "10:00:00",
    }
],
Range.of(2020-10-31 ... 2021-01-01) [
    {
      "id": 4,
      "startTime": "21:00:00"
    },
    {
      "id": 5,
      "startTime": "09:00:00",
    }
]

...分组在从DST翻转到DST翻转的范围内。我很难写一个收集器。groupBy（…）并将密钥从单个LocalDateday转换为范围




Answer 1

首先，您需要定义一个具有两个字段（开始日期和结束日期）的类Range，并创建一个范围列表。由于只需要三个实例，因此将此列表声明为实用程序类中的公共静态最终字段是有意义的。

下面是范围的一个例子，为了简洁起见，我将其实现为Java 16 record。给出了一种实用方法，用于检查给定日期是否在此范围内：

public record Range(LocalDate start, LocalDate end) {
    public boolean isWithinRange(LocalDate date) { // start inclusive, end exclusive
        return date.isBefore(start) || date.isBefore(end) && date.isAfter(start);
    }
}

范围列表：

public static final List<Range> RANGES =
    List.of(new Range(LocalDate.of(2020,1, 1), LocalDate.of(2020,3, 27)),
        new Range(LocalDate.of(2020,3, 27), LocalDate.of(2020,10, 31)),
        new Range(LocalDate.of(2020,10, 31), LocalDate.of(2021,1, 1)));

实用方法getRange（）负责根据给定日期从范围列表中检索Range的实例：

public static Range getRange(LocalDate data) {
    return RANGES.stream()
        .filter(range -> range.isWithinRange(data))
        .findFirst()
        .orElseThrow();
}

虚拟记录Shift（用于测试目的）：

public record Shift(int id, LocalTime startTime) {}

为了接近地图的转换

main（）-demo

public static void main(String[] args) {
    Map<LocalDate, Collection<Shift>> shiftsByDate =
        Map.of(LocalDate.of(2020,3, 26),
                   List.of(new Shift(4, LocalTime.of(21, 0, 0)), new Shift(5, LocalTime.of(9, 0, 0))),
               LocalDate.of(2020,3, 27),
                   List.of(new Shift(4, LocalTime.of(22, 0, 0)), new Shift(5, LocalTime.of(10, 0, 0)))
            );
    
    Map<Range, List<Shift>> shiftsByRange = shiftsByDate.entrySet().stream()
        .collect(Collectors.groupingBy(entry -> getRange(entry.getKey()),
            Collectors.flatMapping(entry -> entry.getValue().stream(),
                Collectors.toList())));
    
    shiftsByRange.forEach((k, v) -> System.out.println(k + " : " + v));
}

输出

Range[start=2020-10-31, end=2021-01-01] : [Shift[id=4, startTime=22:00], Shift[id=5, startTime=10:00]]
Range[start=2020-01-01, end=2020-03-27] : [Shift[id=4, startTime=21:00], Shift[id=5, startTime=09:00]]