HHH-18164 Make @IdGeneratorType respect BeanContainer

hibernate-core/src/main/java/org/hibernate/boot/model/internal/GeneratorBinder.java

@@ -65,8 +65,8 @@ import org.hibernate.models.spi.ClassDetails;
 import org.hibernate.models.spi.MemberDetails;
 import org.hibernate.mapping.Value;
 import org.hibernate.resource.beans.container.spi.BeanContainer;
+import org.hibernate.resource.beans.container.spi.ContainedBean;
 import org.hibernate.resource.beans.spi.BeanInstanceProducer;
-import org.hibernate.resource.beans.spi.ManagedBeanRegistry;
 
 import org.hibernate.service.ServiceRegistry;
 import org.hibernate.type.Type;
@@ -552,14 +552,53 @@ public class GeneratorBinder {
 		final Class<? extends Generator> generatorClass = idGeneratorType.value();
 		return creationContext -> {
 			checkGeneratorClass( generatorClass );
-			final Generator generator = instantiateGenerator(
+			Generator generator;
-					annotation,
+			if ( beanContainer != null ) {
-					beanContainer,
+				final ContainedBean<? extends Generator> bean = beanContainer.getBean(
-					creationContext,
+						generatorClass,
-					generatorClass,
+						new BeanContainer.LifecycleOptions() {
-					idAttributeMember,
+							@Override
-					annotationType
+							public boolean canUseCachedReferences() {
-			);
+								return false;
+							}
+
+							@Override
+							public boolean useJpaCompliantCreation() {
+								return true;
+							}
+						},
+						new BeanInstanceProducer() {
+							@SuppressWarnings("unchecked")
+							@Override
+							public <B> B produceBeanInstance(Class<B> beanType) {
+								return (B) instantiateGenerator(
+										annotation,
+										idAttributeMember,
+										annotationType,
+										CustomIdGeneratorCreationContext.class,
+										generatorClass,
+										creationContext
+								);
+							}
+
+							@Override
+							public <B> B produceBeanInstance(String name, Class<B> beanType) {
+								return produceBeanInstance( beanType );
+							}
+						}
+				);
+				generator = bean.getBeanInstance();
+			}
+			else {
+				generator = instantiateGenerator(
+						annotation,
+						beanContainer,
+						creationContext,
+						generatorClass,
+						idAttributeMember,
+						annotationType
+				);
+			}
 			callInitialize( annotation, idAttributeMember, creationContext, generator );
 			callConfigure( creationContext, generator );
 			checkIdGeneratorTiming( annotationType, generator );

hibernate-core/src/main/java/org/hibernate/boot/model/internal/PropertyBinder.java

@@ -63,6 +63,11 @@ import org.hibernate.resource.beans.spi.ManagedBeanRegistry;
 import org.hibernate.service.ServiceRegistry;
 import org.hibernate.usertype.CompositeUserType;
 
+import org.hibernate.resource.beans.container.spi.BeanContainer;
+import org.hibernate.resource.beans.internal.Helper;
+import org.hibernate.resource.beans.spi.ManagedBeanRegistry;
+import org.hibernate.service.ServiceRegistry;
+import org.hibernate.usertype.CompositeUserType;
 import org.jboss.logging.Logger;
 
 import jakarta.persistence.Basic;