Fix reading Iceberg $manifests table when contains_nan is NULL

trinodb · Mar 25, 2022 · cf92324 · cf92324
1 parent 15a422d
commit cf92324
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/plugin/trino-iceberg/src/main/java/io/trino/plugin/iceberg/ManifestsTable.java b/plugin/trino-iceberg/src/main/java/io/trino/plugin/iceberg/ManifestsTable.java
@@ -137,7 +137,13 @@ private static void writePartitionSummaries(BlockBuilder arrayBlockBuilder, List
 
             BlockBuilder rowBuilder = singleArrayWriter.beginBlockEntry();
             BOOLEAN.writeBoolean(rowBuilder, summary.containsNull());
-            BOOLEAN.writeBoolean(rowBuilder, summary.containsNaN());
+            Boolean containsNan = summary.containsNaN();
+            if (containsNan == null) {
+                rowBuilder.appendNull();
+            }
+            else {
+                BOOLEAN.writeBoolean(rowBuilder, containsNan);
+            }
             VARCHAR.writeString(rowBuilder, field.transform().toHumanString(
                     Conversions.fromByteBuffer(nestedType, summary.lowerBound())));
             VARCHAR.writeString(rowBuilder, field.transform().toHumanString(