backport apache#5126 to close compressor and avoid memory leak (found…

… in Flink iceberg connector) (apache#40)
zzhhhzz · Apr 17, 2023 · 9931da1 · 9931da1
1 parent 632c0a9
commit 9931da1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/parquet/src/main/java/org/apache/iceberg/parquet/ParquetWriter.java b/parquet/src/main/java/org/apache/iceberg/parquet/ParquetWriter.java
@@ -210,7 +210,12 @@ public void close() throws IOException {
       this.closed = true;
       flushRowGroup(true);
       writeStore.close();
-      writer.end(metadata);
+      if (writer != null) {
+        writer.end(metadata);
+      }
+      if (compressor != null) {
+        compressor.release();
+      }
     }
   }
 }