Add getReservation to Accountant

dremio · Jan 15, 2018 · 4a661ed · 4a661ed
1 parent 707af38
commit 4a661ed
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/java/memory/src/main/java/org/apache/arrow/memory/Accountant.java b/java/memory/src/main/java/org/apache/arrow/memory/Accountant.java
@@ -219,6 +219,15 @@ public long getLimit() {
     return allocationLimit.get();
   }
 
+  /**
+   * Return the initial reservation.
+   *
+   * @return reservation in bytes.
+   */
+  public long getInitReservation() {
+    return reservation;
+  }
+
   /**
    * Set the maximum amount of memory that can be allocated in the this Accountant before failing
    * an allocation.

diff --git a/java/memory/src/main/java/org/apache/arrow/memory/BufferAllocator.java b/java/memory/src/main/java/org/apache/arrow/memory/BufferAllocator.java
@@ -91,6 +91,13 @@ public interface BufferAllocator extends AutoCloseable {
    */
   public long getLimit();
 
+  /**
+   * Return the initial reservation.
+   *
+   * @return reservation in bytes.
+   */
+  public long getInitReservation();
+
   /**
    * Set the maximum amount of memory this allocator is allowed to allocate.
    *