NIFI-13579 Corrected number case in switch statement to account for a…

…ll type of numbers.
apache · Sep 6, 2024 · 27f4fa5 · 27f4fa5
1 parent 57c1962
commit 27f4fa5
Showing 1 changed file with 7 additions and 10 deletions.
diff --git a/...c/main/java/org/apache/nifi/serialization/record/field/ObjectTimestampFieldConverter.java b/...c/main/java/org/apache/nifi/serialization/record/field/ObjectTimestampFieldConverter.java
@@ -79,16 +79,13 @@ public Timestamp convertField(final Object field, final Optional<String> pattern
                 final Instant instant = Instant.ofEpochMilli(epochMilli);
                 return Timestamp.from(instant);
             }
-            case Double number -> {
-                final Instant instant = FractionalSecondsUtils.toInstant(number);
-                return Timestamp.from(instant);
-            }
-            case Float number -> {
-                final Instant instant = FractionalSecondsUtils.toInstant(number.doubleValue());
-                return Timestamp.from(instant);
-            }
-            case Long number -> {
-                final Instant instant = FractionalSecondsUtils.toInstant(number);
+            case Number number -> {
+                final Instant instant;
+                switch (field) {
+                    case Double d -> instant = FractionalSecondsUtils.toInstant(d);
+                    case Float f -> instant = FractionalSecondsUtils.toInstant(f.doubleValue());
+                    default -> instant = FractionalSecondsUtils.toInstant(number.longValue());
+                }
                 return Timestamp.from(instant);
             }
             case String string -> {