postfix query.when(condition)

Author: fluentfuture

mug-guava/src/main/java/com/google/mu/safesql/SafeQuery.java

@@ -119,6 +119,15 @@ public static SafeQuery when(
     return condition ? of(query, args) : EMPTY;
   }
 
+  /**
+   * Returns this SafeQuery if {@code condition} is true; otherwise returns {@link #EMPTY}.
+   *
+   * @since 8.4
+   */
+  public SafeQuery when(boolean condition) {
+    return condition ? this : EMPTY;
+  }
+
   /**
    * An optional query that's only rendered if {@code arg} is present; otherwise returns {@link
    * #EMPTY}. It's for use cases where a subquery is only added when present, for example the

mug-guava/src/main/java/com/google/mu/safesql/SafeSql.java

@@ -412,6 +412,15 @@ public static SafeSql when(
     return condition ? of(template, params) : EMPTY;
   }
 
+  /**
+   * Returns this SafeSql if {@code condition} is true; otherwise returns {@link #EMPTY}.
+   *
+   * @since 8.4
+   */
+  public SafeSql when(boolean condition) {
+    return condition ? this : EMPTY;
+  }
+
   /**
    * Returns a {@link Template} of {@link SafeSql} based on the {@code template} string.
    * Useful for creating a constant to be reused with different parameters.

mug-guava/src/test/java/com/google/mu/safesql/SafeQueryTest.java

@@ -1012,6 +1012,17 @@ public void when_conditionalIsTrue_returnsQuery() {
         .isEqualTo(SafeQuery.of("WHERE id = 1"));
   }
 
+  @Test
+  public void postfixWhen_conditionalIsFalse_returnsEmpty() {
+    assertThat(SafeQuery.of("WHERE id = {id}", 1).when(false)).isEqualTo(SafeQuery.EMPTY);
+  }
+
+  @Test
+  public void postfixWhen_conditionalIsTrue_returnsQuery() {
+    assertThat(SafeQuery.of("WHERE id = {id}", 1).when(true))
+        .isEqualTo(SafeQuery.of("WHERE id = 1"));
+  }
+
   @Test
   public void optionally_optionalArgIsEmpty_returnsEmpty() {
     assertThat(SafeQuery.optionally("WHERE id = {id}", /* id */ Optional.empty()))

mug-guava/src/test/java/com/google/mu/safesql/SafeSqlTest.java

@@ -664,6 +664,17 @@ public void when_conditionalIsTrue_returnsQuery() {
         .isEqualTo(SafeSql.of("WHERE id = {id}", 1));
   }
 
+  @Test
+  public void postfixWhen_conditionalIsFalse_returnsEmpty() {
+    assertThat(SafeSql.of("WHERE id = {id}", 1).when(false)).isEqualTo(SafeSql.EMPTY);
+  }
+
+  @Test
+  public void postfixWhen_conditionalIsTrue_returnsQuery() {
+    assertThat(SafeSql.of("WHERE id = {id}", 1).when(true))
+        .isEqualTo(SafeSql.of("WHERE id = {id}", 1));
+  }
+
   @Test
   public void optionally_optionalArgIsEmpty_returnsEmpty() {
     assertThat(SafeSql.optionally("WHERE id = {id}", /* id */ Optional.empty()))