Avoid writing NaN and Infinity with json format table

trinodb · Dec 22, 2024 · 360d7a0 · 360d7a0
1 parent a401e37
commit 360d7a0
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/lib/trino-hive-formats/src/main/java/io/trino/hive/formats/line/json/JsonSerializer.java b/lib/trino-hive-formats/src/main/java/io/trino/hive/formats/line/json/JsonSerializer.java
@@ -123,7 +123,16 @@ else if (REAL.equals(type)) {
             return (generator, block, position) -> generator.writeNumber(REAL.getFloat(block, position));
         }
         else if (DOUBLE.equals(type)) {
-            return (generator, block, position) -> generator.writeNumber(DOUBLE.getDouble(block, position));
+            return (generator, block, position) -> {
+                Double value = DOUBLE.getDouble(block, position);
+                System.out.println(value);
+                if (!Double.isNaN(value) && !Double.isInfinite(value)) {
+                    generator.writeNumber(value);
+                }
+                else {
+                    throw new TrinoException(StandardErrorCode.INVALID_JSON_LITERAL, "Invalid value to Insert " + value);
+                }
+            };
         }
         else if (DATE.equals(type)) {
             return (generator, block, position) -> generator.writeString(HiveFormatUtils.formatHiveDate(block, position));