Add methods and change method access for Hibernate Reactive

Author: dreab8

hibernate-core/src/main/java/org/hibernate/bytecode/enhance/spi/interceptor/EnhancementAsProxyLazinessInterceptor.java

@@ -300,6 +300,13 @@ public boolean hasWrittenFieldNames() {
 		return writtenFieldNames != null && !writtenFieldNames.isEmpty();
 	}
 
+	/*
+	 * Used by Hibernate Reactive
+	 */
+	protected boolean isIdentifier(String attributeName) {
+		return meta.identifierAttributeNames.contains( attributeName );
+	}
+
 	private enum Status {
 		UNINITIALIZED,
 		INITIALIZING,

hibernate-core/src/main/java/org/hibernate/bytecode/internal/BytecodeEnhancementMetadataPojoImpl.java

@@ -40,11 +40,11 @@
  *
  * @author Steve Ebersole
  */
-public final class BytecodeEnhancementMetadataPojoImpl implements BytecodeEnhancementMetadata {
+public class BytecodeEnhancementMetadataPojoImpl implements BytecodeEnhancementMetadata {
 	/**
 	 * Static constructor
 	 */
-	public static BytecodeEnhancementMetadata from(
+	public static BytecodeEnhancementMetadataPojoImpl from(
 			PersistentClass persistentClass,
 			Set<String> identifierAttributeNames,
 			CompositeType nonAggregatedCidMapper,
@@ -75,7 +75,10 @@ public static BytecodeEnhancementMetadata from(
 	private final LazyAttributeLoadingInterceptor.EntityRelatedState lazyAttributeLoadingInterceptorState;
 	private volatile transient EnhancementAsProxyLazinessInterceptor.EntityRelatedState enhancementAsProxyInterceptorState;
 
-	BytecodeEnhancementMetadataPojoImpl(
+	/*
+	 * Used by Hibernate Reactive
+	 */
+	protected BytecodeEnhancementMetadataPojoImpl(
 			String entityName,
 			Class<?> entityClass,
 			Set<String> identifierAttributeNames,
@@ -248,9 +251,12 @@ public void injectEnhancedEntityAsProxyInterceptor(
 		);
 	}
 
+	/*
+	 * Used by Hibernate Reactive
+	 */
 	//This state object needs to be lazily initialized as it needs access to the Persister, but once
 	//initialized it can be reused across multiple sessions.
-	private EnhancementAsProxyLazinessInterceptor.EntityRelatedState getEnhancementAsProxyLazinessInterceptorMetastate(SharedSessionContractImplementor session) {
+	public EnhancementAsProxyLazinessInterceptor.EntityRelatedState getEnhancementAsProxyLazinessInterceptorMetastate(SharedSessionContractImplementor session) {
 		EnhancementAsProxyLazinessInterceptor.EntityRelatedState state = this.enhancementAsProxyInterceptorState;
 		if ( state == null ) {
 			final EntityPersister entityPersister = session.getFactory().getMappingMetamodel()
@@ -311,4 +317,17 @@ public void injectInterceptor(
 		return (BytecodeLazyAttributeInterceptor) interceptor;
 	}
 
+	/*
+	 * Used by Hibernate Reactive
+	 */
+	public Class<?> getEntityClass() {
+		return entityClass;
+	}
+
+	/*
+	 * Used by Hibernate Reactive
+	 */
+	public LazyAttributeLoadingInterceptor.EntityRelatedState getLazyAttributeLoadingInterceptorState() {
+		return lazyAttributeLoadingInterceptorState;
+	}
 }

hibernate-core/src/main/java/org/hibernate/metamodel/internal/AbstractEntityInstantiatorPojo.java

@@ -66,4 +66,18 @@ public boolean isInstance(Object object) {
 			// this one needed only for guessEntityMode()
 			|| proxyInterface!=null && proxyInterface.isInstance(object);
 	}
+
+	/*
+	 * Used by Hibernate Reactive
+	 */
+	protected boolean isApplyBytecodeInterception() {
+		return applyBytecodeInterception;
+	}
+
+	/*
+	 * Used by Hibernate Reactive
+	 */
+	protected LazyAttributeLoadingInterceptor.EntityRelatedState getLoadingInterceptorState() {
+		return loadingInterceptorState;
+	}
 }

hibernate-core/src/main/java/org/hibernate/metamodel/internal/EntityRepresentationStrategyPojoStandard.java

@@ -163,7 +163,10 @@ private Map<String, PropertyAccess> buildPropertyAccessMap(PersistentClass bootD
 		return propertyAccessMap;
 	}
 
-	private EntityInstantiator determineInstantiator(PersistentClass bootDescriptor, EntityMetamodel entityMetamodel) {
+	/*
+	 * Used by Hibernate Reactive
+	 */
+	protected EntityInstantiator determineInstantiator(PersistentClass bootDescriptor, EntityMetamodel entityMetamodel) {
 		if ( reflectionOptimizer != null && reflectionOptimizer.getInstantiationOptimizer() != null ) {
 			return new EntityInstantiatorPojoOptimized(
 					entityMetamodel,

hibernate-core/src/main/java/org/hibernate/tuple/entity/EntityMetamodel.java

@@ -205,13 +205,12 @@ public EntityMetamodel(
 				idAttributeNames = singleton( identifierAttribute.getName() );
 			}
 
-			bytecodeEnhancementMetadata = BytecodeEnhancementMetadataPojoImpl.from(
+			bytecodeEnhancementMetadata = getBytecodeEnhancementMetadataPojo(
 					persistentClass,
+					creationContext,
 					idAttributeNames,
 					nonAggregatedCidMapper,
-					collectionsInDefaultFetchGroupEnabled,
-					creationContext.getMetadata()
-			);
+					collectionsInDefaultFetchGroupEnabled );
 		}
 		else {
 			bytecodeEnhancementMetadata = new BytecodeEnhancementMetadataNonPojoImpl( persistentClass.getEntityName() );
@@ -512,6 +511,19 @@ && isAbstractClass( persistentClass.getMappedClass() ) ) {
 //		entityNameByInheritanceClassMap = toSmallMap( entityNameByInheritanceClassMapLocal );
 	}
 
+	/*
+	 * Used by Hibernate Reactive
+	 */
+	protected BytecodeEnhancementMetadata getBytecodeEnhancementMetadataPojo(PersistentClass persistentClass, RuntimeModelCreationContext creationContext, Set<String> idAttributeNames, CompositeType nonAggregatedCidMapper, boolean collectionsInDefaultFetchGroupEnabled) {
+		return BytecodeEnhancementMetadataPojoImpl.from(
+				persistentClass,
+				idAttributeNames,
+				nonAggregatedCidMapper,
+				collectionsInDefaultFetchGroupEnabled,
+				creationContext.getMetadata()
+		);
+	}
+
 	private static boolean writePropertyValue(OnExecutionGenerator generator) {
 		final boolean writePropertyValue = generator.writePropertyValue();
 		// TODO: move this validation somewhere else!