Hibernate/hibernate-orm
1
0
Fork 0
mirror of https://github.com/hibernate/hibernate-orm synced 2025-02-27 22:39:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

HHH-15858 Improvements in Session casting

Browse Source
This commit is contained in:
Sanne Grinovero 2022-12-13 15:45:16 +00:00 committed by Sanne Grinovero
parent 2c3ac612db
commit a7274875ca
10 changed files with 63 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
hibernate-core/src/main/java/org/hibernate/engine/internal/AbstractEntityEntry.java
View File

@ -276,7 +276,7 @@ public void postUpdate(Object entity, Object[] updatedState, Object nextVersion)
session
.getFactory()
.getCustomEntityDirtinessStrategy()
.resetDirty( entity, persister, (Session) session );
.resetDirty( entity, persister, session.asSessionImplementor() );
}
private static void clearDirtyAttributes(final SelfDirtinessTracker entity) {
@ -370,8 +370,9 @@ else if ( isHibernateProxy( entity ) ) {
final CustomEntityDirtinessStrategy customEntityDirtinessStrategy =
getPersistenceContext().getSession().getFactory().getCustomEntityDirtinessStrategy();
if ( customEntityDirtinessStrategy.canDirtyCheck( entity, getPersister(), (Session) getPersistenceContext().getSession() ) ) {
return ! customEntityDirtinessStrategy.isDirty( entity, getPersister(), (Session) getPersistenceContext().getSession() );
final Session session = getPersistenceContext().getSession().asSessionImplementor();
if ( customEntityDirtinessStrategy.canDirtyCheck( entity, getPersister(), session ) ) {
return ! customEntityDirtinessStrategy.isDirty( entity, getPersister(), session );
}
if ( getPersister().hasMutableProperties() ) {

11
hibernate-core/src/main/java/org/hibernate/engine/spi/SessionImplementor.java
View File

@ -112,4 +112,15 @@ default SessionImplementor getSession() {
*/
@Deprecated
void removeOrphanBeforeUpdates(String entityName, Object child);
@Override
default SessionImplementor asSessionImplementor() {
return this;
}
@Override
default boolean isSessionImplementor() {
return true;
}
}

27
hibernate-core/src/main/java/org/hibernate/engine/spi/SharedSessionContractImplementor.java
View File

@ -15,6 +15,7 @@
import org.hibernate.FlushMode;
import org.hibernate.HibernateException;
import org.hibernate.Interceptor;
import org.hibernate.StatelessSession;
import org.hibernate.event.spi.EventSource;
import org.hibernate.query.Query;
import org.hibernate.SharedSessionContract;
@ -440,4 +441,30 @@ default void setEnforcingFetchGraph(boolean enforcingFetchGraph) {
*/
void afterOperation(boolean success);
/**
* If this can be casted to a @{@link SessionImplementor},
* you'll get this returned after an efficient cast.
* @throws ClassCastException if this is not compatible!
*/
default SessionImplementor asSessionImplementor() {
throw new ClassCastException();
}
default boolean isSessionImplementor() {
return false;
}
/**
* If this can be casted to a @{@link StatelessSession},
* you'll get this returned after an efficient cast.
* @throws ClassCastException if this is not compatible!
*/
default StatelessSession asStatelessSession() {
throw new ClassCastException();
}
default boolean isStatelessSession() {
return false;
}
}

2
hibernate-core/src/main/java/org/hibernate/generator/internal/CurrentTimestampGeneration.java
View File

@ -92,7 +92,7 @@ public EnumSet<EventType> getEventTypes() {
@Override
public Object generate(SharedSessionContractImplementor session, Object owner, Object currentValue, EventType eventType) {
return generator.generateValue( (Session) session, owner, currentValue );
return generator.generateValue( session.asSessionImplementor(), owner, currentValue );
}
@Override

2
hibernate-core/src/main/java/org/hibernate/generator/internal/SourceGeneration.java
View File

@ -90,7 +90,7 @@ public EnumSet<EventType> getEventTypes() {
@Override
public Object generate(SharedSessionContractImplementor session, Object owner, Object currentValue, EventType eventType) {
return valueGenerator.generateValue( (Session) session, owner, currentValue );
return valueGenerator.generateValue( session.asSessionImplementor(), owner, currentValue );
}
public Object generateValue(Session session, Object owner) {

13
hibernate-core/src/main/java/org/hibernate/id/ForeignGenerator.java
View File

@ -10,13 +10,10 @@
import org.hibernate.MappingException;
import org.hibernate.Session;
import org.hibernate.StatelessSession;
import org.hibernate.TransientObjectException;
import org.hibernate.engine.internal.ForeignKeys;
import org.hibernate.engine.spi.SharedSessionContractImplementor;
import org.hibernate.id.factory.spi.StandardGenerator;
import org.hibernate.internal.CoreMessageLogger;
import org.hibernate.loader.PropertyPath;
import org.hibernate.persister.entity.EntityPersister;
import org.hibernate.service.ServiceRegistry;
import org.hibernate.type.EntityType;
@ -118,18 +115,18 @@ public Object generate(SharedSessionContractImplementor sessionImplementor, Obje
associatedEntityName
);
}
if ( sessionImplementor instanceof Session ) {
id = ((Session) sessionImplementor).save( associatedEntityName, associatedObject );
if ( sessionImplementor.isSessionImplementor() ) {
id = sessionImplementor.asSessionImplementor().save( associatedEntityName, associatedObject );
}
else if ( sessionImplementor instanceof StatelessSession ) {
id = ((StatelessSession) sessionImplementor).insert( associatedEntityName, associatedObject );
else if ( sessionImplementor.isStatelessSession() ) {
id = sessionImplementor.asStatelessSession().insert( associatedEntityName, associatedObject );
}
else {
throw new IdentifierGenerationException("sessionImplementor is neither Session nor StatelessSession");
}
}
if ( sessionImplementor instanceof Session && ((Session) sessionImplementor).contains( entityName, object ) ) {
if ( sessionImplementor.isSessionImplementor() && sessionImplementor.asSessionImplementor().contains( entityName, object ) ) {
//abort the save (the object is already saved by a circular cascade)
return SHORT_CIRCUIT_INDICATOR;
//throw new IdentifierGenerationException("save associated object first, or disable cascade for inverse association");

11
hibernate-core/src/main/java/org/hibernate/internal/StatelessSessionImpl.java
View File

@ -725,4 +725,15 @@ private JtaPlatform getJtaPlatform() {
private LockMode getNullSafeLockMode(LockMode lockMode) {
return lockMode == null ? LockMode.NONE : lockMode;
}
@Override
public StatelessSession asStatelessSession() {
return this;
}
@Override
public boolean isStatelessSession() {
return true;
}
}

2
hibernate-core/src/main/java/org/hibernate/sql/results/jdbc/internal/DeferredResultSetAccess.java
View File

@ -119,7 +119,7 @@ public DeferredResultSetAccess(
executionContext.getCallback().registerAfterLoadAction(
(session, entity, persister) ->
( (SessionImplementor) session ).lock(
session.asSessionImplementor().lock(
persister.getEntityName(),
entity,
lockOptionsToUse

2
hibernate-core/src/main/java/org/hibernate/tuple/ValueGeneration.java
View File

@ -59,7 +59,7 @@ default EnumSet<EventType> getEventTypes() {
@Override
default Object generate(SharedSessionContractImplementor session, Object owner, Object currentValue, EventType eventType) {
return getValueGenerator().generateValue( (Session) session, owner, currentValue );
return getValueGenerator().generateValue( session.asSessionImplementor(), owner, currentValue );
}
/**

2
hibernate-core/src/main/java/org/hibernate/tuple/VmValueGeneration.java
View File

@ -52,6 +52,6 @@ public EnumSet<EventType> getEventTypes() {
@Override
public Object generate(SharedSessionContractImplementor session, Object owner, Object currentValue, EventType eventType) {
return generator.generateValue( (Session) session, owner, currentValue );
return generator.generateValue( session.asSessionImplementor(), owner, currentValue );
}
}