HHH-18714 add an explicit type check instead of the cast to raw

Author: gavinking

hibernate-core/src/main/java/org/hibernate/metamodel/model/domain/internal/JpaMetamodelImpl.java

@@ -21,6 +21,7 @@
 
 import org.checkerframework.checker.nullness.qual.Nullable;
 
+import org.hibernate.AnnotationException;
 import org.hibernate.boot.model.NamedEntityGraphDefinition;
 import org.hibernate.boot.query.NamedQueryDefinition;
 import org.hibernate.boot.registry.classloading.spi.ClassLoaderService;
@@ -501,7 +502,7 @@ private void applyNamedEntityGraphs(Collection<NamedEntityGraphDefinition> named
 			}
 
 			final NamedEntityGraph namedEntityGraph = definition.getAnnotation();
-			final RootGraphImpl<?> entityGraph =
+			final RootGraphImplementor<?> entityGraph =
 					createEntityGraph(
 							namedEntityGraph,
 							definition.getRegisteredName(),
@@ -513,19 +514,25 @@ private void applyNamedEntityGraphs(Collection<NamedEntityGraphDefinition> named
 		}
 	}
 
-	private <T> RootGraphImpl<T> createEntityGraph(
+	private <T> RootGraphImplementor<T> createEntityGraph(
 			NamedEntityGraph namedEntityGraph,
 			String registeredName,
 			EntityDomainType<T> entityType,
 			boolean includeAllAttributes) {
-		final RootGraphImpl<T> entityGraph =
+		final RootGraphImplementor<T> entityGraph =
 				createRootGraph( registeredName, entityType, includeAllAttributes );
 
 		if ( namedEntityGraph.subclassSubgraphs() != null ) {
-			for ( var subclassSubgraph : namedEntityGraph.subclassSubgraphs() ) {
-				GraphImplementor<?> subgraph = (GraphImplementor<?>) entityGraph.addTreatedSubgraph(
-						(Class) subclassSubgraph.type() );
-
+			for ( NamedSubgraph subclassSubgraph : namedEntityGraph.subclassSubgraphs() ) {
+				final Class<?> subgraphType = subclassSubgraph.type();
+				final Class<T> graphJavaType = entityGraph.getGraphedType().getJavaType();
+				if ( !graphJavaType.isAssignableFrom( subgraphType ) ) {
+					throw new AnnotationException( "Named subgraph type '" + subgraphType.getName()
+							+ "' is not a subtype of the graph type '" + graphJavaType.getName() + "'" );
+				}
+				@SuppressWarnings("unchecked") // Safe, because we just checked
+				final Class<? extends T> subtype = (Class<? extends T>) subgraphType;
+				final GraphImplementor<? extends T> subgraph = entityGraph.addTreatedSubgraph( subtype );
 				applyNamedAttributeNodes(
 						subclassSubgraph.attributeNodes(),
 						namedEntityGraph,
@@ -541,7 +548,7 @@ private <T> RootGraphImpl<T> createEntityGraph(
 		return entityGraph;
 	}
 
-	private static <T> RootGraphImpl<T> createRootGraph(
+	private static <T> RootGraphImplementor<T> createRootGraph(
 			String name, EntityDomainType<T> entityType, boolean includeAllAttributes) {
 		final RootGraphImpl<T> entityGraph = new RootGraphImpl<>( name, entityType );
 		if ( includeAllAttributes ) {
@@ -783,30 +790,11 @@ public void processJpa(
 
 	private void populateStaticMetamodel(MetadataImplementor bootMetamodel, MetadataContext context) {
 		bootMetamodel.visitNamedHqlQueryDefinitions( definition
-															-> injectTypedQueryReference(
-				definition,
-				namedQueryMetamodelClass(
-						definition,
-						context
-				)
-		) );
+				-> injectTypedQueryReference( definition, namedQueryMetamodelClass( definition, context ) ) );
 		bootMetamodel.visitNamedNativeQueryDefinitions( definition
-																-> injectTypedQueryReference(
-				definition,
-				namedQueryMetamodelClass(
-						definition,
-						context
-				)
-		) );
+				-> injectTypedQueryReference( definition, namedQueryMetamodelClass( definition, context ) ) );
 		bootMetamodel.getNamedEntityGraphs().values().forEach( definition
-																	-> injectEntityGraph(
-				definition,
-				graphMetamodelClass(
-						definition,
-						context
-				),
-				this
-		) );
+				-> injectEntityGraph( definition, graphMetamodelClass( definition, context ), this ) );
 	}
 
 	private Class<?> namedQueryMetamodelClass(NamedQueryDefinition<?> definition, MetadataContext context) {