Remove extra copy when decompressing data page with encryption on (ap…

…ache#17)
parthchandra · May 11, 2022 · 9fbda8d · 9fbda8d
1 parent 0a2c831
commit 9fbda8d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/parquet-hadoop/src/main/java/org/apache/parquet/hadoop/ColumnChunkPageReadStore.java b/parquet-hadoop/src/main/java/org/apache/parquet/hadoop/ColumnChunkPageReadStore.java
@@ -167,10 +167,12 @@ public DataPage readPage() {
         public DataPage visit(DataPageV1 dataPageV1) {
           try {
             BytesInput bytes = dataPageV1.getBytes();
-            ByteBuffer byteBuffer = bytes.toByteBuffer();
             long compressedSize = bytes.size();
 
-            if (null != blockDecryptor) {
+            ByteBuffer byteBuffer;
+            if (blockDecryptor == null) {
+              byteBuffer = bytes.toByteBuffer();
+            } else {
               byte[] decrypted = blockDecryptor.decrypt(bytes.toByteArray(), dataPageAAD);
               compressedSize = decrypted.length;
               byteBuffer = ByteBuffer.allocateDirect((int) compressedSize);