Merge pull request #3472 from ebean-orm/feature/3461-take-2

#3461 - Add also() for query beans

Author: rbygrave

ebean-api/src/main/java/io/ebean/QueryBuilder.java

@@ -27,6 +27,13 @@ public interface QueryBuilder<SELF, T> extends QueryBuilderProjection<SELF, T> {
    */
   SELF alias(String alias);
 
+  /**
+   * Apply changes to the query using a function.
+   *
+   * @param apply Function that applies changes to the query.
+   */
+  SELF also(Consumer<SELF> apply);
+
   /**
    * Apply changes to the query conditional on the supplied predicate.
    * <p>

ebean-core/src/main/java/io/ebeaninternal/server/query/DefaultFetchGroupQuery.java

@@ -230,6 +230,11 @@ public Query<T> apply(FetchPath fetchPath) {
     throw new RuntimeException("EB102: Only select() and fetch() clause is allowed on FetchGroup");
   }
 
+  @Override
+  public Query<T> also(Consumer<Query<T>> apply) {
+    throw new RuntimeException("EB102: Only select() and fetch() clause is allowed on FetchGroup");
+  }
+
   @Override
   public Query<T> alsoIf(BooleanSupplier predicate, Consumer<Query<T>> apply) {
     throw new RuntimeException("EB102: Only select() and fetch() clause is allowed on FetchGroup");

ebean-core/src/main/java/io/ebeaninternal/server/querydefn/DefaultOrmQuery.java

@@ -293,6 +293,12 @@ public final Query<T> apply(FetchPath fetchPath) {
     return this;
   }
 
+  @Override
+  public Query<T> also(Consumer<Query<T>> apply) {
+    apply.accept(this);
+    return this;
+  }
+
   @Override
   public Query<T> alsoIf(BooleanSupplier predicate, Consumer<Query<T>> consumer) {
     if (predicate.getAsBoolean()) {

ebean-querybean/src/main/java/io/ebean/typequery/QueryBean.java

@@ -275,6 +275,12 @@ public final R apply(FetchPath pathProperties) {
     return root;
   }
 
+  @Override
+  public final R also(Consumer<R> apply) {
+    apply.accept(root);
+    return root;
+  }
+
   @Override
   public final R alsoIf(BooleanSupplier predicate, Consumer<R> apply) {
     if (predicate.getAsBoolean()) {

ebean-querybean/src/test/java/org/querytest/QueryAlsoIfTest.java

@@ -11,6 +11,18 @@ class QueryAlsoIfTest {
 
   int dummy = 1;
 
+  @Test
+  void also() {
+    var q = new QCustomer()
+      .select(name)
+      .name.isNotNull()
+      .also(query -> query.email.isNotNull())
+      .query();
+
+    q.findList();
+    assertThat(q.getGeneratedSql()).isEqualTo("select /* QueryAlsoIfTest.also */ t0.id, t0.name from be_customer t0 where t0.name is not null and t0.email is not null");
+  }
+
   @Test
   void apply() {
     var q = new QCustomer()