[ARROW-5881][Java] Provide benchmark for allBitsNull

apache · Jul 11, 2019 · c57cb6d · c57cb6d
1 parent 3deedaf
commit c57cb6d
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/java/performance/src/test/java/org/apache/arrow/vector/BitVectorHelperBenchmarks.java b/java/performance/src/test/java/org/apache/arrow/vector/BitVectorHelperBenchmarks.java
@@ -50,6 +50,8 @@ public class BitVectorHelperBenchmarks {
 
   private ArrowBuf validityBuffer;
 
+  private ArrowBuf oneBitValidityBuffer;
+
   /**
    * Setup benchmarks.
    */
@@ -65,6 +67,11 @@ public void prepare() {
         BitVectorHelper.setValidityBit(validityBuffer, i, (byte) 0);
       }
     }
+
+    // only one 1 bit in the middle of the buffer
+    oneBitValidityBuffer = allocator.buffer(VALIDITY_BUFFER_CAPACITY / 8);
+    oneBitValidityBuffer.setZero(0, VALIDITY_BUFFER_CAPACITY / 8);
+    BitVectorHelper.setValidityBit(oneBitValidityBuffer, VALIDITY_BUFFER_CAPACITY / 2, (byte) 1);
   }
 
   /**
@@ -73,6 +80,7 @@ public void prepare() {
   @TearDown
   public void tearDown() {
     validityBuffer.close();
+    oneBitValidityBuffer.close();
     allocator.close();
   }
 
@@ -83,6 +91,13 @@ public int getNullCountBenchmark() {
     return BitVectorHelper.getNullCount(validityBuffer, VALIDITY_BUFFER_CAPACITY);
   }
 
+  @Benchmark
+  @BenchmarkMode(Mode.AverageTime)
+  @OutputTimeUnit(TimeUnit.NANOSECONDS)
+  public boolean allBitsNullBenchmark() {
+    return BitVectorHelper.allBitsNullOrig(oneBitValidityBuffer, VALIDITY_BUFFER_CAPACITY);
+  }
+
   //@Test
   public static void main(String [] args) throws RunnerException {
     Options opt = new OptionsBuilder()