Explicitly call Deflater/Inflater#close to free native resources (#203)…

… (#204)
awslabs · Aug 1, 2022 · 529b240 · 529b240
1 parent 8b3a416
commit 529b240
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/...va/com/amazonaws/services/schemaregistry/common/GlueSchemaRegistryCompressionHandler.java b/...va/com/amazonaws/services/schemaregistry/common/GlueSchemaRegistryCompressionHandler.java
@@ -81,7 +81,10 @@ static byte[] writeToByteArrayOutputStream(Inflater inflater, ByteArrayOutputStr
      * @throws IOException
      */
     default byte[] compress(byte[] record) throws IOException {
-        return writeToDeflatorObject(record, getDeflatorObject(record));
+        Deflater deflator = getDeflatorObject(record);
+        byte[] compressed = writeToDeflatorObject(record, deflator);
+        deflator.end();
+        return compressed;
     }
 
     /**
@@ -95,6 +98,9 @@ default byte[] compress(byte[] record) throws IOException {
      * @throws IOException
      */
     default byte[] decompress(byte[] compressedRecord, int start, int end) throws IOException {
-        return decompress(getInflatorObject(compressedRecord, start, end), compressedRecord.length);
+        Inflater inflator = getInflatorObject(compressedRecord, start, end);
+        byte[] decompressed = decompress(inflator, compressedRecord.length);
+        inflator.end();
+        return decompressed;
     }
 }