recall bug fix for odfe>=1.8 (#155)

* recall bug fix for odfe>=1.8
opendistro-for-elasticsearch · Jul 1, 2020 · 5663cdc · 5663cdc
1 parent 2f0d23e
commit 5663cdc
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/main/java/com/amazon/opendistroforelasticsearch/knn/index/codec/KNNCodecUtil.java b/src/main/java/com/amazon/opendistroforelasticsearch/knn/index/codec/KNNCodecUtil.java
@@ -18,6 +18,7 @@
 
 import org.apache.lucene.index.BinaryDocValues;
 import org.apache.lucene.search.DocIdSetIterator;
+import org.apache.lucene.util.BytesRef;
 
 import java.io.ByteArrayInputStream;
 import java.io.IOException;
@@ -43,10 +44,9 @@ public static KNNCodecUtil.Pair getFloats(BinaryDocValues values) throws IOExcep
         ArrayList<float[]> vectorList = new ArrayList<>();
         ArrayList<Integer> docIdList = new ArrayList<>();
         for (int doc = values.nextDoc(); doc != DocIdSetIterator.NO_MORE_DOCS; doc = values.nextDoc()) {
-            byte[] value = values.binaryValue().bytes;
-
-            try (ByteArrayInputStream byteStream = new ByteArrayInputStream(value);
-                 ObjectInputStream objectStream = new ObjectInputStream(byteStream)) {
+            BytesRef bytesref = values.binaryValue();
+            try (ByteArrayInputStream byteStream = new ByteArrayInputStream(bytesref.bytes, bytesref.offset, bytesref.length);
+                ObjectInputStream objectStream = new ObjectInputStream(byteStream)) {
                 float[] vector = (float[]) objectStream.readObject();
                 vectorList.add(vector);
             } catch (ClassNotFoundException e) {