HHH-18511 ArrayIndexOutOfBoundsException in ImmutableFetchList

Author: dreab8

hibernate-core/src/main/java/org/hibernate/metamodel/mapping/internal/CompoundNaturalIdMapping.java

@@ -61,6 +61,11 @@ public class CompoundNaturalIdMapping extends AbstractNaturalIdMapping implement
 	private final List<SingularAttributeMapping> attributes;
 
 	private List<JdbcMapping> jdbcMappings;
+	/*
+		This value is used to determine the size of the array used to create the ImmutableFetchList (see org.hibernate.sql.results.graph.internal.ImmutableFetchList#Builder)
+		The Fetch is inserted into the array at a position corresponding to its Fetchable key value.
+	 */
+	private final int maxFetchableKeyIndex;
 
 	public CompoundNaturalIdMapping(
 			EntityMappingType declaringType,
@@ -69,6 +74,14 @@ public CompoundNaturalIdMapping(
 		super( declaringType, isMutable( attributes ) );
 		this.attributes = attributes;
 
+		int maxIndex = 0;
+		for ( SingularAttributeMapping attribute : attributes ) {
+			if ( attribute.getFetchableKey() > maxIndex ) {
+				maxIndex = attribute.getFetchableKey();
+			}
+		}
+		this.maxFetchableKeyIndex = maxIndex + 1;
+
 		creationProcess.registerInitializationCallback(
 				"Determine compound natural-id JDBC mappings ( " + declaringType.getEntityName() + ")",
 				() -> {
@@ -531,6 +544,10 @@ public void visitSubParts(Consumer<ModelPart> consumer, EntityMappingType treatT
 		attributes.forEach( consumer );
 	}
 
+	@Override
+	public int getNumberOfFetchableKeys() {
+		return maxFetchableKeyIndex;
+	}
 
 	public static class DomainResultImpl implements DomainResult<Object[]>, FetchParent {
 		private final NavigablePath navigablePath;