Add roundOff option to TemporalAmountParser.

Author: vLuckyyy

eternalcode-commons-shared/src/main/java/com/eternalcode/commons/shared/time/TemporalAmountParser.java

@@ -9,9 +9,11 @@
 import java.time.temporal.TemporalAmount;
 import java.util.ArrayList;
 import java.util.Collections;
+import java.util.HashSet;
 import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.Set;
 import java.util.function.Predicate;
 
 /**
@@ -41,6 +43,12 @@ public abstract class TemporalAmountParser<T extends TemporalAmount> {
     private static final Map<ChronoUnit, Long> UNIT_TO_NANO = new LinkedHashMap<>();
     private static final Map<ChronoUnit, Integer> PART_TIME_UNITS = new LinkedHashMap<>();
 
+    private Set<ChronoUnit> roundedUnits = new HashSet<>();
+
+    public void roundOff(ChronoUnit unit) {
+        roundedUnits.add(unit);
+    }
+
     static {
         UNIT_TO_NANO.put(ChronoUnit.NANOS, 1L);
         UNIT_TO_NANO.put(ChronoUnit.MICROS, 1_000L);
@@ -276,6 +284,11 @@ public String format(T temporalAmount) {
 
         for (String key : keys) {
             ChronoUnit chronoUnit = this.units.get(key);
+
+            if (roundedUnits.contains(chronoUnit)) {
+                continue;
+            }
+
             Long part = UNIT_TO_NANO.get(chronoUnit);
 
             if (part == null) {

eternalcode-commons-shared/test/com/eternalcode/commons/shared/time/TemporalAmountParserTest.java

@@ -192,6 +192,34 @@ void testFormat(int seconds, String expected) {
         assertEquals(expected, formatted);
     }
 
+    @Test
+    void testRoundOff() {
+        TemporalAmountParser<Duration> temporalAmountParser = new DurationParser()
+            .withUnit("s", ChronoUnit.SECONDS)
+            .withUnit("m", ChronoUnit.MINUTES)
+            .withUnit("h", ChronoUnit.HOURS)
+            .withUnit("d", ChronoUnit.DAYS);
+        temporalAmountParser.roundOff(ChronoUnit.MILLIS);
+
+        Duration temporalAmount = Duration.ofDays(1)
+            .plus(Duration.ofHours(1))
+            .plus(Duration.ofMinutes(1))
+            .plus(Duration.ofSeconds(1))
+            .plus(Duration.ofMillis(500));
+
+        String formatted = temporalAmountParser.format(temporalAmount);
+        assertEquals("1d1h1m1s", formatted);
+
+        temporalAmount = Duration.ofDays(1)
+            .plus(Duration.ofHours(1))
+            .plus(Duration.ofMinutes(1))
+            .plus(Duration.ofSeconds(1))
+            .plus(Duration.ofMillis(499));
+
+        formatted = temporalAmountParser.format(temporalAmount);
+        assertEquals("1d1h1m1s", formatted);
+    }
+
     @Test
     void testNotSupportedChronoUnit() {
         TemporalAmountParser<Period> temporalAmountParser = new PeriodParser()