feat: minor change

vibhatha · Feb 1, 2024 · 184d827 · 184d827
1 parent d09533c
commit 184d827
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/java/vector/src/main/java/org/apache/arrow/vector/ipc/ArrowWriter.java b/java/vector/src/main/java/org/apache/arrow/vector/ipc/ArrowWriter.java
@@ -101,7 +101,9 @@ protected ArrowWriter(VectorSchemaRoot root, DictionaryProvider provider, Writab
     this.compressionFactory = compressionFactory;
     this.codecType = codecType;
     this.compressionLevel = compressionLevel;
-    this.codec = getCodec();
+    this.codec = this.compressionLevel.isPresent() ?
+            this.compressionFactory.createCodec(this.codecType, this.compressionLevel.get()) :
+            this.compressionFactory.createCodec(this.codecType);
     this.unloader = new VectorUnloader(root, /*includeNullCount*/ true, codec,
         /*alignBuffers*/ true);
 
@@ -181,12 +183,6 @@ public long bytesWritten() {
     return out.getCurrentPosition();
   }
 
-  private CompressionCodec getCodec() {
-    return this.compressionLevel.isPresent() ?
-        this.compressionFactory.createCodec(this.codecType, this.compressionLevel.get()) :
-        this.compressionFactory.createCodec(this.codecType);
-  }
-
   private void ensureStarted() throws IOException {
     if (!started) {
       started = true;