Adjust ParquetFileWriter.getWritteBytes to match OrcFileWriter behaviour

trinodb · Feb 10, 2022 · 47a3850 · 47a3850
1 parent 3788633
commit 47a3850
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/lib/trino-parquet/src/main/java/io/trino/parquet/writer/ParquetWriter.java b/lib/trino-parquet/src/main/java/io/trino/parquet/writer/ParquetWriter.java
@@ -179,6 +179,7 @@ public void close()
             flush();
             writeFooter();
         }
+        bufferedBytes = 0;
     }
 
     // Parquet File Layout:

diff --git a/plugin/trino-hive/src/main/java/io/trino/plugin/hive/parquet/ParquetFileWriter.java b/plugin/trino-hive/src/main/java/io/trino/plugin/hive/parquet/ParquetFileWriter.java
@@ -86,7 +86,7 @@ public ParquetFileWriter(
     @Override
     public long getWrittenBytes()
     {
-        return parquetWriter.getWrittenBytes();
+        return parquetWriter.getWrittenBytes() + parquetWriter.getBufferedBytes();
     }
 
     @Override