Make embeddable diggable

Author: ghost81

ebean-core/src/main/java/io/ebeaninternal/server/deploy/BeanDescriptor.java

@@ -2362,6 +2362,12 @@ ElPropertyValue buildElGetValue(String propName, ElPropertyChainBuilder chain, b
       if (assocProp == null) {
         return null;
       }
+      // this method is an entry-point, although it introduces recursive calls via
+      // buildElPropertyValue -> createElPropertyValue -> buildElGetValue (back to here)
+      // it seams we can initialize ElPropertyChainBuilder at this point and skip further checks.
+      if (chain == null) {
+        chain = new ElPropertyChainBuilder(propName);
+      }
       String remainder = propName.substring(basePos + 1);
       return assocProp.buildElPropertyValue(propName, remainder, chain, propertyDeploy);
     }
@@ -2373,9 +2379,7 @@ ElPropertyValue buildElGetValue(String propName, ElPropertyChainBuilder chain, b
     if (property == null) {
       throw new PersistenceException("No property found for [" + propName + "] in expression " + chain.expression());
     }
-    if (property.containsMany()) {
-      chain.setContainsMany();
-    }
+
     return chain.add(property).build();
   }
 

ebean-core/src/main/java/io/ebeaninternal/server/deploy/BeanPropertyAssoc.java

@@ -161,13 +161,7 @@ public boolean hasForeignKeyIndex() {
   ElPropertyValue createElPropertyValue(String propName, String remainder, ElPropertyChainBuilder chain, boolean propertyDeploy) {
     // associated or embedded bean
     BeanDescriptor<?> embDesc = targetDescriptor();
-    if (chain == null) {
-      chain = new ElPropertyChainBuilder(isEmbedded(), propName);
-    }
     chain.add(this);
-    if (containsMany()) {
-      chain.setContainsMany();
-    }
     return embDesc.buildElGetValue(remainder, chain, propertyDeploy);
   }
 

ebean-core/src/main/java/io/ebeaninternal/server/deploy/BeanPropertyAssocOne.java

@@ -210,17 +210,17 @@ Object naturalKeyVal(Map<String, Object> values) {
   @Override
   public ElPropertyValue buildElPropertyValue(String propName, String remainder, ElPropertyChainBuilder chain, boolean propertyDeploy) {
     if (embedded) {
-      BeanProperty embProp = embeddedPropsMap.get(remainder);
+      String embName = remainder;
+      int basePos = remainder.indexOf('.');
+      if (basePos > -1) {
+        embName = remainder.substring(0, basePos);
+      }
+      BeanProperty embProp = embeddedPropsMap.get(embName);
+
       if (embProp == null) {
-        String msg = "Embedded Property " + remainder + " not found in " + fullName();
+        String msg = "Embedded Property " + embName + " not found in " + fullName();
         throw new PersistenceException(msg);
       }
-      if (chain == null) {
-        chain = new ElPropertyChainBuilder(true, propName);
-      }
-      chain.add(this);
-      chain.setEmbedded(true);
-      return chain.add(embProp).build();
     }
     return createElPropertyValue(propName, remainder, chain, propertyDeploy);
   }

ebean-core/src/main/java/io/ebeaninternal/server/el/ElPropertyChain.java

@@ -37,38 +37,32 @@ public final class ElPropertyChain implements ElPropertyValue {
   private final ScalarType<?> scalarType;
   private final ElPropertyValue lastElPropertyValue;
 
-  public ElPropertyChain(boolean containsMany, boolean embedded, String expression, ElPropertyValue[] chain) {
-    this.containsMany = containsMany;
+  public ElPropertyChain(String expression, boolean containsMany, ElPropertyValue chain[]) {
     this.chain = chain;
     this.expression = expression;
+    this.containsMany = containsMany;
+    
     int dotPos = expression.lastIndexOf('.');
     if (dotPos > -1) {
       this.name = expression.substring(dotPos + 1);
-      if (embedded) {
-        int embPos = expression.lastIndexOf('.', dotPos - 1);
-        this.prefix = embPos == -1 ? null : expression.substring(0, embPos);
-
-      } else {
-        this.prefix = expression.substring(0, dotPos);
-      }
+      this.prefix = expression.substring(0, dotPos);
     } else {
       this.prefix = null;
       this.name = expression;
     }
 
-    this.assocId = chain[chain.length - 1].isAssocId();
-
-    this.last = chain.length - 1;
-    this.lastBeanProperty = chain[chain.length - 1].beanProperty();
+    this.last = this.chain.length - 1;
+    this.lastElPropertyValue = this.chain[this.last];
+    this.assocId = this.lastElPropertyValue.isAssocId();
+    this.lastBeanProperty = lastElPropertyValue.beanProperty();
     if (lastBeanProperty != null) {
       this.scalarType = lastBeanProperty.scalarType();
     } else {
       // case for nested compound type (non-scalar)
       this.scalarType = null;
     }
-    this.lastElPropertyValue = chain[chain.length - 1];
-    this.placeHolder = placeHolder(prefix, lastElPropertyValue, false);
-    this.placeHolderEncrypted = placeHolder(prefix, lastElPropertyValue, true);
+    this.placeHolder = placeHolder(this.prefix, lastElPropertyValue, false);
+    this.placeHolderEncrypted = placeHolder(this.prefix, lastElPropertyValue, true);
   }
 
   @Override

ebean-core/src/main/java/io/ebeaninternal/server/el/ElPropertyChainBuilder.java

@@ -17,25 +17,19 @@ public final class ElPropertyChainBuilder {
 
   private final String expression;
   private final List<ElPropertyValue> chain = new ArrayList<>();
-  private boolean embedded;
-  private boolean containsMany;
+  private boolean containsMany = false;
 
   /**
    * Create with the original expression.
    */
-  public ElPropertyChainBuilder(boolean embedded, String expression) {
-    this.embedded = embedded;
+  public ElPropertyChainBuilder(String expression) {
     this.expression = expression;
   }
 
   public boolean isContainsMany() {
     return containsMany;
   }
 
-  public void setContainsMany() {
-    this.containsMany = true;
-  }
-
   public String expression() {
     return expression;
   }
@@ -48,20 +42,16 @@ public ElPropertyChainBuilder add(ElPropertyValue element) {
       throw new NullPointerException("element null in expression " + expression);
     }
     chain.add(element);
+    if (element.containsMany()) {
+      containsMany = true;
+    }
     return this;
   }
 
   /**
    * Build the immutable ElGetChain from the build information.
    */
   public ElPropertyChain build() {
-    return new ElPropertyChain(containsMany, embedded, expression, chain.toArray(new ElPropertyValue[0]));
-  }
-
-  /**
-   * Permits to set whole chain as embedded when the leaf is embedded
-   */
-  public void setEmbedded(boolean embedded) {
-    this.embedded = embedded;
+    return new ElPropertyChain(expression, containsMany, chain.toArray(new ElPropertyValue[0]));
   }
 }

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

@@ -4,7 +4,6 @@
 import io.ebeaninternal.api.SpiQuery;
 
 import java.util.HashMap;
-import java.util.HashSet;
 import java.util.Map;
 import java.util.Set;
 import java.util.TreeSet;
@@ -16,7 +15,8 @@ final class SqlTreeAlias {
 
   private final SpiQuery.TemporalMode temporalMode;
   private final TreeSet<String> joinProps = new TreeSet<>();
-  private HashSet<String> embeddedPropertyJoins;
+  // embedded property as key, and true if many-where-property
+  private HashMap<String, Boolean> embeddedPropertyJoins;
   private final TreeSet<String> manyWhereJoinProps = new TreeSet<>();
   private final HashMap<String, String> aliasMap = new HashMap<>();
   private final HashMap<String, String> manyWhereAliasMap = new HashMap<>();
@@ -32,19 +32,19 @@ final class SqlTreeAlias {
   /**
    * Add joins to support where predicates
    */
-  void addManyWhereJoins(Set<String> manyWhereJoins) {
+  void addManyWhereJoins(Set<String> manyWhereJoins, STreeType desc) {
     if (manyWhereJoins != null) {
       for (String include : manyWhereJoins) {
-        addPropertyJoin(include, manyWhereJoinProps);
+        addPropertyJoin(include, manyWhereJoinProps, desc, true);
       }
     }
   }
 
-  private void addEmbeddedPropertyJoin(String embProp) {
+  private void addEmbeddedPropertyJoin(String embProp, Boolean isManyWhere) {
     if (embeddedPropertyJoins == null) {
-      embeddedPropertyJoins = new HashSet<>();
+      embeddedPropertyJoins = new HashMap<>();
     }
-    embeddedPropertyJoins.add(embProp);
+    embeddedPropertyJoins.put(embProp, isManyWhere);
   }
 
   /**
@@ -53,21 +53,21 @@ private void addEmbeddedPropertyJoin(String embProp) {
   public void addJoin(Set<String> propJoins, STreeType desc) {
     if (propJoins != null) {
       for (String propJoin : propJoins) {
-        if (desc.isEmbeddedPath(propJoin)) {
-          addEmbeddedPropertyJoin(propJoin);
-        } else {
-          addPropertyJoin(propJoin, joinProps);
-        }
+        addPropertyJoin(propJoin, joinProps, desc, false);
       }
     }
   }
 
-  private void addPropertyJoin(String include, TreeSet<String> set) {
-    if (set.add(include)) {
-      String[] split = SplitName.split(include);
-      if (split[0] != null) {
-        addPropertyJoin(split[0], set);
-      }
+  private void addPropertyJoin(String include, TreeSet<String> set, STreeType desc, Boolean isManyWhere) {
+    if (include == null) {
+      return;
+    }
+    String[] split = SplitName.split(include);
+    if (desc.isEmbeddedPath(include)) {
+      addEmbeddedPropertyJoin(include, isManyWhere);
+      addPropertyJoin(split[0], set, desc, isManyWhere);
+    } else if (set.add(include)) {
+      addPropertyJoin(split[0], set, desc, isManyWhere);
     }
   }
 
@@ -86,11 +86,14 @@ void buildAlias() {
 
   private void mapEmbeddedPropertyAlias() {
     if (embeddedPropertyJoins != null) {
-      for (String propJoin : embeddedPropertyJoins) {
-        String[] split = SplitName.split(propJoin);
+      for (Map.Entry<String,Boolean> propJoin : embeddedPropertyJoins.entrySet()) {
+        String[] split = SplitName.split(propJoin.getKey());
         // the table alias of the parent path
-        String alias = tableAlias(split[0]);
-        aliasMap.put(propJoin, alias);
+        if (Boolean.TRUE.equals(propJoin.getValue())) {
+          manyWhereAliasMap.put(propJoin.getKey(), tableAliasManyWhere(split[0]));
+        } else {
+          aliasMap.put(propJoin.getKey(), tableAlias(split[0]));
+        }
       }
     }
   }

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

@@ -220,7 +220,7 @@ private SqlTreeNode buildRootNode(STreeType desc) {
     if (!rawSql) {
       alias.addJoin(queryDetail.getFetchPaths(), desc);
       alias.addJoin(predicates.predicateIncludes(), desc);
-      alias.addManyWhereJoins(manyWhereJoins.propertyNames());
+      alias.addManyWhereJoins(manyWhereJoins.propertyNames(), desc);
       // build set of table alias
       alias.buildAlias();
       predicates.parseTableAlias(alias);
@@ -681,6 +681,12 @@ private SqlTreeNodeExtraJoin findExtraJoinRoot(String includeProp, SqlTreeNodeEx
             // parent already handled by select
             return childJoin;
           }
+          if (desc.isEmbeddedPath(parentPropertyName)) {
+            // digging in embedded property
+            // so we have to join parent property path
+            includeProp = parentPropertyName;
+            continue;
+          }
 
           SqlTreeNodeExtraJoin parentJoin = joinRegister.get(parentPropertyName);
           if (parentJoin == null) {

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

@@ -83,6 +83,9 @@ public void appendFrom(DbSqlContext ctx, SqlJoinType currentJoinType) {
    * intersection table if this is a ManyToMany node.
    */
   private void appendFromBaseTable(DbSqlContext ctx, SqlJoinType joinType) {
+    if (nodeBeanProp.isEmbedded()) {
+      return;
+    }
     String alias = ctx.tableAliasManyWhere(prefix);
     String parentAlias = ctx.tableAliasManyWhere(parentPrefix);