Enable configurable write buffer block size.

amazon-ion · Nov 16, 2023 · fa11a5f · fa11a5f
1 parent 0d0258c
commit fa11a5f
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/com/amazon/ion/impl/_Private_IonBinaryWriterBuilder.java b/src/com/amazon/ion/impl/_Private_IonBinaryWriterBuilder.java
@@ -223,6 +223,13 @@ public _Private_IonBinaryWriterBuilder withLocalSymbolTableAppendDisabled()
         return b;
     }
 
+    @Override
+    public _Private_IonBinaryWriterBuilder withDefaultBlockSize(int size) {
+        _Private_IonBinaryWriterBuilder b = mutable();
+        b.setDefaultBlockSize(size);
+        return b;
+    }
+
     @Override
     public void setIsFloatBinary32Enabled(boolean enabled) {
         mutationCheck();
@@ -275,6 +282,9 @@ public void setStreamCopyOptimized(final boolean optimized)
 
     //=========================================================================
 
+    public void setDefaultBlockSize(int size) {
+        myBinaryWriterBuilder.withUserBlockSize(size);
+    }
 
     /**
      * Fills all properties and returns an immutable builder.

diff --git a/src/com/amazon/ion/system/IonBinaryWriterBuilder.java b/src/com/amazon/ion/system/IonBinaryWriterBuilder.java
@@ -194,6 +194,12 @@ public IvmMinimizing getIvmMinimizing()
      */
     public abstract IonBinaryWriterBuilder withLocalSymbolTableAppendDisabled();
 
+    /**
+     * Specify the default block size for write buffer.
+     * @param size represent the specified block size in bytes.
+     * If this functionality is not set, the default block size is 32768 bytes.
+     */
+    public abstract IonBinaryWriterBuilder withDefaultBlockSize(int size);
 
     /**
      * Enables or disables writing Binary32 (4-byte, single precision,