Browse Source

attempt to solve bytebuffer compile issue

Signed-off-by: Zhiru Zhu <zzhu@fandm.edu>
Zhiru Zhu 5 years ago
parent
commit
b95805fd07

+ 1 - 1
pom.xml

@@ -25,7 +25,7 @@
 
     <groupId>io.milvus</groupId>
     <artifactId>milvus-sdk-java</artifactId>
-    <version>0.5.0</version>
+    <version>0.6.0-SNAPSHOT</version>
     <packaging>jar</packaging>
 
     <name>io.milvus:milvus-sdk-java</name>

+ 1 - 1
src/main/java/io/milvus/client/MilvusClient.java

@@ -24,7 +24,7 @@ import java.util.List;
 /** The Milvus Client Interface */
 public interface MilvusClient {
 
-  String clientVersion = "0.5.0";
+  String clientVersion = "0.6.0";
 
   /** @return current Milvus client version: 0.5.0 */
   default String getClientVersion() {

+ 2 - 1
src/main/java/io/milvus/client/MilvusGrpcClient.java

@@ -27,6 +27,7 @@ import io.grpc.StatusRuntimeException;
 import org.apache.commons.collections4.ListUtils;
 
 import javax.annotation.Nonnull;
+import java.nio.Buffer;
 import java.nio.ByteBuffer;
 import java.util.ArrayList;
 import java.util.List;
@@ -1077,7 +1078,7 @@ public class MilvusGrpcClient implements MilvusClient {
         rowRecordBuilder.addAllFloatData(floatVectors.get(i));
       }
       if (i < binaryVectors.size()) {
-        binaryVectors.get(i).rewind();
+        ((Buffer)binaryVectors.get(i)).rewind();
         rowRecordBuilder.setBinaryData(ByteString.copyFrom(binaryVectors.get(i)));
       }
 

+ 6 - 0
src/test/java/io/milvus/client/MilvusGrpcClientTest.java

@@ -60,6 +60,12 @@ class MilvusClientTest {
     List<ByteBuffer> vectors = new ArrayList<>();
     final long dimensionInByte = dimension / 8;
     for (long i = 0; i < vectorCount; ++i) {
+//      byte[] array = new byte[(int) dimensionInByte];
+//      for (int j = 0; j < dimensionInByte; j++) {
+//        Long num = Math.round(Math.random());
+//        array[j] = num.byteValue();
+//      }
+//      ByteBuffer buffer = ByteBuffer.wrap(array);
       ByteBuffer byteBuffer = ByteBuffer.allocate((int) dimensionInByte);
       random.nextBytes(byteBuffer.array());
       vectors.add(byteBuffer);