HHH-15866 Handle DependantValue when building embedded attr mapping

hibernate · Jan 4, 2023 · 60fc0c0 · 60fc0c0
1 parent 461383c
commit 60fc0c0
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/...ore/src/main/java/org/hibernate/metamodel/mapping/internal/EmbeddableMappingTypeImpl.java b/...ore/src/main/java/org/hibernate/metamodel/mapping/internal/EmbeddableMappingTypeImpl.java
@@ -29,6 +29,7 @@
 import org.hibernate.mapping.BasicValue;
 import org.hibernate.mapping.Column;
 import org.hibernate.mapping.Component;
+import org.hibernate.mapping.DependantValue;
 import org.hibernate.mapping.IndexedConsumer;
 import org.hibernate.mapping.Property;
 import org.hibernate.mapping.Selectable;
@@ -92,6 +93,7 @@ public static EmbeddableMappingTypeImpl from(
 				null,
 				null,
 				null,
+				null,
 				insertability,
 				updateability,
 				embeddedPartBuilder,
@@ -105,6 +107,7 @@ public static EmbeddableMappingTypeImpl from(
 			String rootTableExpression,
 			String[] rootTableKeyColumnNames,
 			Property componentProperty,
+			DependantValue dependantValue,
 			boolean[] insertability,
 			boolean[] updateability,
 			Function<EmbeddableMappingType,EmbeddableValuedModelPart> embeddedPartBuilder,
@@ -130,6 +133,7 @@ public static EmbeddableMappingTypeImpl from(
 								compositeType,
 								rootTableExpression,
 								rootTableKeyColumnNames,
+								dependantValue,
 								insertability,
 								updateability,
 								creationProcess
@@ -294,6 +298,7 @@ private boolean finishInitialization(
 			CompositeType compositeType,
 			String rootTableExpression,
 			String[] rootTableKeyColumnNames,
+			DependantValue dependantValue,
 			boolean[] insertability,
 			boolean[] updateability,
 			MappingModelCreationProcess creationProcess) {
@@ -344,7 +349,9 @@ private boolean finishInitialization(
 			final Value value = bootPropertyDescriptor.getValue();
 			if ( subtype instanceof BasicType ) {
 				final BasicValue basicValue = (BasicValue) value;
-				final Selectable selectable = basicValue.getColumn();
+				final Selectable selectable = dependantValue != null ?
+						dependantValue.getColumns().get( columnPosition ) :
+						basicValue.getColumn();
 				final String containingTableExpression;
 				final String columnExpression;
 				if ( rootTableKeyColumnNames == null ) {

diff --git a/...re/src/main/java/org/hibernate/metamodel/mapping/internal/MappingModelCreationHelper.java b/...re/src/main/java/org/hibernate/metamodel/mapping/internal/MappingModelCreationHelper.java
@@ -31,6 +31,7 @@
 import org.hibernate.mapping.BasicValue;
 import org.hibernate.mapping.Collection;
 import org.hibernate.mapping.Component;
+import org.hibernate.mapping.DependantValue;
 import org.hibernate.mapping.IndexedCollection;
 import org.hibernate.mapping.KeyValue;
 import org.hibernate.mapping.ManyToOne;
@@ -134,6 +135,7 @@ public static EntityIdentifierMapping buildEncapsulatedCompositeIdentifierMappin
 				rootTableName,
 				rootTableKeyColumnNames,
 				bootProperty,
+				null,
 				component.getColumnInsertability(),
 				component.getColumnUpdateability(),
 				embeddable -> new EmbeddedIdentifierMappingImpl(
@@ -266,13 +268,21 @@ public static EmbeddedAttributeMapping buildEmbeddedAttributeMapping(
 				creationProcess
 		);
 
-		final Component component = (Component) bootProperty.getValue();
+		Value componentValue = bootProperty.getValue();
+		DependantValue dependantValue = null;
+		if ( componentValue instanceof DependantValue ) {
+			dependantValue = ( (DependantValue) componentValue );
+			componentValue = dependantValue.getWrappedValue();
+		}
+
+		final Component component = (Component) componentValue;
 		final EmbeddableMappingTypeImpl embeddableMappingType = EmbeddableMappingTypeImpl.from(
 				component,
 				attrType,
 				tableExpression,
 				rootTableKeyColumnNames,
 				bootProperty,
+				dependantValue,
 				component.getColumnInsertability(),
 				component.getColumnUpdateability(),
 				attributeMappingType -> {