新增了TcVectorQuerier

Author: hemiao

moql-querier-milvus/src/main/java/org/datayoo/moql/querier/milvus/PartitionBy.java

@@ -17,14 +17,11 @@
  */
 package org.datayoo.moql.querier.milvus;
 
-import io.milvus.param.MetricType;
 import org.datayoo.moql.EntityMap;
 import org.datayoo.moql.Operand;
 import org.datayoo.moql.operand.function.AbstractFunction;
 
 import java.util.Arrays;
-import java.util.Collections;
-import java.util.LinkedList;
 import java.util.List;
 
 /**

moql-querier-tcvector/pom.xml

@@ -0,0 +1,45 @@
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+    <parent>
+        <groupId>org.datayoo.moql</groupId>
+        <artifactId>moql-parent</artifactId>
+        <version>1.1.0</version>
+    </parent>
+
+
+    <artifactId>moql-querier-tcvector</artifactId>
+    <version>1.0.0</version>
+
+    <name>moql-querier-tcvector</name>
+    <url>http://maven.apache.org</url>
+
+    <properties>
+        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
+    </properties>
+
+    <dependencies>
+        <dependency>
+            <groupId>junit</groupId>
+            <artifactId>junit</artifactId>
+            <version>4.13.1</version>
+            <scope>test</scope>
+        </dependency>
+
+        <dependency>
+            <groupId>org.datayoo.moql</groupId>
+            <artifactId>moql-querier</artifactId>
+            <version>RELEASE</version>
+            <scope>compile</scope>
+        </dependency>
+
+        <dependency>
+            <groupId>com.tencent.tcvectordb</groupId>
+            <artifactId>vectordatabase-sdk-java</artifactId>
+            <version>1.0.2</version>
+        </dependency>
+
+
+    </dependencies>
+</project>

moql-querier-tcvector/src/main/java/org/datayoo/moql/querier/tcvector/BuilderProxy.java

@@ -0,0 +1,74 @@
+package org.datayoo.moql.querier.tcvector;
+
+import com.tencent.tcvectordb.model.param.dml.*;
+
+import java.util.List;
+
+public class BuilderProxy {
+  SearchByVectorParam.Builder searchBuilder;
+
+  QueryParam.Builder queryBuilder;
+
+  protected boolean search = false;
+
+  public BuilderProxy() {
+    searchBuilder = SearchByVectorParam.newBuilder();
+    searchBuilder.withLimit(1);
+
+    queryBuilder = QueryParam.newBuilder();
+  }
+
+  public BuilderProxy withFilter(String expr) {
+    expr = expr.replace('\'', '"');
+    searchBuilder.withFilter(new Filter(expr));
+    queryBuilder.withFilter(new Filter(expr));
+    return this;
+  }
+
+  public BuilderProxy withOutFields(List<String> fields) {
+    queryBuilder.withOutputFields(fields);
+    searchBuilder.withOutputFields(fields);
+    return this;
+  }
+
+  public boolean isSearchMode() {
+    return search;
+  }
+
+  public BuilderProxy withSearchVectors(List<List<Double>> vectors) {
+    this.search = true;
+    searchBuilder.withVectors(vectors);
+    return this;
+  }
+
+  public BuilderProxy withQueryId(List<String> documentIds) {
+    queryBuilder.withDocumentIds(documentIds);
+    return this;
+  }
+
+  public BuilderProxy withLimit(int limit) {
+    queryBuilder.withLimit(limit);
+    searchBuilder.withLimit(limit);
+    return this;
+  }
+
+  public BuilderProxy withOffset(int offset) {
+    queryBuilder.withOffset(offset);
+    return this;
+  }
+
+  public SearchByVectorParam buildSearch() {
+
+    return searchBuilder.build();
+  }
+
+  public BuilderProxy withParams(Params params) {
+    searchBuilder.withParams(params);
+    return this;
+  }
+
+  public QueryParam buildQuery() {
+    return queryBuilder.build();
+  }
+
+}