Add comment and fix formatting

Author: cindy-peng

google-cloud-datastore/src/main/java/com/google/cloud/datastore/Value.java

@@ -218,6 +218,7 @@ public static Value<?> fromPb(com.google.datastore.v1.Value proto) {
     if (valueType == null) return RawValue.MARSHALLER.fromProto(proto).build();
 
     Value<?> returnValue = valueType.getMarshaller().fromProto(proto).build();
+    // If the proto  is a list of doubles with a meaning of 31, use the VectorValue marshaller.
     if (valueType == ValueType.LIST && proto.getMeaning() == VECTOR_MEANING) {
       for (com.google.datastore.v1.Value item : proto.getArrayValue().getValuesList()) {
         if (item.getValueTypeCase() != ValueTypeCase.DOUBLE_VALUE) {

google-cloud-datastore/src/test/java/com/google/cloud/datastore/it/ITDatastoreTest.java

@@ -46,6 +46,7 @@
 import com.google.cloud.datastore.Entity;
 import com.google.cloud.datastore.EntityQuery;
 import com.google.cloud.datastore.EntityValue;
+import com.google.cloud.datastore.FindNearest;
 import com.google.cloud.datastore.FullEntity;
 import com.google.cloud.datastore.GqlQuery;
 import com.google.cloud.datastore.IncompleteKey;
@@ -56,8 +57,6 @@
 import com.google.cloud.datastore.LatLngValue;
 import com.google.cloud.datastore.ListValue;
 import com.google.cloud.datastore.NullValue;
-import com.google.cloud.datastore.VectorValue;
-import com.google.cloud.datastore.FindNearest;
 import com.google.cloud.datastore.PathElement;
 import com.google.cloud.datastore.ProjectionEntity;
 import com.google.cloud.datastore.Query;
@@ -72,6 +71,7 @@
 import com.google.cloud.datastore.TimestampValue;
 import com.google.cloud.datastore.Transaction;
 import com.google.cloud.datastore.ValueType;
+import com.google.cloud.datastore.VectorValue;
 import com.google.cloud.datastore.models.ExecutionStats;
 import com.google.cloud.datastore.models.ExplainMetrics;
 import com.google.cloud.datastore.models.ExplainOptions;
@@ -2120,48 +2120,54 @@ public void testQueryWithStartCursor() {
   @Test
   public void testQueryWithVectorSearch() {
     Entity entity1 =
-            Entity.newBuilder(
-                            Key.newBuilder(PROJECT_ID, KIND1, "name-01", options.getDatabaseId()).build())
-                    .set(
-                            "vector_property",
-                            VectorValue.newBuilder(3.0, 1.0, 2.0).setExcludeFromIndexes(true).build())
-                    .build();
+        Entity.newBuilder(
+                Key.newBuilder(PROJECT_ID, KIND1, "name-01", options.getDatabaseId()).build())
+            .set(
+                "vector_property",
+                VectorValue.newBuilder(3.0, 1.0, 2.0).setExcludeFromIndexes(true).build())
+            .build();
     datastore.put(entity1);
 
     VectorValue vectorValue = VectorValue.newBuilder(1.78, 2.56, 3.88).build();
 
     // Query to find the nearest 1 neighbor
     FindNearest vectorQuery =
-            new FindNearest(
-                    "vector_property", vectorValue, FindNearest.DistanceMeasure.COSINE, 1, "distance");
-    Query<Entity> queryWithVectorSearch = Query.newEntityQueryBuilder().setKind(KIND1).setFindNearest(vectorQuery).build();
+        new FindNearest(
+            "vector_property", vectorValue, FindNearest.DistanceMeasure.COSINE, 1, "distance");
+    Query<Entity> queryWithVectorSearch =
+        Query.newEntityQueryBuilder().setKind(KIND1).setFindNearest(vectorQuery).build();
     QueryResults<Entity> vectorSearchResult = datastore.run(queryWithVectorSearch);
     assertTrue(vectorSearchResult.hasNext());
     assertEquals(entity1, vectorSearchResult.next());
     assertFalse(vectorSearchResult.hasNext());
 
     Entity entity2 =
-            Entity.newBuilder(
-                            Key.newBuilder(PROJECT_ID, KIND1, "name-02", options.getDatabaseId()).build())
-                    .set(
-                            "vector_property",
-                            VectorValue.newBuilder(5.0, 0.7, 2.0).setExcludeFromIndexes(true).build())
-                    .build();
+        Entity.newBuilder(
+                Key.newBuilder(PROJECT_ID, KIND1, "name-02", options.getDatabaseId()).build())
+            .set(
+                "vector_property",
+                VectorValue.newBuilder(5.0, 0.7, 2.0).setExcludeFromIndexes(true).build())
+            .build();
     Entity entity3 =
-            Entity.newBuilder(
-                            Key.newBuilder(PROJECT_ID, KIND1, "name-03", options.getDatabaseId()).build())
-                    .set(
-                            "vector_property",
-                            VectorValue.newBuilder(2.0, 1.7, 1.0).setExcludeFromIndexes(true).build())
-                    .build();
+        Entity.newBuilder(
+                Key.newBuilder(PROJECT_ID, KIND1, "name-03", options.getDatabaseId()).build())
+            .set(
+                "vector_property",
+                VectorValue.newBuilder(2.0, 1.7, 1.0).setExcludeFromIndexes(true).build())
+            .build();
     datastore.put(entity2, entity3);
 
     // Query to find the nearest 2 neighbors
     FindNearest vectorQueryWithLimit =
-            new FindNearest(
-                    "vector_property", VectorValue.newBuilder(2.8, 2.56, 3.88).build(), FindNearest.DistanceMeasure.EUCLIDEAN, 2, "distance");
-
-    Query<Entity> queryWithVectorSearchLimit = Query.newEntityQueryBuilder().setKind(KIND1).setFindNearest(vectorQueryWithLimit).build();
+        new FindNearest(
+            "vector_property",
+            VectorValue.newBuilder(2.8, 2.56, 3.88).build(),
+            FindNearest.DistanceMeasure.EUCLIDEAN,
+            2,
+            "distance");
+
+    Query<Entity> queryWithVectorSearchLimit =
+        Query.newEntityQueryBuilder().setKind(KIND1).setFindNearest(vectorQueryWithLimit).build();
     QueryResults<Entity> resultsWithVectorLimit = datastore.run(queryWithVectorSearchLimit);
     List<Entity> resultsCopy = makeResultsCopy(resultsWithVectorLimit);
     assertEquals(2, resultsCopy.size());