diff --git a/core-java-modules/core-java-date-operations/src/main/java/com/baeldung/datetime/AddSubtractDaysSkippingWeekendsUtils.java b/core-java-modules/core-java-date-operations/src/main/java/com/baeldung/datetime/AddSubtractDaysSkippingWeekendsUtils.java index c54a7823f3..bf32fdb79a 100644 --- a/core-java-modules/core-java-date-operations/src/main/java/com/baeldung/datetime/AddSubtractDaysSkippingWeekendsUtils.java +++ b/core-java-modules/core-java-date-operations/src/main/java/com/baeldung/datetime/AddSubtractDaysSkippingWeekendsUtils.java @@ -5,17 +5,27 @@ import java.time.LocalDate; public class AddSubtractDaysSkippingWeekendsUtils { - public static LocalDate addSubtractDaysSkippingWeekendsIterativeMethod(LocalDate date, int days) { + public static LocalDate addDaysSkippingWeekends(LocalDate date, int days) { LocalDate result = date; int addedDays = 0; - int absDays = Math.abs(days); - int day = (days > 0) ? 1 : -1; - while (addedDays < absDays) { - result = result.plusDays(day); + while (addedDays < days) { + result = result.plusDays(1); if (!(result.getDayOfWeek() == DayOfWeek.SATURDAY || result.getDayOfWeek() == DayOfWeek.SUNDAY)) { ++addedDays; } } return result; } + + public static LocalDate subtractDaysSkippingWeekends(LocalDate date, int days) { + LocalDate result = date; + int subtractedDays = 0; + while (subtractedDays < days) { + result = result.minusDays(1); + if (!(result.getDayOfWeek() == DayOfWeek.SATURDAY || result.getDayOfWeek() == DayOfWeek.SUNDAY)) { + ++subtractedDays; + } + } + return result; + } } diff --git a/core-java-modules/core-java-date-operations/src/test/java/com/baeldung/datetime/AddSubtractDaysSkippingWeekendsUtilsUnitTest.java b/core-java-modules/core-java-date-operations/src/test/java/com/baeldung/datetime/AddSubtractDaysSkippingWeekendsUtilsUnitTest.java index 2c8320f035..d60a0f4015 100644 --- a/core-java-modules/core-java-date-operations/src/test/java/com/baeldung/datetime/AddSubtractDaysSkippingWeekendsUtilsUnitTest.java +++ b/core-java-modules/core-java-date-operations/src/test/java/com/baeldung/datetime/AddSubtractDaysSkippingWeekendsUtilsUnitTest.java @@ -9,18 +9,18 @@ import java.time.LocalDate; public class AddSubtractDaysSkippingWeekendsUtilsUnitTest { @Test - public void givenLocalDateAndDaysToAdd_thenAddDaysSkippingWeekendsUsingIterativeMethod() { + public void givenLocalDateAndDaysToAdd_thenAddDaysSkippingWeekends() { LocalDate initialDate = LocalDate.of(2019, 11, 7); LocalDate expectedDate = LocalDate.of(2019, 11, 13); - LocalDate result = AddSubtractDaysSkippingWeekendsUtils.addSubtractDaysSkippingWeekendsIterativeMethod(initialDate, 4); + LocalDate result = AddSubtractDaysSkippingWeekendsUtils.addDaysSkippingWeekends(initialDate, 4); assertEquals(expectedDate, result); } @Test - public void givenLocalDateAndDaysToAdd_thenSubtractDaysSkippingWeekendsUsingIterativeMethod() { + public void givenLocalDateAndDaysToSubtract_thenSubtractDaysSkippingWeekends() { LocalDate initialDate = LocalDate.of(2019, 11, 7); LocalDate expectedDate = LocalDate.of(2019, 11, 1); - LocalDate result = AddSubtractDaysSkippingWeekendsUtils.addSubtractDaysSkippingWeekendsIterativeMethod(initialDate, -4); + LocalDate result = AddSubtractDaysSkippingWeekendsUtils.subtractDaysSkippingWeekends(initialDate, 4); assertEquals(expectedDate, result); }