Implement primitives reads in ByteArrayIndexInput (#79885)

enhance performance of those actions
elastic · Oct 27, 2021 · 036f881 · 036f881
1 parent 8f5069d
commit 036f881
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/server/src/main/java/org/elasticsearch/common/lucene/store/ByteArrayIndexInput.java b/server/src/main/java/org/elasticsearch/common/lucene/store/ByteArrayIndexInput.java
@@ -8,6 +8,7 @@
 package org.elasticsearch.common.lucene.store;
 
 import org.apache.lucene.store.IndexInput;
+import org.apache.lucene.util.BitUtil;
 
 import java.io.EOFException;
 import java.io.IOException;
@@ -85,4 +86,31 @@ public void readBytes(final byte[] b, final int offset, int len) throws IOExcept
         System.arraycopy(bytes, this.offset + pos, b, offset, len);
         pos += len;
     }
+
+    @Override
+    public short readShort() throws IOException {
+        try {
+            return (short) BitUtil.VH_LE_SHORT.get(bytes, pos);
+        } finally {
+            pos += Short.BYTES;
+        }
+    }
+
+    @Override
+    public int readInt() throws IOException {
+        try {
+            return (int) BitUtil.VH_LE_INT.get(bytes, pos);
+        } finally {
+            pos += Integer.BYTES;
+        }
+    }
+
+    @Override
+    public long readLong() throws IOException {
+        try {
+            return (long) BitUtil.VH_LE_LONG.get(bytes, pos);
+        } finally {
+            pos += Long.BYTES;
+        }
+    }
 }