Add proper vector value support

Author: wu-hui

google-cloud-firestore/src/main/java/com/google/cloud/firestore/pipeline/expressions/Constant.java

@@ -7,12 +7,12 @@
 import com.google.cloud.Timestamp;
 import com.google.cloud.firestore.Blob;
 import com.google.cloud.firestore.DocumentReference;
+import com.google.cloud.firestore.FieldValue;
 import com.google.cloud.firestore.GeoPoint;
 import com.google.firestore.v1.Value;
 import java.util.Arrays;
 import java.util.Date;
 import java.util.Map;
-import java.util.stream.Collectors;
 
 @BetaApi
 public final class Constant implements Expr {
@@ -112,9 +112,8 @@ public static <T> Constant of(Map<String, T> value) {
   }
 
   @BetaApi
-  public static Constant ofVector(double[] value) {
-    // Convert double array to List<Double>
-    return new Constant(Arrays.stream(value).boxed().collect(Collectors.toList()));
+  public static Constant vector(double[] value) {
+    return new Constant(FieldValue.vector(value));
   }
 
   @InternalApi

google-cloud-firestore/src/main/java/com/google/cloud/firestore/pipeline/expressions/Expr.java

@@ -1084,7 +1084,7 @@ default CosineDistance cosineDistance(Expr other) {
    */
   @BetaApi
   default CosineDistance cosineDistance(double[] other) {
-    return new CosineDistance(this, Constant.ofVector(other));
+    return new CosineDistance(this, Constant.vector(other));
   }
 
   /**
@@ -1102,7 +1102,7 @@ default CosineDistance cosineDistance(double[] other) {
    */
   @BetaApi
   default EuclideanDistance euclideanDistance(double[] other) {
-    return new EuclideanDistance(this, Constant.ofVector(other));
+    return new EuclideanDistance(this, Constant.vector(other));
   }
 
   /**
@@ -1138,7 +1138,7 @@ default EuclideanDistance euclideanDistance(Expr other) {
    */
   @BetaApi
   default DotProductDistance dotProductDistance(double[] other) {
-    return new DotProductDistance(this, Constant.ofVector(other));
+    return new DotProductDistance(this, Constant.vector(other));
   }
 
   /**

google-cloud-firestore/src/main/java/com/google/cloud/firestore/pipeline/expressions/Function.java

@@ -2277,7 +2277,7 @@ public static CosineDistance cosineDistance(Expr expr, Expr other) {
    */
   @BetaApi
   public static CosineDistance cosineDistance(Expr expr, double[] other) {
-    return new CosineDistance(expr, Constant.ofVector(other));
+    return new CosineDistance(expr, Constant.vector(other));
   }
 
   /**
@@ -2315,7 +2315,7 @@ public static CosineDistance cosineDistance(String field, Expr other) {
    */
   @BetaApi
   public static CosineDistance cosineDistance(String field, double[] other) {
-    return new CosineDistance(Field.of(field), Constant.ofVector(other));
+    return new CosineDistance(Field.of(field), Constant.vector(other));
   }
 
   /**
@@ -2353,7 +2353,7 @@ public static DotProductDistance dotProductDistance(Expr expr, Expr other) {
    */
   @BetaApi
   public static DotProductDistance dotProductDistance(Expr expr, double[] other) {
-    return new DotProductDistance(expr, Constant.ofVector(other));
+    return new DotProductDistance(expr, Constant.vector(other));
   }
 
   /**
@@ -2391,7 +2391,7 @@ public static DotProductDistance dotProductDistance(String field, Expr other) {
    */
   @BetaApi
   public static DotProductDistance dotProductDistance(String field, double[] other) {
-    return new DotProductDistance(Field.of(field), Constant.ofVector(other));
+    return new DotProductDistance(Field.of(field), Constant.vector(other));
   }
 
   /**
@@ -2429,7 +2429,7 @@ public static EuclideanDistance euclideanDistance(Expr expr, Expr other) {
    */
   @BetaApi
   public static EuclideanDistance euclideanDistance(Expr expr, double[] other) {
-    return new EuclideanDistance(expr, Constant.ofVector(other));
+    return new EuclideanDistance(expr, Constant.vector(other));
   }
 
   /**
@@ -2467,7 +2467,7 @@ public static EuclideanDistance euclideanDistance(String field, Expr other) {
    */
   @BetaApi
   public static EuclideanDistance euclideanDistance(String field, double[] other) {
-    return new EuclideanDistance(Field.of(field), Constant.ofVector(other));
+    return new EuclideanDistance(Field.of(field), Constant.vector(other));
   }
 
   /**

google-cloud-firestore/src/test/java/com/google/cloud/firestore/it/ITPipelineTest.java

@@ -828,10 +828,10 @@ public void testDistanceFunctions() throws Exception {
             .pipeline()
             .collection(collection.getPath())
             .select(
-                cosineDistance(Constant.ofVector(sourceVector), targetVector).as("cosineDistance"),
-                dotProductDistance(Constant.ofVector(sourceVector), targetVector)
+                cosineDistance(Constant.vector(sourceVector), targetVector).as("cosineDistance"),
+                dotProductDistance(Constant.vector(sourceVector), targetVector)
                     .as("dotProductDistance"),
-                euclideanDistance(Constant.ofVector(sourceVector), targetVector)
+                euclideanDistance(Constant.vector(sourceVector), targetVector)
                     .as("euclideanDistance"))
             .limit(1)
             .execute()