Java跨年度日期场景下动态调用getUserForMonth方法的实现

我之前也碰到过一模一样的场景，用YearMonth类来处理年月遍历简直是精准适配！它能自动搞定跨年度的月份递推，不用你手动去算“12月之后要跳转到下一年1月”这种麻烦逻辑。下面是具体的实现步骤：

1. 把起始/结束日期转成YearMonth #

先把你已经拿到的startDate和endDate转换成YearMonth类型，这样我们就能聚焦在年月维度处理：

YearMonth startYm = YearMonth.from(startDate);
YearMonth endYm = YearMonth.from(endDate);

2. 遍历所有目标年月并调用方法 #

接下来用循环或者Stream API，遍历从startYm到endYm的每一个年月，逐个调用getUserForMonth，最后把所有结果合并成一个总集合：

方式一：普通循环（直观易懂，适合新手）

Collection<User> allUsers = new ArrayList<>();
YearMonth currentYm = startYm;

// 循环到当前年月超过结束年月为止
while (!currentYm.isAfter(endYm)) {
    // 调用目标方法，传入当前年月的年和月
    Collection<User> monthlyUsers = getUserForMonth(id, currentYm.getYear(), currentYm.getMonthValue());
    // 加个空判断，避免空指针异常
    Optional.ofNullable(monthlyUsers).ifPresent(allUsers::addAll);
    
    // 月份自动递增，跨年逻辑由YearMonth处理
    currentYm = currentYm.plusMonths(1);
}

方式二：Stream API（简洁优雅，适合函数式编程爱好者）

如果习惯用Stream写法，也可以这样实现：

Collection<User> allUsers = Stream.iterate(startYm, ym -> ym.plusMonths(1))
        // 计算需要遍历的月份总数，确保不超出结束年月
        .limit(startYm.until(endYm, ChronoUnit.MONTHS) + 1)
        // 调用方法并将每个结果集合转为Stream
        .flatMap(ym -> getUserForMonth(id, ym.getYear(), ym.getMonthValue()).stream())
        // 收集到最终集合中
        .collect(Collectors.toList());

为什么推荐用YearMonth？ #

• 自动处理跨年：比如当前年月是2023-12时，调用plusMonths(1)会直接得到2024-01，完全不用你手动判断月份是否为12、年份要不要加1。
• 边界场景全覆盖：不管起始和结束日期在同一个月、跨几个月还是跨年度，这个逻辑都能正确生成所有需要的年月参数。

内容的提问来源于stack exchange，提问作者Tom