startsWith and endsWith

Author: wu-hui

google-cloud-firestore/src/main/java/com/google/cloud/firestore/Pipeline.java

@@ -81,6 +81,7 @@
  */
 @BetaApi
 public final class Pipeline {
+  private static Logger logger = Logger.getLogger(Pipeline.class.getName());
   private final ImmutableList<Stage> stages;
   private final Firestore db;
 
@@ -388,7 +389,7 @@ public void onComplete() {
           }
         };
 
-    Logger.getLogger("Pipeline").log(Level.WARNING, "Sending request: " + request);
+    logger.log(Level.INFO, "Sending pipeline request: " + request.getStructuredPipeline());
 
     rpcContext.streamRequest(request, observer, rpcContext.getClient().executePipelineCallable());
   }

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

@@ -0,0 +1,13 @@
+package com.google.cloud.firestore.pipeline.expressions;
+
+import com.google.api.core.BetaApi;
+import com.google.api.core.InternalApi;
+import com.google.common.collect.Lists;
+
+@BetaApi
+public final class EndsWith extends Function implements FilterCondition {
+  @InternalApi
+  EndsWith(Expr expr, Expr postfix) {
+    super("ends_with", Lists.newArrayList(expr, postfix));
+  }
+}

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

@@ -228,6 +228,26 @@ default RegexMatch regexMatches(String regex) {
     return new RegexMatch(this, Constant.of(regex));
   }
 
+  @BetaApi
+  default StartsWith startsWith(String prefix) {
+    return new StartsWith(this, Constant.of(prefix));
+  }
+
+  @BetaApi
+  default StartsWith startsWith(Expr prefix) {
+    return new StartsWith(this, prefix);
+  }
+
+  @BetaApi
+  default EndsWith endsWith(String postfix) {
+    return new EndsWith(this, Constant.of(postfix));
+  }
+
+  @BetaApi
+  default EndsWith endsWith(Expr postfix) {
+    return new EndsWith(this, postfix);
+  }
+
   @BetaApi
   default StrConcat strConcat(List<Expr> elements) {
     return new StrConcat(this, elements);

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

@@ -485,6 +485,36 @@ public static RegexMatch regexMatch(String field, String pattern) {
     return new RegexMatch(Field.of(field), Constant.of(pattern));
   }
 
+  @BetaApi
+  public static StartsWith startsWith(String field, String prefix) {
+    return new StartsWith(Field.of(field), Constant.of(prefix));
+  }
+
+  @BetaApi
+  public static StartsWith startsWith(String field, Expr prefix) {
+    return new StartsWith(Field.of(field), prefix);
+  }
+
+  @BetaApi
+  public static StartsWith startsWith(Expr expr, Expr prefix) {
+    return new StartsWith(expr, prefix);
+  }
+
+  @BetaApi
+  public static EndsWith endsWith(String field, String prefix) {
+    return new EndsWith(Field.of(field), Constant.of(prefix));
+  }
+
+  @BetaApi
+  public static EndsWith endsWith(String field, Expr prefix) {
+    return new EndsWith(Field.of(field), prefix);
+  }
+
+  @BetaApi
+  public static EndsWith endsWith(Expr expr, Expr prefix) {
+    return new EndsWith(expr, prefix);
+  }
+
   @BetaApi
   public static StrConcat strConcat(Expr expr, Expr... elements) {
     return new StrConcat(expr, Arrays.asList(elements));

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

@@ -0,0 +1,13 @@
+package com.google.cloud.firestore.pipeline.expressions;
+
+import com.google.api.core.BetaApi;
+import com.google.api.core.InternalApi;
+import com.google.common.collect.Lists;
+
+@BetaApi
+public final class StartsWith extends Function implements FilterCondition {
+  @InternalApi
+  StartsWith(Expr expr, Expr prefix) {
+    super("starts_with", Lists.newArrayList(expr, prefix));
+  }
+}

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

@@ -30,6 +30,7 @@
 import static com.google.cloud.firestore.pipeline.expressions.Function.cosineDistance;
 import static com.google.cloud.firestore.pipeline.expressions.Function.countAll;
 import static com.google.cloud.firestore.pipeline.expressions.Function.dotProductDistance;
+import static com.google.cloud.firestore.pipeline.expressions.Function.endsWith;
 import static com.google.cloud.firestore.pipeline.expressions.Function.eq;
 import static com.google.cloud.firestore.pipeline.expressions.Function.euclideanDistance;
 import static com.google.cloud.firestore.pipeline.expressions.Function.gt;
@@ -39,6 +40,7 @@
 import static com.google.cloud.firestore.pipeline.expressions.Function.not;
 import static com.google.cloud.firestore.pipeline.expressions.Function.or;
 import static com.google.cloud.firestore.pipeline.expressions.Function.parent;
+import static com.google.cloud.firestore.pipeline.expressions.Function.startsWith;
 import static com.google.cloud.firestore.pipeline.expressions.Function.strConcat;
 import static com.google.cloud.firestore.pipeline.expressions.Function.subtract;
 import static com.google.common.truth.Truth.assertThat;
@@ -526,6 +528,43 @@ public void testStrConcat() throws Exception {
                 map("bookInfo", "Douglas Adams - The Hitchhiker's Guide to the Galaxy")));
   }
 
+  @Test
+  public void testStartsWith() throws Exception {
+    List<PipelineResult> results =
+        collection.pipeline()
+            .where(startsWith("title", "The"))
+            .select("title")
+            .sort(Field.of("title").ascending())
+            .execute().get();
+
+    assertThat(data(results))
+        .isEqualTo(
+            Lists.newArrayList(
+                map("title", "The Great Gatsby"),
+                map("title", "The Handmaid's Tale"),
+                map("title", "The Hitchhiker's Guide to the Galaxy"),
+                map("title", "The Lord of the Rings")
+                ));
+  }
+
+  @Test
+  public void testEndsWith() throws Exception {
+    List<PipelineResult> results =
+        collection
+            .pipeline()
+            .where(endsWith(Field.of("title"), Constant.of("y")))
+            .select("title")
+            .sort(Field.of("title").descending())
+            .execute()
+            .get();
+
+    assertThat(data(results))
+        .isEqualTo(
+            Lists.newArrayList(
+                map("title", "The Hitchhiker's Guide to the Galaxy"),
+                map("title", "The Great Gatsby")));
+  }
+
   @Test
   public void testLength() throws Exception {
     List<PipelineResult> results =