diff --git a/src/main/java/com/codingwasabi/howtodo/web/policy/planMaking/MonthPlanMakingPolicy.java b/src/main/java/com/codingwasabi/howtodo/web/policy/planMaking/MonthPlanMakingPolicy.java index 6d2c42b..7a5459d 100644 --- a/src/main/java/com/codingwasabi/howtodo/web/policy/planMaking/MonthPlanMakingPolicy.java +++ b/src/main/java/com/codingwasabi/howtodo/web/policy/planMaking/MonthPlanMakingPolicy.java @@ -3,6 +3,7 @@ import static com.codingwasabi.howtodo.web.policy.util.ExamDateSorting.*; import java.time.LocalDate; +import java.time.temporal.ChronoUnit; import java.util.ArrayList; import java.util.HashMap; import java.util.List; @@ -143,13 +144,17 @@ private Map initDailyHours(List exams, LocalDate today LocalDate endDate = exams.get(exams.size() - 1) .getDueDateTime() .toLocalDate(); + int days = getDays(today, endDate); - for (int day = 1; day < today.until(endDate) - .getDays(); day++) { + for (int day = 1; day < days; day++) { nextDay = nextDay.plusDays(1); dailyHours.put(nextDay, 0); } return dailyHours; } + + private int getDays(LocalDate start, LocalDate end) { + return (int)ChronoUnit.DAYS.between(start, end); + } }