Use toIntExact in PrimitiveColumnWriter to cast long to int safely

trinodb · Jul 19, 2022 · 35f6831 · 35f6831
1 parent 822f43e
commit 35f6831
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/trino-parquet/src/main/java/io/trino/parquet/writer/PrimitiveColumnWriter.java b/lib/trino-parquet/src/main/java/io/trino/parquet/writer/PrimitiveColumnWriter.java
@@ -230,8 +230,8 @@ private void flushCurrentPageToBuffer()
         columnStatistics.mergeStatistics(statistics);
 
         ByteArrayOutputStream pageHeaderOutputStream = new ByteArrayOutputStream();
-        parquetMetadataConverter.writeDataPageV1Header((int) uncompressedSize,
-                (int) compressedSize,
+        parquetMetadataConverter.writeDataPageV1Header(toIntExact(uncompressedSize),
+                toIntExact(compressedSize),
                 valueCount,
                 repetitionLevelWriter.getEncoding(),
                 definitionLevelWriter.getEncoding(),
@@ -284,7 +284,8 @@ private List<ParquetDataOutput> getDataStreams()
             long compressedSize = pageData.size();
 
             ByteArrayOutputStream dictStream = new ByteArrayOutputStream();
-            parquetMetadataConverter.writeDictionaryPageHeader(toIntExact(uncompressedSize),
+            parquetMetadataConverter.writeDictionaryPageHeader(
+                    toIntExact(uncompressedSize),
                     toIntExact(compressedSize),
                     dictionaryPage.getDictionarySize(),
                     dictionaryPage.getEncoding(),