Hibernate
/
hibernate-orm
mirror of https://github.com/hibernate/hibernate-orm
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

HHH-17983 Remove entity keys from batch queue before loading

This commit is contained in:
Marco Belladelli 2024-04-23 15:54:44 +02:00
parent 882709bdf8
commit 181399f84c
2 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
hibernate-core/src/main/java/org/hibernate/loader/ast/internal/EntityBatchLoaderArrayParam.java
View File

@ -134,6 +134,13 @@ public class EntityBatchLoaderArrayParam<T>
getLoadable().getEntityName(), id, Arrays.toString(idsToInitialize) );
}
for ( Object initializedId : idsToInitialize ) {
if ( initializedId != null ) {
// found or not, remove the key from the batch-fetch queue
removeBatchLoadableEntityKey( initializedId, getLoadable(), session );
}
}
LoaderHelper.loadByArrayParameter(
idsToInitialize,
sqlAst,
@ -147,13 +154,6 @@ public class EntityBatchLoaderArrayParam<T>
readOnly,
session
);
for ( Object initializedId : idsToInitialize ) {
if ( initializedId != null ) {
// found or not, remove the key from the batch-fetch queue
removeBatchLoadableEntityKey( initializedId, getLoadable(), session );
}
}
}
@Override

9
hibernate-core/src/main/java/org/hibernate/loader/ast/internal/EntityBatchLoaderInPredicate.java
View File

@ -141,7 +141,6 @@ public class EntityBatchLoaderInPredicate<T>
);
final BatchFetchQueue batchFetchQueue = session.getPersistenceContextInternal().getBatchFetchQueue();
final List<EntityKey> entityKeys = arrayList( sqlBatchSize );
chunker.processChunks(
idsToInitialize,
@ -166,7 +165,11 @@ public class EntityBatchLoaderInPredicate<T>
},
(key, relativePosition, absolutePosition) -> {
if ( key != null ) {
entityKeys.add( session.generateEntityKey( key, getLoadable().getEntityPersister() ) );
final EntityKey entityKey = session.generateEntityKey(
key,
getLoadable().getEntityPersister()
);
batchFetchQueue.removeBatchLoadableEntityKey( entityKey );
}
},
(startIndex) -> {
@ -181,8 +184,6 @@ public class EntityBatchLoaderInPredicate<T>
}
},
(startIndex, nonNullElementCount) -> {
entityKeys.forEach( batchFetchQueue::removeBatchLoadableEntityKey );
entityKeys.clear();
},
session
);