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

Lookup RevisionTypeType rather than re-create CustomType instances

This commit is contained in:
Chris Cranford 2021-03-11 15:33:14 -05:00 committed by Christian Beikov
parent fd73a1af85
commit 5dfbaf58ef
2 changed files with 16 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
hibernate-envers/src/main/java/org/hibernate/envers/internal/entities/mapper/relation/query/AbstractRelationQueryGenerator.java
View File

@ -22,7 +22,7 @@ import org.hibernate.envers.internal.tools.query.Parameters;
import org.hibernate.envers.internal.tools.query.QueryBuilder;
import org.hibernate.envers.strategy.AuditStrategy;
import org.hibernate.query.Query;
import org.hibernate.type.CustomType;
import org.hibernate.type.BasicType;
import static org.hibernate.envers.internal.entities.mapper.relation.query.QueryConstants.DEL_REVISION_TYPE_PARAMETER;
import static org.hibernate.envers.internal.entities.mapper.relation.query.QueryConstants.REVISION_PARAMETER;
@ -66,12 +66,13 @@ public abstract class AbstractRelationQueryGenerator implements RelationQueryGen
public Query getQuery(SharedSessionContractImplementor session, Object primaryKey, Number revision, boolean removed) {
final String queryString = getQueryString( session.getFactory(), removed );
final BasicType<?> revisionType = session.getFactory()
.getTypeConfiguration()
.getBasicTypeRegistry()
.getRegisteredType( RevisionTypeType.class );
final Query query = session.createQuery( queryString );
query.setParameter(
DEL_REVISION_TYPE_PARAMETER,
RevisionType.DEL,
new CustomType( new RevisionTypeType(), session.getFactory().getTypeConfiguration() )
);
query.setParameter( DEL_REVISION_TYPE_PARAMETER, RevisionType.DEL, revisionType );
query.setParameter( REVISION_PARAMETER, revision );
final IdMapper prefixIdMapper = referencingIdData.getPrefixedMapper();

18
hibernate-envers/src/main/java/org/hibernate/envers/internal/tools/query/QueryBuilder.java
View File

@ -8,13 +8,11 @@ package org.hibernate.envers.internal.tools.query;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import javax.persistence.criteria.JoinType;
import org.hibernate.HibernateException;
import org.hibernate.NotYetImplementedFor6Exception;
import org.hibernate.Session;
import org.hibernate.engine.spi.SessionFactoryImplementor;
@ -25,8 +23,7 @@ import org.hibernate.envers.internal.tools.StringTools;
import org.hibernate.envers.internal.tools.Triple;
import org.hibernate.envers.tools.Pair;
import org.hibernate.query.Query;
import org.hibernate.sql.Template;
import org.hibernate.type.CustomType;
import org.hibernate.type.BasicType;
/**
* A class for incrementally building a HQL query.
@ -68,6 +65,8 @@ public class QueryBuilder {
private final SessionFactoryImplementor sessionFactory;
private final BasicType<?> revisionType;
/**
* @param entityName Main entity which should be selected.
* @param alias Alias of the entity
@ -89,6 +88,10 @@ public class QueryBuilder {
this.paramCounter = paramCounter;
this.sessionFactory = sessionFactory;
this.revisionType = sessionFactory.getTypeConfiguration()
.getBasicTypeRegistry()
.getRegisteredType( RevisionTypeType.class );
final Parameters rootParameters = new Parameters( alias, "and", paramCounter );
parameters.add( rootParameters );
@ -105,6 +108,7 @@ public class QueryBuilder {
this.entityName = other.entityName;
this.alias = other.alias;
this.sessionFactory = other.sessionFactory;
this.revisionType = other.revisionType;
this.aliasCounter = other.aliasCounter.deepCopy();
this.paramCounter = other.paramCounter.deepCopy();
for (final Parameters params : other.parameters) {
@ -318,11 +322,7 @@ public class QueryBuilder {
for ( Map.Entry<String, Object> paramValue : queryParamValues.entrySet() ) {
if ( paramValue.getValue() instanceof RevisionType ) {
// this is needed when the ClassicQueryTranslatorFactory is used
query.setParameter(
paramValue.getKey(),
paramValue.getValue(),
new CustomType( new RevisionTypeType(), sessionFactory.getTypeConfiguration() )
);
query.setParameter( paramValue.getKey(), paramValue.getValue(), revisionType );
}
else {
query.setParameter( paramValue.getKey(), paramValue.getValue() );