Please wait. This can take some minutes ...
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.blazebit.persistence.impl.AbstractCommonQueryBuilder Maven / Gradle / Ivy
/*
* Copyright 2014 - 2019 Blazebit.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.blazebit.persistence.impl;
import com.blazebit.persistence.BaseSubqueryBuilder;
import com.blazebit.persistence.CTEBuilder;
import com.blazebit.persistence.CaseWhenStarterBuilder;
import com.blazebit.persistence.CriteriaBuilderFactory;
import com.blazebit.persistence.From;
import com.blazebit.persistence.FullQueryBuilder;
import com.blazebit.persistence.FullSelectCTECriteriaBuilder;
import com.blazebit.persistence.HavingOrBuilder;
import com.blazebit.persistence.JoinOnBuilder;
import com.blazebit.persistence.JoinType;
import com.blazebit.persistence.Keyset;
import com.blazebit.persistence.KeysetBuilder;
import com.blazebit.persistence.LeafOngoingFinalSetOperationCTECriteriaBuilder;
import com.blazebit.persistence.MultipleSubqueryInitiator;
import com.blazebit.persistence.ObjectBuilder;
import com.blazebit.persistence.Path;
import com.blazebit.persistence.RestrictionBuilder;
import com.blazebit.persistence.ReturningModificationCriteriaBuilderFactory;
import com.blazebit.persistence.SelectRecursiveCTECriteriaBuilder;
import com.blazebit.persistence.SimpleCaseWhenStarterBuilder;
import com.blazebit.persistence.StartOngoingSetOperationCTECriteriaBuilder;
import com.blazebit.persistence.SubqueryBuilder;
import com.blazebit.persistence.SubqueryInitiator;
import com.blazebit.persistence.WhereOrBuilder;
import com.blazebit.persistence.WindowBuilder;
import com.blazebit.persistence.impl.function.entity.ValuesEntity;
import com.blazebit.persistence.impl.keyset.KeysetBuilderImpl;
import com.blazebit.persistence.impl.keyset.KeysetImpl;
import com.blazebit.persistence.impl.keyset.KeysetLink;
import com.blazebit.persistence.impl.keyset.KeysetManager;
import com.blazebit.persistence.impl.keyset.KeysetMode;
import com.blazebit.persistence.impl.keyset.SimpleKeysetLink;
import com.blazebit.persistence.impl.query.AbstractCustomQuery;
import com.blazebit.persistence.impl.query.CTENode;
import com.blazebit.persistence.impl.query.CustomQuerySpecification;
import com.blazebit.persistence.impl.query.CustomSQLQuery;
import com.blazebit.persistence.impl.query.CustomSQLTypedQuery;
import com.blazebit.persistence.impl.query.DefaultQuerySpecification;
import com.blazebit.persistence.impl.query.EntityFunctionNode;
import com.blazebit.persistence.impl.query.ObjectBuilderTypedQuery;
import com.blazebit.persistence.impl.query.QuerySpecification;
import com.blazebit.persistence.impl.transform.ExpressionModifierVisitor;
import com.blazebit.persistence.impl.transform.ExpressionTransformerGroup;
import com.blazebit.persistence.impl.transform.OuterFunctionVisitor;
import com.blazebit.persistence.impl.transform.SimpleTransformerGroup;
import com.blazebit.persistence.impl.transform.SizeTransformationVisitor;
import com.blazebit.persistence.impl.transform.SizeTransformerGroup;
import com.blazebit.persistence.impl.transform.SubqueryRecursiveExpressionVisitor;
import com.blazebit.persistence.impl.util.SqlUtils;
import com.blazebit.persistence.parser.AliasReplacementVisitor;
import com.blazebit.persistence.parser.EntityMetamodel;
import com.blazebit.persistence.parser.expression.Expression;
import com.blazebit.persistence.parser.expression.ExpressionFactory;
import com.blazebit.persistence.parser.expression.PathExpression;
import com.blazebit.persistence.parser.expression.Subquery;
import com.blazebit.persistence.parser.expression.SubqueryExpressionFactory;
import com.blazebit.persistence.parser.expression.VisitorAdapter;
import com.blazebit.persistence.parser.expression.modifier.ExpressionModifier;
import com.blazebit.persistence.parser.predicate.Predicate;
import com.blazebit.persistence.parser.util.JpaMetamodelUtils;
import com.blazebit.persistence.spi.ConfigurationSource;
import com.blazebit.persistence.spi.DbmsDialect;
import com.blazebit.persistence.spi.DbmsModificationState;
import com.blazebit.persistence.spi.DbmsStatementType;
import com.blazebit.persistence.spi.ExtendedAttribute;
import com.blazebit.persistence.spi.ExtendedManagedType;
import com.blazebit.persistence.spi.JpqlMacro;
import com.blazebit.persistence.spi.ServiceProvider;
import com.blazebit.persistence.spi.SetOperationType;
import com.blazebit.persistence.spi.ValuesStrategy;
import javax.persistence.EntityManager;
import javax.persistence.Parameter;
import javax.persistence.Query;
import javax.persistence.TemporalType;
import javax.persistence.Tuple;
import javax.persistence.TypedQuery;
import javax.persistence.metamodel.Attribute;
import javax.persistence.metamodel.BasicType;
import javax.persistence.metamodel.EntityType;
import javax.persistence.metamodel.IdentifiableType;
import javax.persistence.metamodel.ManagedType;
import javax.persistence.metamodel.MapAttribute;
import javax.persistence.metamodel.PluralAttribute;
import javax.persistence.metamodel.SingularAttribute;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Calendar;
import java.util.Collection;
import java.util.Collections;
import java.util.Date;
import java.util.EnumSet;
import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
/**
*
* @param The query result type
* @param The concrete builder type
* @param The builder type that should be returned on set operations
* @param The builder type that should be returned on subquery set operations
* @author Christian Beikov
* @author Moritz Becker
* @since 1.0.0
*/
public abstract class AbstractCommonQueryBuilder> implements ServiceProvider, ConfigurationSource {
public static final String ID_PARAM_NAME = "ids";
protected final MainQuery mainQuery;
protected final QueryContext queryContext;
/* This might change when transitioning to a set operation */
protected boolean isMainQuery;
protected final CriteriaBuilderFactoryImpl cbf;
protected final EntityManager em;
protected final DbmsStatementType statementType;
protected final Map, Map> explicitVersionEntities = new HashMap, Map>(0);
protected final ParameterManager parameterManager;
protected final SelectManager selectManager;
protected final WhereManager whereManager;
protected final HavingManager havingManager;
protected final GroupByManager groupByManager;
protected final OrderByManager orderByManager;
protected final JoinManager joinManager;
protected final WindowManager windowManager;
protected final KeysetManager keysetManager;
protected final ResolvingQueryGenerator queryGenerator;
protected final SubqueryInitiatorFactory subqueryInitFactory;
protected final EmbeddableSplittingVisitor embeddableSplittingVisitor;
protected final FunctionalDependencyAnalyzerVisitor functionalDependencyAnalyzerVisitor;
// This builder will be passed in when using set operations
protected FinalSetReturn finalSetOperationBuilder;
protected boolean setOperationEnded;
protected final AliasManager aliasManager;
protected final ExpressionFactory expressionFactory;
// Mutable state
protected Class resultType;
protected int firstResult = 0;
protected int maxResults = Integer.MAX_VALUE;
protected boolean fromClassExplicitlySet = false;
protected final List> transformerGroups;
// Cache
protected String cachedQueryString;
protected String cachedExternalQueryString;
protected ResolvedExpression[] cachedGroupByIdentifierExpressions;
protected boolean hasGroupBy = false;
protected boolean needsCheck = true;
protected boolean hasCollections = false;
// Fetch owner's are evaluated during implicit joining
protected Set nodesToFetch;
private boolean checkSetBuilderEnded = true;
private boolean implicitJoinsApplied = false;
/**
* Create flat copy of builder
*
* @param builder
*/
@SuppressWarnings("unchecked")
protected AbstractCommonQueryBuilder(AbstractCommonQueryBuilder builder) {
this.mainQuery = builder.mainQuery;
this.queryContext = builder.queryContext;
this.isMainQuery = builder.isMainQuery;
this.cbf = builder.cbf;
this.statementType = builder.statementType;
this.orderByManager = builder.orderByManager;
this.parameterManager = builder.parameterManager;
this.selectManager = builder.selectManager;
this.whereManager = (WhereManager) builder.whereManager;
this.havingManager = (HavingManager) builder.havingManager;
this.groupByManager = builder.groupByManager;
this.keysetManager = builder.keysetManager;
this.joinManager = builder.joinManager;
this.windowManager = (WindowManager) builder.windowManager;
this.queryGenerator = builder.queryGenerator;
this.em = builder.em;
this.finalSetOperationBuilder = (FinalSetReturn) builder.finalSetOperationBuilder;
this.subqueryInitFactory = builder.subqueryInitFactory;
this.embeddableSplittingVisitor = builder.embeddableSplittingVisitor;
this.functionalDependencyAnalyzerVisitor = builder.functionalDependencyAnalyzerVisitor;
this.aliasManager = builder.aliasManager;
this.expressionFactory = builder.expressionFactory;
this.transformerGroups = builder.transformerGroups;
this.resultType = builder.resultType;
}
/**
* Create fully copy of builder. Intended for CTEs only.
*
* @param builder
*/
@SuppressWarnings("unchecked")
protected AbstractCommonQueryBuilder(AbstractCommonQueryBuilder builder, MainQuery mainQuery, QueryContext queryContext) {
this.mainQuery = mainQuery;
if (isMainQuery) {
mainQuery.cteManager.init(this);
}
this.queryContext = queryContext;
this.isMainQuery = builder.isMainQuery;
this.statementType = builder.statementType;
this.cbf = mainQuery.cbf;
this.parameterManager = mainQuery.parameterManager;
this.em = mainQuery.em;
this.aliasManager = new AliasManager(queryContext.getParent().aliasManager);
this.expressionFactory = builder.expressionFactory;
this.queryGenerator = new ResolvingQueryGenerator(this.aliasManager, parameterManager, mainQuery.parameterTransformerFactory, mainQuery.jpaProvider, mainQuery.registeredFunctions);
this.joinManager = new JoinManager(mainQuery, this, queryGenerator, this.aliasManager, queryContext.getParent().joinManager, expressionFactory);
this.fromClassExplicitlySet = builder.fromClassExplicitlySet;
this.subqueryInitFactory = joinManager.getSubqueryInitFactory();
SplittingVisitor splittingVisitor = new SplittingVisitor(mainQuery.metamodel, mainQuery.jpaProvider, this.aliasManager);
this.embeddableSplittingVisitor = new EmbeddableSplittingVisitor(mainQuery.metamodel, mainQuery.jpaProvider, this.aliasManager, splittingVisitor);
GroupByExpressionGatheringVisitor groupByExpressionGatheringVisitor = new GroupByExpressionGatheringVisitor(false, this.aliasManager, mainQuery.dbmsDialect);
this.functionalDependencyAnalyzerVisitor = new FunctionalDependencyAnalyzerVisitor(mainQuery.metamodel, splittingVisitor, mainQuery.jpaProvider, this.aliasManager);
this.windowManager = new WindowManager<>(queryGenerator, parameterManager, subqueryInitFactory);
this.whereManager = new WhereManager<>(queryGenerator, parameterManager, subqueryInitFactory, expressionFactory);
this.groupByManager = new GroupByManager(queryGenerator, parameterManager, subqueryInitFactory, mainQuery.jpaProvider, this.aliasManager, embeddableSplittingVisitor, groupByExpressionGatheringVisitor);
this.havingManager = new HavingManager<>(queryGenerator, parameterManager, subqueryInitFactory, expressionFactory, groupByExpressionGatheringVisitor);
this.selectManager = new SelectManager<>(queryGenerator, parameterManager, this, this.joinManager, this.aliasManager, subqueryInitFactory, expressionFactory, mainQuery.jpaProvider, mainQuery, groupByExpressionGatheringVisitor, builder.resultType);
this.orderByManager = new OrderByManager(queryGenerator, parameterManager, subqueryInitFactory, this.joinManager, this.aliasManager, embeddableSplittingVisitor, functionalDependencyAnalyzerVisitor, mainQuery.metamodel, mainQuery.jpaProvider, groupByExpressionGatheringVisitor);
this.keysetManager = new KeysetManager(this, queryGenerator, parameterManager, mainQuery.jpaProvider, mainQuery.dbmsDialect);
final SizeTransformationVisitor sizeTransformationVisitor = new SizeTransformationVisitor(mainQuery, subqueryInitFactory, joinManager, mainQuery.jpaProvider);
this.transformerGroups = Arrays.>asList(
new SimpleTransformerGroup(new OuterFunctionVisitor(joinManager)),
new SimpleTransformerGroup(new SubqueryRecursiveExpressionVisitor()),
new SizeTransformerGroup(sizeTransformationVisitor, orderByManager, selectManager, joinManager, groupByManager));
this.resultType = builder.resultType;
applyFrom(builder, true);
}
protected AbstractCommonQueryBuilder(MainQuery mainQuery, QueryContext queryContext, boolean isMainQuery, DbmsStatementType statementType, Class resultClazz, String alias, AliasManager aliasManager, JoinManager parentJoinManager, ExpressionFactory expressionFactory, FinalSetReturn finalSetOperationBuilder, boolean implicitFromClause) {
if (mainQuery == null) {
throw new NullPointerException("mainQuery");
}
if (statementType == null) {
throw new NullPointerException("statementType");
}
if (resultClazz == null) {
throw new NullPointerException("resultClazz");
}
this.mainQuery = mainQuery;
if (isMainQuery) {
mainQuery.cteManager.init(this);
}
this.queryContext = queryContext;
this.isMainQuery = isMainQuery;
this.statementType = statementType;
this.cbf = mainQuery.cbf;
this.parameterManager = mainQuery.parameterManager;
this.em = mainQuery.em;
this.aliasManager = new AliasManager(aliasManager);
this.expressionFactory = expressionFactory;
this.queryGenerator = new ResolvingQueryGenerator(this.aliasManager, parameterManager, mainQuery.parameterTransformerFactory, mainQuery.jpaProvider, mainQuery.registeredFunctions);
this.joinManager = new JoinManager(mainQuery, this, queryGenerator, this.aliasManager, parentJoinManager, expressionFactory);
if (implicitFromClause) {
// set defaults
if (alias != null) {
// If the user supplies an alias, the intention is clear
fromClassExplicitlySet = true;
}
EntityType type = mainQuery.metamodel.getEntity(resultClazz);
if (type == null) {
// the result class might not be an entity
if (fromClassExplicitlySet) {
// If the intention was to use that as from clause, we have to throw an exception
throw new IllegalArgumentException("The class [" + resultClazz.getName() + "] is not an entity and therefore can't be aliased!");
}
} else {
this.joinManager.addRoot(type, alias);
}
}
this.subqueryInitFactory = joinManager.getSubqueryInitFactory();
SplittingVisitor splittingVisitor = new SplittingVisitor(mainQuery.metamodel, mainQuery.jpaProvider, this.aliasManager);
this.embeddableSplittingVisitor = new EmbeddableSplittingVisitor(mainQuery.metamodel, mainQuery.jpaProvider, this.aliasManager, splittingVisitor);
GroupByExpressionGatheringVisitor groupByExpressionGatheringVisitor = new GroupByExpressionGatheringVisitor(false, this.aliasManager, mainQuery.dbmsDialect);
this.functionalDependencyAnalyzerVisitor = new FunctionalDependencyAnalyzerVisitor(mainQuery.metamodel, splittingVisitor, mainQuery.jpaProvider, this.aliasManager);
this.windowManager = new WindowManager<>(queryGenerator, parameterManager, subqueryInitFactory);
this.whereManager = new WhereManager<>(queryGenerator, parameterManager, subqueryInitFactory, expressionFactory);
this.groupByManager = new GroupByManager(queryGenerator, parameterManager, subqueryInitFactory, mainQuery.jpaProvider, this.aliasManager, embeddableSplittingVisitor, groupByExpressionGatheringVisitor);
this.havingManager = new HavingManager<>(queryGenerator, parameterManager, subqueryInitFactory, expressionFactory, groupByExpressionGatheringVisitor);
this.selectManager = new SelectManager<>(queryGenerator, parameterManager, this, this.joinManager, this.aliasManager, subqueryInitFactory, expressionFactory, mainQuery.jpaProvider, mainQuery, groupByExpressionGatheringVisitor, resultClazz);
this.orderByManager = new OrderByManager(queryGenerator, parameterManager, subqueryInitFactory, this.joinManager, this.aliasManager, embeddableSplittingVisitor, functionalDependencyAnalyzerVisitor, mainQuery.metamodel, mainQuery.jpaProvider, groupByExpressionGatheringVisitor);
this.keysetManager = new KeysetManager(this, queryGenerator, parameterManager, mainQuery.jpaProvider, mainQuery.dbmsDialect);
final SizeTransformationVisitor sizeTransformationVisitor = new SizeTransformationVisitor(mainQuery, subqueryInitFactory, joinManager, mainQuery.jpaProvider);
this.transformerGroups = Arrays.>asList(
new SimpleTransformerGroup(new OuterFunctionVisitor(joinManager)),
new SimpleTransformerGroup(new SubqueryRecursiveExpressionVisitor()),
new SizeTransformerGroup(sizeTransformationVisitor, orderByManager, selectManager, joinManager, groupByManager));
this.resultType = resultClazz;
this.finalSetOperationBuilder = finalSetOperationBuilder;
}
public AbstractCommonQueryBuilder(MainQuery mainQuery, QueryContext queryContext, boolean isMainQuery, DbmsStatementType statementType, Class resultClazz, String alias, FinalSetReturn finalSetOperationBuilder, boolean implicitFromClause) {
this(mainQuery, queryContext, isMainQuery, statementType, resultClazz, alias, null, null, mainQuery.expressionFactory, finalSetOperationBuilder, implicitFromClause);
}
public AbstractCommonQueryBuilder(MainQuery mainQuery, QueryContext queryContext, boolean isMainQuery, DbmsStatementType statementType, Class resultClazz, String alias, FinalSetReturn finalSetOperationBuilder) {
this(mainQuery, queryContext, isMainQuery, statementType, resultClazz, alias, null, null, mainQuery.expressionFactory, finalSetOperationBuilder, true);
}
public AbstractCommonQueryBuilder(MainQuery mainQuery, QueryContext queryContext, boolean isMainQuery, DbmsStatementType statementType, Class resultClazz, String alias) {
this(mainQuery, queryContext, isMainQuery, statementType, resultClazz, alias, null);
}
abstract AbstractCommonQueryBuilder copy(QueryContext queryContext);
void applyFrom(AbstractCommonQueryBuilder builder, boolean fixedSelect) {
if (isMainQuery) {
parameterManager.copyFrom(builder.parameterManager);
mainQuery.cteManager.applyFrom(builder.mainQuery.cteManager);
}
aliasManager.applyFrom(builder.aliasManager);
Map nodeMapping = joinManager.applyFrom(builder.joinManager);
windowManager.applyFrom(builder.windowManager);
whereManager.applyFrom(builder.whereManager);
havingManager.applyFrom(builder.havingManager);
groupByManager.applyFrom(builder.groupByManager);
orderByManager.applyFrom(builder.orderByManager);
setFirstResult(builder.firstResult);
setMaxResults(builder.maxResults);
// TODO: select aliases that are ordered by?
selectManager.setDefaultSelect(nodeMapping, builder.selectManager.getSelectInfos());
if (fixedSelect) {
selectManager.unsetDefaultSelect();
}
// No need to copy the finalSetOperationBuilder as that is only necessary for further builders which isn't possible after copying
collectParameters();
}
public CriteriaBuilderFactory getCriteriaBuilderFactory() {
return cbf;
}
public DbmsStatementType getStatementType() {
return statementType;
}
@SuppressWarnings("unchecked")
public T getService(Class serviceClass) {
if (CriteriaBuilderFactory.class.equals(serviceClass)) {
return (T) cbf;
} else if (ConfigurationSource.class.equals(serviceClass)) {
return (T) this;
} else if (EntityManager.class.equals(serviceClass)) {
return (T) em;
} else if (DbmsDialect.class.equals(serviceClass)) {
return (T) mainQuery.dbmsDialect;
} else if (SubqueryExpressionFactory.class.equals(serviceClass)) {
return (T) mainQuery.subqueryExpressionFactory;
} else if (ExpressionFactory.class.equals(serviceClass)) {
return (T) mainQuery.expressionFactory;
} else if (JoinOnBuilder.class.equals(serviceClass)) {
// TODO: We should think of a better way to expose a where builder to clients as an on builder
// TODO: Setting the expression via this does not clear the cache
return (T) whereManager.startOnBuilder(this);
}
return cbf.getService(serviceClass);
}
@SuppressWarnings("unchecked")
public BuilderType registerMacro(String macroName, JpqlMacro jpqlMacro) {
prepareForModification(null);
this.mainQuery.registerMacro(macroName, jpqlMacro);
return (BuilderType) this;
}
@SuppressWarnings("unchecked")
public BuilderType setProperty(String propertyName, String propertyValue) {
prepareForModification(null);
this.mainQuery.getMutableQueryConfiguration().setProperty(propertyName, propertyValue);
return (BuilderType) this;
}
@SuppressWarnings("unchecked")
public BuilderType setProperties(Map properties) {
prepareForModification(null);
this.mainQuery.getMutableQueryConfiguration().setProperties(properties);
return (BuilderType) this;
}
@SuppressWarnings("unchecked")
public BuilderType setCacheable(boolean cacheable) {
this.mainQuery.getMutableQueryConfiguration().setCacheable(cacheable);
return (BuilderType) this;
}
public boolean isCacheable() {
return this.mainQuery.getQueryConfiguration().isCacheable();
}
public Map getProperties() {
return this.mainQuery.getQueryConfiguration().getProperties();
}
public String getProperty(String name) {
return this.mainQuery.getQueryConfiguration().getProperty(name);
}
@SuppressWarnings("unchecked")
public StartOngoingSetOperationCTECriteriaBuilder> withStartSet(Class cteClass) {
if (!mainQuery.dbmsDialect.supportsWithClause()) {
throw new UnsupportedOperationException("The database does not support the with clause!");
}
prepareForModification(ClauseType.CTE);
return mainQuery.cteManager.withStartSet(cteClass, (BuilderType) this);
}
public boolean hasCte(Class cte) {
return mainQuery.cteManager.hasCte(cte);
}
@SuppressWarnings("unchecked")
public FullSelectCTECriteriaBuilder with(Class cteClass) {
if (!mainQuery.dbmsDialect.supportsWithClause()) {
throw new UnsupportedOperationException("The database does not support the with clause!");
}
prepareForModification(ClauseType.CTE);
return mainQuery.cteManager.with(cteClass, (BuilderType) this);
}
public BuilderType withCtesFrom(CTEBuilder cteBuilder) {
MainQuery mainQuery = ((AbstractCommonQueryBuilder) cteBuilder).mainQuery;
if (this.mainQuery == mainQuery) {
throw new IllegalStateException("Can't copy the CTEs from itself back into itself!");
}
if (mainQuery.cteManager.hasCtes()) {
if (!mainQuery.dbmsDialect.supportsWithClause()) {
throw new UnsupportedOperationException("The database does not support the with clause!");
}
prepareForModification(ClauseType.CTE);
this.parameterManager.applyToCteFrom(mainQuery.parameterManager);
this.mainQuery.cteManager.applyFrom(mainQuery.cteManager);
}
return (BuilderType) this;
}
@SuppressWarnings("unchecked")
public SelectRecursiveCTECriteriaBuilder withRecursive(Class cteClass) {
if (!mainQuery.dbmsDialect.supportsWithClause()) {
throw new UnsupportedOperationException("The database does not support the with clause!");
}
prepareForModification(ClauseType.CTE);
return mainQuery.cteManager.withRecursive(cteClass, (BuilderType) this);
}
@SuppressWarnings("unchecked")
public ReturningModificationCriteriaBuilderFactory withReturning(Class cteClass) {
if (!mainQuery.dbmsDialect.supportsWithClause()) {
throw new UnsupportedOperationException("The database does not support the with clause!");
}
if (!mainQuery.dbmsDialect.supportsModificationQueryInWithClause()) {
throw new UnsupportedOperationException("The database does not support modification queries in the with clause!");
}
prepareForModification(ClauseType.CTE);
return mainQuery.cteManager.withReturning(cteClass, (BuilderType) this);
}
public SetReturn union() {
return addSetOperation(SetOperationType.UNION);
}
public SetReturn unionAll() {
return addSetOperation(SetOperationType.UNION_ALL);
}
public SetReturn intersect() {
return addSetOperation(SetOperationType.INTERSECT);
}
public SetReturn intersectAll() {
return addSetOperation(SetOperationType.INTERSECT_ALL);
}
public SetReturn except() {
return addSetOperation(SetOperationType.EXCEPT);
}
public SetReturn exceptAll() {
return addSetOperation(SetOperationType.EXCEPT_ALL);
}
public SubquerySetReturn startUnion() {
return addSubquerySetOperation(SetOperationType.UNION);
}
public SubquerySetReturn startUnionAll() {
return addSubquerySetOperation(SetOperationType.UNION_ALL);
}
public SubquerySetReturn startIntersect() {
return addSubquerySetOperation(SetOperationType.INTERSECT);
}
public SubquerySetReturn startIntersectAll() {
return addSubquerySetOperation(SetOperationType.INTERSECT_ALL);
}
public SubquerySetReturn startExcept() {
return addSubquerySetOperation(SetOperationType.EXCEPT);
}
public SubquerySetReturn startExceptAll() {
return addSubquerySetOperation(SetOperationType.EXCEPT_ALL);
}
public SubquerySetReturn startSet() {
return addSubquerySetOperation(null);
}
private SetReturn addSetOperation(SetOperationType type) {
// We support set operations in subqueries since we use custom functions
if (!(this instanceof BaseSubqueryBuilder)) {
mainQuery.assertSupportsAdvancedSql("Illegal use of SET operation!");
}
prepareForModification(ClauseType.SELECT);
this.setOperationEnded = true;
// We only check non-empty queries since empty ones will be replaced
if (!isEmpty()) {
prepareAndCheck();
}
FinalSetReturn finalSetOperationBuilder = this.finalSetOperationBuilder;
if (finalSetOperationBuilder == null) {
finalSetOperationBuilder = createFinalSetOperationBuilder(type, false);
finalSetOperationBuilder.setOperationManager.setStartQueryBuilder(this);
this.finalSetOperationBuilder = finalSetOperationBuilder;
} else {
SetOperationManager oldOperationManager = finalSetOperationBuilder.setOperationManager;
if (oldOperationManager.getOperator() == null) {
oldOperationManager.setOperator(type);
} else if (oldOperationManager.getOperator() != type) {
// Put existing set operands into a sub builder and use the sub builder as new start
FinalSetReturn subFinalSetOperationBuilder = createFinalSetOperationBuilder(oldOperationManager.getOperator(), false);
subFinalSetOperationBuilder.setOperationManager.setStartQueryBuilder(oldOperationManager.getStartQueryBuilder());
subFinalSetOperationBuilder.setOperationManager.getSetOperations().addAll(oldOperationManager.getSetOperations());
oldOperationManager.setStartQueryBuilder(subFinalSetOperationBuilder);
oldOperationManager.getSetOperations().clear();
oldOperationManager.setOperator(type);
}
}
SetReturn setOperand = createSetOperand(finalSetOperationBuilder);
finalSetOperationBuilder.setOperationManager.addSetOperation((AbstractCommonQueryBuilder) setOperand);
return setOperand;
}
private SubquerySetReturn addSubquerySetOperation(SetOperationType type) {
prepareForModification(ClauseType.SELECT);
this.setOperationEnded = true;
// We only check non-empty queries since empty ones will be replaced
if (!isEmpty()) {
prepareAndCheck();
}
FinalSetReturn parentFinalSetOperationBuilder = this.finalSetOperationBuilder;
if (parentFinalSetOperationBuilder == null) {
parentFinalSetOperationBuilder = createFinalSetOperationBuilder(type, false);
parentFinalSetOperationBuilder.setOperationManager.setStartQueryBuilder(this);
this.finalSetOperationBuilder = parentFinalSetOperationBuilder;
this.needsCheck = true;
} else {
SetOperationManager oldParentOperationManager = finalSetOperationBuilder.setOperationManager;
if (oldParentOperationManager.getOperator() == null) {
oldParentOperationManager.setOperator(type);
} else if (oldParentOperationManager.getOperator() != type) {
// Put existing set operands into a sub builder and use the sub builder as new start
FinalSetReturn subFinalSetOperationBuilder = createFinalSetOperationBuilder(oldParentOperationManager.getOperator(), false);
subFinalSetOperationBuilder.setOperationManager.setStartQueryBuilder(oldParentOperationManager.getStartQueryBuilder());
subFinalSetOperationBuilder.setOperationManager.getSetOperations().addAll(oldParentOperationManager.getSetOperations());
oldParentOperationManager.setStartQueryBuilder(subFinalSetOperationBuilder);
oldParentOperationManager.getSetOperations().clear();
oldParentOperationManager.setOperator(type);
}
}
FinalSetReturn finalSetOperationBuilder = createFinalSetOperationBuilder(type, true);
SubquerySetReturn subquerySetOperand = createSubquerySetOperand(finalSetOperationBuilder, parentFinalSetOperationBuilder);
finalSetOperationBuilder.setOperationManager.setStartQueryBuilder((AbstractCommonQueryBuilder) subquerySetOperand);
if (type != null) {
parentFinalSetOperationBuilder.setOperationManager.addSetOperation(finalSetOperationBuilder);
} else {
parentFinalSetOperationBuilder.setOperationManager.setStartQueryBuilder(finalSetOperationBuilder);
}
return subquerySetOperand;
}
protected FinalSetReturn createFinalSetOperationBuilder(SetOperationType operator, boolean nested) {
throw new IllegalArgumentException("Set operations aren't supported!");
}
protected SetReturn createSetOperand(FinalSetReturn baseQueryBuilder) {
throw new IllegalArgumentException("Set operations aren't supported!");
}
protected SubquerySetReturn createSubquerySetOperand(FinalSetReturn baseQueryBuilder, FinalSetReturn resultFinalSetOperationBuilder) {
throw new IllegalArgumentException("Set operations aren't supported!");
}
public BuilderType from(String correlationPath) {
return from(correlationPath, null);
}
public BuilderType from(String correlationPath, String alias) {
if (!(this instanceof SubqueryBuilder)) {
throw new IllegalStateException("Cannot use a correlation path in a non-subquery!");
}
prepareForModification(ClauseType.JOIN);
joinManager.addRoot(correlationPath, alias);
return (BuilderType) this;
}
public BuilderType from(Class clazz) {
return from(clazz, null);
}
public BuilderType from(Class clazz, String alias) {
return from(clazz, alias, null);
}
public BuilderType from(EntityType entityType) {
return from(entityType, null);
}
public BuilderType from(EntityType entityType, String alias) {
return from(entityType, alias, null);
}
public BuilderType fromCte(Class clazz, String cteName) {
return fromCte(clazz, cteName, null);
}
public BuilderType fromCte(Class clazz, String cteName, String alias) {
return from(clazz, alias, null);
}
public BuilderType fromOld(Class clazz) {
return fromOld(clazz, null);
}
public BuilderType fromOld(Class clazz, String alias) {
return from(clazz, alias, DbmsModificationState.OLD);
}
public BuilderType fromNew(Class clazz) {
return fromNew(clazz, null);
}
public BuilderType fromNew(Class clazz, String alias) {
return from(clazz, alias, DbmsModificationState.NEW);
}
public BuilderType fromValues(Class valueClass, String alias, Collection values) {
BuilderType result = fromValues(valueClass, alias, values.size());
setParameter(alias, values);
return result;
}
public BuilderType fromValues(Class entityBaseClass, String attributeName, String alias, Collection values) {
BuilderType result = fromValues(entityBaseClass, attributeName, alias, values.size());
setParameter(alias, values);
return result;
}
public BuilderType fromIdentifiableValues(Class valueClass, String alias, Collection values) {
BuilderType result = fromIdentifiableValues(valueClass, alias, values.size());
setParameter(alias, values);
return result;
}
public BuilderType fromIdentifiableValues(Class valueClass, String alias, int valueCount) {
prepareForModification(ClauseType.JOIN);
if (!fromClassExplicitlySet) {
// When from is explicitly called we have to revert the implicit root
if (joinManager.getRoots().size() > 0) {
joinManager.removeRoot();
}
}
ManagedType type = mainQuery.metamodel.getManagedType(valueClass);
if (!JpaMetamodelUtils.isIdentifiable(type)) {
throw new IllegalArgumentException("Only identifiable types allowed!");
}
joinManager.addRootValues(valueClass, valueClass, alias, valueCount, null, null, true, true, null, null, null, null);
fromClassExplicitlySet = true;
return (BuilderType) this;
}
public BuilderType fromValues(Class valueClass, String alias, int valueCount) {
ManagedType type = mainQuery.metamodel.getManagedType(valueClass);
if (type == null) {
String sqlType = mainQuery.dbmsDialect.getSqlType(valueClass);
if (sqlType == null) {
throw new IllegalArgumentException("The basic type " + valueClass.getSimpleName() + " has no column type registered in the DbmsDialect '" + mainQuery.dbmsDialect.getClass().getName() + "'! Register a column type or consider using the fromValues variant that extracts column types from entity attributes!");
}
String typeName = cbf.getNamedTypes().get(valueClass);
if (typeName == null) {
throw new IllegalArgumentException("Unsupported non-managed type for VALUES clause: " + valueClass.getName() + ". You can register the type via com.blazebit.persistence.spi.CriteriaBuilderConfiguration.registerNamedType. Please report this so that we can add the type definition as well!");
}
String castedParameter = mainQuery.dbmsDialect.cast("?", sqlType);
ExtendedAttribute valuesLikeAttribute = mainQuery.metamodel.getManagedType(ExtendedManagedType.class, ValuesEntity.class).getAttribute("value");
prepareFromModification();
joinManager.addRootValues(ValuesEntity.class, valueClass, alias, valueCount, typeName, castedParameter, false, true, "value", valuesLikeAttribute, null, null);
} else if (type instanceof EntityType) {
prepareFromModification();
joinManager.addRootValues(valueClass, valueClass, alias, valueCount, null, null, false, true, null, null, null, null);
} else {
ExtendedManagedType extendedManagedType = mainQuery.metamodel.getManagedType(ExtendedManagedType.class, valueClass);
Map.Entry, String> entry = extendedManagedType.getEmbeddableSingularOwner();
boolean singular = true;
if (entry == null) {
singular = false;
entry = extendedManagedType.getEmbeddablePluralOwner();
}
if (entry == null) {
throw new IllegalArgumentException("Unsupported use of embeddable type [" + valueClass + "] for values clause! Use the entity type and fromIdentifiableValues instead or introduce a CTE entity containing just the embeddable to be able to query it!");
}
Class valueHolderEntityClass = entry.getKey().getJavaType();
String valuesLikeAttributeName = entry.getValue();
ExtendedAttribute valuesLikeAttribute = mainQuery.metamodel.getManagedType(ExtendedManagedType.class, valueHolderEntityClass).getAttribute(valuesLikeAttributeName);
prepareFromModification();
joinManager.addRootValues(valueHolderEntityClass, valueClass, alias, valueCount, null, null, false, singular, valuesLikeAttributeName, valuesLikeAttribute, null, null);
}
fromClassExplicitlySet = true;
return (BuilderType) this;
}
public BuilderType fromValues(Class entityBaseClass, String originalAttributeName, String alias, int valueCount) {
ExtendedManagedType extendedManagedType = mainQuery.metamodel.getManagedType(ExtendedManagedType.class, entityBaseClass);
String keyFunction = "key(";
String indexFunction = "index(";
String attributeName = originalAttributeName;
ExtendedAttribute valuesLikeAttribute;
Class elementClass;
boolean index = false;
boolean singular = false;
String valuesLikeClause;
String qualificationExpression;
if (attributeName.regionMatches(true, 0, keyFunction, 0, keyFunction.length())) {
attributeName = attributeName.substring(keyFunction.length(), attributeName.length() - 1);
valuesLikeAttribute = extendedManagedType.getAttribute(attributeName);
index = true;
if (valuesLikeAttribute.getAttributePath().size() > 1) {
ExtendedAttribute superAttr = extendedManagedType.getAttribute(valuesLikeAttribute.getAttributePathString().substring(0, valuesLikeAttribute.getAttributePathString().lastIndexOf('.')));
elementClass = JpaMetamodelUtils.resolveKeyClass(superAttr.getElementClass(), (MapAttribute) valuesLikeAttribute.getAttribute());
} else {
elementClass = JpaMetamodelUtils.resolveKeyClass(entityBaseClass, (MapAttribute) valuesLikeAttribute.getAttribute());
}
valuesLikeClause = "KEY(" + ((EntityType) extendedManagedType.getType()).getName() + "." + attributeName + ")";
qualificationExpression = "KEY";
} else if (attributeName.regionMatches(true, 0, indexFunction, 0, indexFunction.length())) {
attributeName = attributeName.substring(indexFunction.length(), attributeName.length() - 1);
valuesLikeAttribute = extendedManagedType.getAttribute(attributeName);
index = true;
elementClass = Integer.class;
valuesLikeClause = "INDEX(" + ((EntityType) extendedManagedType.getType()).getName() + "." + attributeName + ")";
qualificationExpression = "INDEX";
} else {
valuesLikeAttribute = extendedManagedType.getAttribute(attributeName);
elementClass = valuesLikeAttribute.getElementClass();
valuesLikeClause = ((EntityType) extendedManagedType.getType()).getName() + "." + attributeName;
qualificationExpression = null;
}
if (valuesLikeAttribute.getAttribute() instanceof SingularAttribute) {
singular = true;
if (((SingularAttribute) valuesLikeAttribute.getAttribute()).getType() instanceof BasicType) {
if (valuesLikeAttribute.getColumnTypes().length != 1) {
throw new IllegalArgumentException("Unsupported VALUES clause use with multi-column attribute type " + Arrays.toString(valuesLikeAttribute.getColumnTypes()) + "! Consider creating a synthetic type like a @CTE entity to hold this attribute and use that type via fromIdentifiableValues instead!");
}
return fromValuesLike(entityBaseClass, elementClass, valuesLikeAttribute.getColumnTypes()[0], alias, valueCount, valuesLikeClause, valuesLikeAttribute, true, null);
}
} else if (index) {
Map keyColumnTypes = valuesLikeAttribute.getJoinTable().getKeyColumnTypes();
if (keyColumnTypes.size() != 1) {
throw new IllegalArgumentException("Unsupported VALUES clause use with multi-column attribute type " + keyColumnTypes.values() + "! Consider creating a synthetic type like a @CTE entity to hold this attribute and use that type via fromIdentifiableValues instead!");
}
String columnType = keyColumnTypes.values().iterator().next();
return fromValuesLike(entityBaseClass, elementClass, columnType, alias, valueCount, valuesLikeClause, valuesLikeAttribute, false, qualificationExpression);
} else {
if (((PluralAttribute) valuesLikeAttribute.getAttribute()).getElementType() instanceof BasicType) {
if (valuesLikeAttribute.getColumnTypes().length != 1) {
throw new IllegalArgumentException("Unsupported VALUES clause use with multi-column attribute type " + Arrays.toString(valuesLikeAttribute.getColumnTypes()) + "! Consider creating a synthetic type like a @CTE entity to hold this attribute and use that type via fromIdentifiableValues instead!");
}
return fromValuesLike(entityBaseClass, elementClass, valuesLikeAttribute.getColumnTypes()[0], alias, valueCount, valuesLikeClause, valuesLikeAttribute, false, null);
}
}
return fromValuesLike(entityBaseClass, elementClass, null, alias, valueCount, valuesLikeClause, valuesLikeAttribute, singular, null);
}
private BuilderType fromValuesLike(Class valueHolderEntityClass, Class valueClass, String sqlType, String alias, int valueCount, String valuesLikeClause, ExtendedAttribute valuesLikeAttribute, boolean valueLikeAttributeSingular, String qualificationExpression) {
prepareFromModification();
String castedParameter = sqlType == null ? null : mainQuery.dbmsDialect.cast("?", sqlType);
joinManager.addRootValues(valueHolderEntityClass, valueClass, alias, valueCount, null, castedParameter, false, valueLikeAttributeSingular, valuesLikeAttribute.getAttributePathString(), valuesLikeAttribute, valuesLikeClause, qualificationExpression);
fromClassExplicitlySet = true;
return (BuilderType) this;
}
private BuilderType from(Class clazz, String alias, DbmsModificationState state) {
EntityType type = mainQuery.metamodel.entity(clazz);
return from(type, alias, state);
}
@SuppressWarnings("unchecked")
private BuilderType from(EntityType type, String alias, DbmsModificationState state) {
prepareFromModification();
String finalAlias = joinManager.addRoot(type, alias);
fromClassExplicitlySet = true;
// Handle old and new references
if (state != null) {
mainQuery.assertSupportsAdvancedSql("Illegal use of modification state clause OLD/NEW!");
Class clazz = type.getJavaType();
Map versionEntities = explicitVersionEntities.get(clazz);
if (versionEntities == null) {
versionEntities = new HashMap<>(1);
explicitVersionEntities.put(clazz, versionEntities);
}
versionEntities.put(finalAlias, state);
}
return (BuilderType) this;
}
private void prepareFromModification() {
prepareForModification(ClauseType.JOIN);
if (!fromClassExplicitlySet) {
// When from is explicitly called we have to revert the implicit root
if (joinManager.getRoots().size() > 0) {
joinManager.removeRoot();
}
}
}
public Set getRoots() {
return new LinkedHashSet(joinManager.getRoots());
}
public JoinNode getRoot() {
return joinManager.getRootNodeOrFail("This should never happen. Please report this error!");
}
public JoinNode getFrom(String alias) {
AliasInfo info = aliasManager.getAliasInfo(alias);
if (info == null || !(info instanceof JoinAliasInfo)) {
return null;
}
return ((JoinAliasInfo) info).getJoinNode();
}
public JoinNode getFromByPath(String path) {
if (path == null || path.isEmpty()) {
JoinNode node = joinManager.getRootNodeOrFail("No or multiple query roots, can't find single root!");
return node;
}
PathExpression pathExpression = expressionFactory.createPathExpression(path);
joinManager.implicitJoin(pathExpression, true, true, null, null, new HashSet(), false, false, true, false);
return (JoinNode) pathExpression.getBaseNode();
}
public Path getPath(String path) {
if (path == null || path.isEmpty()) {
JoinNode node = joinManager.getRootNodeOrFail("No or multiple query roots, can't find single root!");
return new SimplePathReference(node, null, node.getType());
}
PathExpression pathExpression = expressionFactory.createPathExpression(path);
joinManager.implicitJoin(pathExpression, true, true, null, null, new HashSet(), false, false, true, false);
return (Path) pathExpression.getPathReference();
}
public boolean isEmpty() {
return joinManager.getRoots().isEmpty()
|| (
!fromClassExplicitlySet
&& joinManager.getRoots().size() == 1
&& joinManager.getRoots().get(0).getNodes().isEmpty()
&& joinManager.getRoots().get(0).getTreatedJoinNodes().isEmpty()
&& joinManager.getRoots().get(0).getEntityJoinNodes().isEmpty()
)
;
}
@SuppressWarnings("unchecked")
public BuilderType setFirstResult(int firstResult) {
this.firstResult = firstResult;
return (BuilderType) this;
}
@SuppressWarnings("unchecked")
public BuilderType setMaxResults(int maxResults) {
this.maxResults = maxResults;
return (BuilderType) this;
}
public int getFirstResult() {
return firstResult;
}
public int getMaxResults() {
return maxResults;
}
public EntityManager getEntityManager() {
return em;
}
public EntityMetamodel getMetamodel() {
return mainQuery.metamodel;
}
@SuppressWarnings("unchecked")
public BuilderType setParameter(String name, Object value) {
parameterManager.satisfyParameter(name, value);
return (BuilderType) this;
}
@SuppressWarnings("unchecked")
public BuilderType setParameter(String name, Calendar value, TemporalType temporalType) {
parameterManager.satisfyParameter(name, value, temporalType);
return (BuilderType) this;
}
@SuppressWarnings("unchecked")
public BuilderType setParameter(String name, Date value, TemporalType temporalType) {
parameterManager.satisfyParameter(name, value, temporalType);
return (BuilderType) this;
}
public boolean containsParameter(String name) {
return parameterManager.containsParameter(name);
}
public boolean isParameterSet(String name) {
return parameterManager.isParameterSet(name);
}
public Parameter getParameter(String name) {
return parameterManager.getParameter(name);
}
public Set> getParameters() {
return parameterManager.getParameters();
}
public Object getParameterValue(String name) {
return parameterManager.getParameterValue(name);
}
@SuppressWarnings("unchecked")
public BuilderType setParameterType(String name, Class type) {
parameterManager.setParameterType(name, type);
return (BuilderType) this;
}
/*
* Select methods
*/
@SuppressWarnings("unchecked")
public BuilderType distinct() {
prepareForModification(ClauseType.SELECT);
selectManager.distinct();
return (BuilderType) this;
}
public CaseWhenStarterBuilder selectCase() {
return selectCase(null);
}
/* CASE (WHEN condition THEN scalarExpression)+ ELSE scalarExpression END */
@SuppressWarnings("unchecked")
public CaseWhenStarterBuilder selectCase(String selectAlias) {
if (selectAlias != null && selectAlias.isEmpty()) {
throw new IllegalArgumentException("selectAlias");
}
prepareForModification(ClauseType.SELECT);
return selectManager.selectCase((BuilderType) this, selectAlias);
}
public SimpleCaseWhenStarterBuilder selectSimpleCase(String expression) {
return selectSimpleCase(expression, null);
}
/* CASE caseOperand (WHEN scalarExpression THEN scalarExpression)+ ELSE scalarExpression END */
@SuppressWarnings("unchecked")
public SimpleCaseWhenStarterBuilder selectSimpleCase(String caseOperandExpression, String selectAlias) {
if (selectAlias != null && selectAlias.isEmpty()) {
throw new IllegalArgumentException("selectAlias");
}
prepareForModification(ClauseType.SELECT);
return selectManager.selectSimpleCase((BuilderType) this, selectAlias, expressionFactory.createSimpleExpression(caseOperandExpression, false));
}
public BuilderType select(String expression) {
return select(expression, null);
}
@SuppressWarnings("unchecked")
public BuilderType select(String expression, String selectAlias) {
Expression expr = expressionFactory.createSimpleExpression(expression, false);
if (selectAlias != null && selectAlias.isEmpty()) {
throw new IllegalArgumentException("selectAlias");
}
verifyBuilderEnded();
prepareForModification(ClauseType.SELECT);
selectManager.select(expr, selectAlias);
if (selectManager.getSelectInfos().size() > 1) {
// TODO: don't know if we should override this here
resultType = (Class) Tuple.class;
}
return (BuilderType) this;
}
public SubqueryInitiator selectSubquery() {
return selectSubquery((String) null);
}
@SuppressWarnings("unchecked")
public SubqueryInitiator selectSubquery(String selectAlias) {
if (selectAlias != null && selectAlias.isEmpty()) {
throw new IllegalArgumentException("selectAlias");
}
verifyBuilderEnded();
prepareForModification(ClauseType.SELECT);
return selectManager.selectSubquery((BuilderType) this, selectAlias);
}
public SubqueryInitiator selectSubquery(String subqueryAlias, String expression) {
return selectSubquery(subqueryAlias, expression, (String) null);
}
@SuppressWarnings("unchecked")
public SubqueryInitiator selectSubquery(String subqueryAlias, String expression, String selectAlias) {
if (selectAlias != null && selectAlias.isEmpty()) {
throw new IllegalArgumentException("selectAlias");
}
if (subqueryAlias == null) {
throw new NullPointerException("subqueryAlias");
}
if (subqueryAlias.isEmpty()) {
throw new IllegalArgumentException("subqueryAlias");
}
if (expression == null) {
throw new NullPointerException("expression");
}
if (!expression.contains(subqueryAlias)) {
throw new IllegalArgumentException("Expression [" + expression + "] does not contain subquery alias [" + subqueryAlias + "]");
}
verifyBuilderEnded();
prepareForModification(ClauseType.SELECT);
return selectManager.selectSubquery((BuilderType) this, subqueryAlias, expressionFactory.createSimpleExpression(expression, false), selectAlias);
}
public MultipleSubqueryInitiator selectSubqueries(String expression) {
return selectSubqueries(null, expression);
}
@SuppressWarnings("unchecked")
public MultipleSubqueryInitiator selectSubqueries(String selectAlias, String expression) {
if (selectAlias != null && selectAlias.isEmpty()) {
throw new IllegalArgumentException("selectAlias");
}
if (expression == null) {
throw new NullPointerException("expression");
}
verifyBuilderEnded();
prepareForModification(ClauseType.SELECT);
return selectManager.selectSubqueries((BuilderType) this, expressionFactory.createSimpleExpression(expression, false), selectAlias);
}
public SubqueryBuilder selectSubquery(FullQueryBuilder criteriaBuilder) {
return selectSubquery(null, criteriaBuilder);
}
@SuppressWarnings("unchecked")
public SubqueryBuilder selectSubquery(String selectAlias, FullQueryBuilder criteriaBuilder) {
if (selectAlias != null && selectAlias.isEmpty()) {
throw new IllegalArgumentException("selectAlias");
}
if (criteriaBuilder == null) {
throw new NullPointerException("criteriaBuilder");
}
verifyBuilderEnded();
prepareForModification(ClauseType.SELECT);
return selectManager.selectSubquery((BuilderType) this, selectAlias, criteriaBuilder);
}
public SubqueryBuilder selectSubquery(String subqueryAlias, String expression, FullQueryBuilder criteriaBuilder) {
return selectSubquery(subqueryAlias, expression, null, criteriaBuilder);
}
@SuppressWarnings("unchecked")
public SubqueryBuilder selectSubquery(String subqueryAlias, String expression, String selectAlias, FullQueryBuilder criteriaBuilder) {
if (selectAlias != null && selectAlias.isEmpty()) {
throw new IllegalArgumentException("selectAlias");
}
if (subqueryAlias == null) {
throw new NullPointerException("subqueryAlias");
}
if (subqueryAlias.isEmpty()) {
throw new IllegalArgumentException("subqueryAlias");
}
if (expression == null) {
throw new NullPointerException("expression");
}
if (criteriaBuilder == null) {
throw new NullPointerException("criteriaBuilder");
}
if (!expression.contains(subqueryAlias)) {
throw new IllegalArgumentException("Expression [" + expression + "] does not contain subquery alias [" + subqueryAlias + "]");
}
verifyBuilderEnded();
prepareForModification(ClauseType.SELECT);
return selectManager.selectSubquery((BuilderType) this, subqueryAlias, expressionFactory.createSimpleExpression(expression, false), selectAlias, criteriaBuilder);
}
/*
* Where methods
*/
public RestrictionBuilder where(String expression) {
prepareForModification(ClauseType.WHERE);
Expression expr = expressionFactory.createSimpleExpression(expression, false);
return whereManager.restrict((BuilderType) this, expr);
}
public CaseWhenStarterBuilder> whereCase() {
prepareForModification(ClauseType.WHERE);
return whereManager.restrictCase((BuilderType) this);
}
public SimpleCaseWhenStarterBuilder> whereSimpleCase(String expression) {
prepareForModification(ClauseType.WHERE);
return whereManager.restrictSimpleCase((BuilderType) this, expressionFactory.createSimpleExpression(expression, false));
}
public WhereOrBuilder whereOr() {
prepareForModification(ClauseType.WHERE);
return whereManager.whereOr(this);
}
@SuppressWarnings("unchecked")
public SubqueryInitiator whereExists() {
prepareForModification(ClauseType.WHERE);
return whereManager.restrictExists((BuilderType) this);
}
@SuppressWarnings("unchecked")
public SubqueryInitiator whereNotExists() {
prepareForModification(ClauseType.WHERE);
return whereManager.restrictNotExists((BuilderType) this);
}
@SuppressWarnings("unchecked")
public SubqueryBuilder whereExists(FullQueryBuilder criteriaBuilder) {
prepareForModification(ClauseType.WHERE);
return whereManager.restrictExists((BuilderType) this, criteriaBuilder);
}
@SuppressWarnings("unchecked")
public SubqueryBuilder whereNotExists(FullQueryBuilder criteriaBuilder) {
prepareForModification(ClauseType.WHERE);
return whereManager.restrictNotExists((BuilderType) this, criteriaBuilder);
}
public SubqueryInitiator> whereSubquery() {
prepareForModification(ClauseType.WHERE);
return whereManager.restrict((BuilderType) this);
}
public SubqueryInitiator> whereSubquery(String subqueryAlias, String expression) {
prepareForModification(ClauseType.WHERE);
return whereManager.restrict((BuilderType) this, subqueryAlias, expression);
}
public MultipleSubqueryInitiator> whereSubqueries(String expression) {
prepareForModification(ClauseType.WHERE);
return whereManager.restrictSubqueries((BuilderType) this, expression);
}
public SubqueryBuilder> whereSubquery(FullQueryBuilder criteriaBuilder) {
prepareForModification(ClauseType.WHERE);
return whereManager.restrict((BuilderType) this, criteriaBuilder);
}
public SubqueryBuilder> whereSubquery(String subqueryAlias, String expression, FullQueryBuilder criteriaBuilder) {
prepareForModification(ClauseType.WHERE);
return whereManager.restrict((BuilderType) this, subqueryAlias, expression, criteriaBuilder);
}
@SuppressWarnings("unchecked")
public BuilderType whereExpression(String expression) {
prepareForModification(ClauseType.WHERE);
Predicate predicate = expressionFactory.createBooleanExpression(expression, false);
whereManager.restrictExpression(predicate);
return (BuilderType) this;
}
@SuppressWarnings("unchecked")
public MultipleSubqueryInitiator whereExpressionSubqueries(String expression) {
prepareForModification(ClauseType.WHERE);
Predicate predicate = expressionFactory.createBooleanExpression(expression, true);
return whereManager.restrictExpressionSubqueries((BuilderType) this, predicate);
}
@SuppressWarnings("unchecked")
public BuilderType setWhereExpression(String expression) {
prepareForModification(ClauseType.WHERE);
Predicate predicate = expressionFactory.createBooleanExpression(expression, false);
whereManager.restrictSetExpression(predicate);
return (BuilderType) this;
}
@SuppressWarnings("unchecked")
public MultipleSubqueryInitiator setWhereExpressionSubqueries(String expression) {
prepareForModification(ClauseType.WHERE);
Predicate predicate = expressionFactory.createBooleanExpression(expression, true);
return whereManager.restrictSetExpressionSubqueries((BuilderType) this, predicate);
}
/*
* Group by methods
*/
@SuppressWarnings("unchecked")
public BuilderType groupBy(String... paths) {
for (String path : paths) {
groupBy(path);
}
return (BuilderType) this;
}
@SuppressWarnings("unchecked")
public BuilderType groupBy(String expression) {
prepareForModification(ClauseType.GROUP_BY);
Expression expr;
if (mainQuery.getQueryConfiguration().isCompatibleModeEnabled()) {
expr = expressionFactory.createPathExpression(expression);
} else {
expr = expressionFactory.createSimpleExpression(expression, false);
}
verifyBuilderEnded();
groupByManager.groupBy(expr);
return (BuilderType) this;
}
/*
* Having methods
*/
public RestrictionBuilder having(String expression) {
prepareForModification(ClauseType.HAVING);
if (groupByManager.isEmpty()) {
throw new IllegalStateException("Having without group by");
}
Expression expr = expressionFactory.createSimpleExpression(expression, false);
return havingManager.restrict((BuilderType) this, expr);
}
public CaseWhenStarterBuilder> havingCase() {
prepareForModification(ClauseType.HAVING);
if (groupByManager.isEmpty()) {
throw new IllegalStateException("Having without group by");
}
return havingManager.restrictCase((BuilderType) this);
}
public SimpleCaseWhenStarterBuilder> havingSimpleCase(String expression) {
prepareForModification(ClauseType.HAVING);
if (groupByManager.isEmpty()) {
throw new IllegalStateException("Having without group by");
}
return havingManager.restrictSimpleCase((BuilderType) this, expressionFactory.createSimpleExpression(expression, false));
}
public HavingOrBuilder havingOr() {
prepareForModification(ClauseType.HAVING);
if (groupByManager.isEmpty()) {
throw new IllegalStateException("Having without group by");
}
return havingManager.havingOr(this);
}
@SuppressWarnings("unchecked")
public SubqueryInitiator havingExists() {
prepareForModification(ClauseType.HAVING);
if (groupByManager.isEmpty()) {
throw new IllegalStateException("Having without group by");
}
return havingManager.restrictExists((BuilderType) this);
}
@SuppressWarnings("unchecked")
public SubqueryInitiator havingNotExists() {
prepareForModification(ClauseType.HAVING);
if (groupByManager.isEmpty()) {
throw new IllegalStateException("Having without group by");
}
return havingManager.restrictNotExists((BuilderType) this);
}
@SuppressWarnings("unchecked")
public SubqueryBuilder havingExists(FullQueryBuilder criteriaBuilder) {
prepareForModification(ClauseType.HAVING);
if (groupByManager.isEmpty()) {
throw new IllegalStateException("Having without group by");
}
return havingManager.restrictExists((BuilderType) this, criteriaBuilder);
}
@SuppressWarnings("unchecked")
public SubqueryBuilder havingNotExists(FullQueryBuilder criteriaBuilder) {
prepareForModification(ClauseType.HAVING);
if (groupByManager.isEmpty()) {
throw new IllegalStateException("Having without group by");
}
return havingManager.restrictNotExists((BuilderType) this, criteriaBuilder);
}
public SubqueryInitiator> havingSubquery() {
prepareForModification(ClauseType.HAVING);
if (groupByManager.isEmpty()) {
throw new IllegalStateException("Having without group by");
}
return havingManager.restrict((BuilderType) this);
}
public SubqueryInitiator> havingSubquery(String subqueryAlias, String expression) {
prepareForModification(ClauseType.HAVING);
if (groupByManager.isEmpty()) {
throw new IllegalStateException("Having without group by");
}
return havingManager.restrict((BuilderType) this, subqueryAlias, expression);
}
public MultipleSubqueryInitiator> havingSubqueries(String expression) {
prepareForModification(ClauseType.HAVING);
if (groupByManager.isEmpty()) {
throw new IllegalStateException("Having without group by");
}
return havingManager.restrictSubqueries((BuilderType) this, expression);
}
public SubqueryBuilder> havingSubquery(FullQueryBuilder criteriaBuilder) {
prepareForModification(ClauseType.HAVING);
if (groupByManager.isEmpty()) {
throw new IllegalStateException("Having without group by");
}
return havingManager.restrict((BuilderType) this, criteriaBuilder);
}
public SubqueryBuilder> havingSubquery(String subqueryAlias, String expression, FullQueryBuilder criteriaBuilder) {
prepareForModification(ClauseType.HAVING);
if (groupByManager.isEmpty()) {
throw new IllegalStateException("Having without group by");
}
return havingManager.restrict((BuilderType) this, subqueryAlias, expression, criteriaBuilder);
}
@SuppressWarnings("unchecked")
public BuilderType havingExpression(String expression) {
prepareForModification(ClauseType.HAVING);
Predicate predicate = expressionFactory.createBooleanExpression(expression, false);
havingManager.restrictExpression(predicate);
return (BuilderType) this;
}
@SuppressWarnings("unchecked")
public MultipleSubqueryInitiator havingExpressionSubqueries(String expression) {
prepareForModification(ClauseType.HAVING);
Predicate predicate = expressionFactory.createBooleanExpression(expression, true);
return havingManager.restrictExpressionSubqueries((BuilderType) this, predicate);
}
@SuppressWarnings("unchecked")
public BuilderType setHavingExpression(String expression) {
prepareForModification(ClauseType.HAVING);
if (groupByManager.isEmpty()) {
throw new IllegalStateException("Having without group by");
}
Predicate predicate = expressionFactory.createBooleanExpression(expression, false);
havingManager.restrictSetExpression(predicate);
return (BuilderType) this;
}
@SuppressWarnings("unchecked")
public MultipleSubqueryInitiator setHavingExpressionSubqueries(String expression) {
prepareForModification(ClauseType.HAVING);
if (groupByManager.isEmpty()) {
throw new IllegalStateException("Having without group by");
}
Predicate predicate = expressionFactory.createBooleanExpression(expression, true);
return havingManager.restrictSetExpressionSubqueries((BuilderType) this, predicate);
}
/*
* Order by methods
*/
public BuilderType orderByDesc(String expression) {
return orderBy(expression, false, false);
}
public BuilderType orderByAsc(String expression) {
return orderBy(expression, true, false);
}
public BuilderType orderByDesc(String expression, boolean nullFirst) {
return orderBy(expression, false, nullFirst);
}
public BuilderType orderByAsc(String expression, boolean nullFirst) {
return orderBy(expression, true, nullFirst);
}
@SuppressWarnings("unchecked")
public BuilderType orderBy(String expression, boolean ascending, boolean nullFirst) {
Expression expr;
if (mainQuery.getQueryConfiguration().isCompatibleModeEnabled()) {
expr = expressionFactory.createPathExpression(expression);
} else {
expr = expressionFactory.createSimpleExpression(expression, false);
}
orderBy(expr, ascending, nullFirst);
return (BuilderType) this;
}
private void orderBy(Expression expression, boolean ascending, boolean nullFirst) {
prepareForModification(ClauseType.ORDER_BY);
verifyBuilderEnded();
orderByManager.orderBy(expression, ascending, nullFirst);
}
/*
* Window methods
*/
public WindowBuilder window(String name) {
prepareForModification(ClauseType.WINDOW);
verifyBuilderEnded();
return windowManager.window(name, (BuilderType) this);
}
protected void verifySetBuilderEnded() {
if (finalSetOperationBuilder != null) {
if (!setOperationEnded) {
throw new IllegalStateException("Set operation builder not properly ended!");
}
}
}
protected void verifyBuilderEnded() {
if (isMainQuery) {
mainQuery.cteManager.verifyBuilderEnded();
}
whereManager.verifyBuilderEnded();
keysetManager.verifyBuilderEnded();
havingManager.verifyBuilderEnded();
selectManager.verifyBuilderEnded();
joinManager.verifyBuilderEnded();
windowManager.verifyBuilderEnded();
}
/*
* Join methods
*/
public BuilderType innerJoin(String path, String alias) {
return join(path, alias, JoinType.INNER);
}
public BuilderType innerJoinDefault(String path, String alias) {
return joinDefault(path, alias, JoinType.INNER);
}
public BuilderType leftJoin(String path, String alias) {
return join(path, alias, JoinType.LEFT);
}
public BuilderType leftJoinDefault(String path, String alias) {
return joinDefault(path, alias, JoinType.LEFT);
}
public BuilderType rightJoin(String path, String alias) {
return join(path, alias, JoinType.RIGHT);
}
public BuilderType rightJoinDefault(String path, String alias) {
return joinDefault(path, alias, JoinType.RIGHT);
}
@SuppressWarnings("unchecked")
public BuilderType join(String path, String alias, JoinType type) {
prepareForModification(ClauseType.JOIN);
checkJoinPreconditions(path, alias, type);
joinManager.join(path, alias, type, false, false);
return (BuilderType) this;
}
@SuppressWarnings("unchecked")
public BuilderType joinDefault(String path, String alias, JoinType type) {
prepareForModification(ClauseType.JOIN);
checkJoinPreconditions(path, alias, type);
joinManager.join(path, alias, type, false, true);
return (BuilderType) this;
}
@SuppressWarnings("unchecked")
public JoinOnBuilder joinOn(String path, String alias, JoinType type) {
prepareForModification(ClauseType.JOIN);
checkJoinPreconditions(path, alias, type);
return joinManager.joinOn((BuilderType) this, path, alias, type, false);
}
@SuppressWarnings("unchecked")
public JoinOnBuilder joinDefaultOn(String path, String alias, JoinType type) {
prepareForModification(ClauseType.JOIN);
checkJoinPreconditions(path, alias, type);
return joinManager.joinOn((BuilderType) this, path, alias, type, true);
}
@SuppressWarnings("unchecked")
public JoinOnBuilder joinOn(Class clazz, String alias, JoinType type) {
return joinOn(joinManager.getRootNodeOrFail("An explicit base join node is required when multiple root nodes are used!").getAlias(), clazz, alias, type);
}
@SuppressWarnings("unchecked")
public JoinOnBuilder joinOn(String base, Class entityClass, String alias, JoinType type) {
prepareForModification(ClauseType.JOIN);
checkJoinPreconditions(base, alias, type);
if (entityClass == null) {
throw new NullPointerException("entityClass");
}
return joinManager.joinOn((BuilderType) this, base, entityClass, alias, type);
}
@SuppressWarnings("unchecked")
public JoinOnBuilder joinOn(EntityType entityType, String alias, JoinType type) {
return joinOn(joinManager.getRootNodeOrFail("An explicit base join node is required when multiple root nodes are used!").getAlias(), entityType, alias, type);
}
@SuppressWarnings("unchecked")
public JoinOnBuilder joinOn(String base, EntityType entityType, String alias, JoinType type) {
prepareForModification(ClauseType.JOIN);
checkJoinPreconditions(base, alias, type);
if (entityType == null) {
throw new NullPointerException("entityType");
}
return joinManager.joinOn((BuilderType) this, base, entityType, alias, type);
}
public JoinOnBuilder innerJoinOn(String path, String alias) {
return joinOn(path, alias, JoinType.INNER);
}
public JoinOnBuilder innerJoinDefaultOn(String path, String alias) {
return joinDefaultOn(path, alias, JoinType.INNER);
}
public JoinOnBuilder innerJoinOn(Class clazz, String alias) {
return joinOn(clazz, alias, JoinType.INNER);
}
public JoinOnBuilder innerJoinOn(String base, Class clazz, String alias) {
return joinOn(base, clazz, alias, JoinType.INNER);
}
public JoinOnBuilder innerJoinOn(EntityType entityType, String alias) {
return joinOn(entityType, alias, JoinType.INNER);
}
public JoinOnBuilder innerJoinOn(String base, EntityType entityType, String alias) {
return joinOn(base, entityType, alias, JoinType.INNER);
}
public JoinOnBuilder leftJoinOn(String path, String alias) {
return joinOn(path, alias, JoinType.LEFT);
}
public JoinOnBuilder leftJoinDefaultOn(String path, String alias) {
return joinDefaultOn(path, alias, JoinType.LEFT);
}
public JoinOnBuilder leftJoinOn(EntityType entityType, String alias) {
return joinOn(entityType, alias, JoinType.LEFT);
}
public JoinOnBuilder leftJoinOn(String base, EntityType entityType, String alias) {
return joinOn(base, entityType, alias, JoinType.LEFT);
}
public JoinOnBuilder leftJoinOn(Class clazz, String alias) {
return joinOn(clazz, alias, JoinType.LEFT);
}
public JoinOnBuilder leftJoinOn(String base, Class clazz, String alias) {
return joinOn(base, clazz, alias, JoinType.LEFT);
}
public JoinOnBuilder rightJoinOn(String path, String alias) {
return joinOn(path, alias, JoinType.RIGHT);
}
public JoinOnBuilder rightJoinDefaultOn(String path, String alias) {
return joinDefaultOn(path, alias, JoinType.RIGHT);
}
public JoinOnBuilder rightJoinOn(Class clazz, String alias) {
return joinOn(clazz, alias, JoinType.RIGHT);
}
public JoinOnBuilder rightJoinOn(String base, Class clazz, String alias) {
return joinOn(base, clazz, alias, JoinType.RIGHT);
}
public JoinOnBuilder rightJoinOn(EntityType entityType, String alias) {
return joinOn(entityType, alias, JoinType.RIGHT);
}
public JoinOnBuilder rightJoinOn(String base, EntityType entityType, String alias) {
return joinOn(base, entityType, alias, JoinType.RIGHT);
}
private void checkJoinPreconditions(String path, String alias, JoinType type) {
if (path == null) {
throw new NullPointerException("path");
}
if (alias == null) {
throw new NullPointerException("alias");
}
if (type == null) {
throw new NullPointerException("type");
}
if (alias.isEmpty()) {
throw new IllegalArgumentException("Empty alias");
}
verifyBuilderEnded();
}
protected boolean isJoinRequiredForSelect() {
return true;
}
protected JoinVisitor applyImplicitJoins(JoinVisitor parentVisitor) {
if (implicitJoinsApplied) {
return null;
}
// The first thing we need to do, is reorder values clauses without joins to the end of the from clause roots
// This is an ugly integration detail, but to me, this seems to be the only way to support the values clause in all situations
// We put the values clauses without joins to the end because we will add synthetic predicates to the where clause
// This is necessary in order to have a single query that has all parameters which will simplify a lot
// If a values clause has a join, we will inject the synthetic predicate into that joins on clause which will preserve the correct order
// We just have to hope that Hibernate will not reorder roots when translating to SQL, if it does, this will break..
// NOTE: If it turns out to be problematic, I can imagine introducing a synthetic SELECT item that is removed in the end for this purpose
joinManager.reorderSimpleValuesClauses();
final JoinVisitor joinVisitor = new JoinVisitor(mainQuery, windowManager, parentVisitor, joinManager, parameterManager, !mainQuery.jpaProvider.supportsSingleValuedAssociationIdExpressions());
joinVisitor.setFromClause(ClauseType.JOIN);
joinManager.acceptVisitor(joinVisitor);
// carry out implicit joins
joinVisitor.setFromClause(ClauseType.SELECT);
// There might be clauses for which joins are not required
joinVisitor.setJoinRequired(isJoinRequiredForSelect());
selectManager.acceptVisitor((SelectInfoVisitor) joinVisitor);
joinVisitor.setJoinRequired(true);
// Only the main query has fetch owners
if (isMainQuery) {
StringBuilder sb = null;
Set fetchOwners = selectManager.collectFetchOwners();
nodesToFetch = new HashSet<>();
// Add all parents of the fetchable nodes to nodesToFetch until the fetch owner is reached
// If we reach a root before a fetch owner, the fetch owner is missing
final List fetchableNodes = joinVisitor.getFetchableNodes();
for (int i = 0; i < fetchableNodes.size(); i++) {
JoinNode fetchableNode = fetchableNodes.get(i);
while (!fetchOwners.contains(fetchableNode)) {
nodesToFetch.add(fetchableNode);
if (fetchableNode.getParent() == null) {
if (sb == null) {
sb = new StringBuilder();
sb.append("Some join nodes specified fetch joining but their fetch owners weren't included in the select clause! Missing fetch owners: [");
} else {
sb.append(", ");
}
sb.append(fetchableNode.getAlias());
break;
}
fetchableNode = fetchableNode.getParent();
// We don't care about treated nodes specifically when fetching as they aren't declarable directly, but only the "main" node
if (fetchableNode.isTreatedJoinNode()) {
fetchableNode = ((TreatedJoinAliasInfo) fetchableNode.getAliasInfo()).getTreatedJoinNode();
}
}
}
if (sb != null) {
sb.append("]");
throw new IllegalStateException(sb.toString());
}
} else {
nodesToFetch = Collections.emptySet();
}
joinVisitor.setFromClause(ClauseType.WHERE);
whereManager.acceptVisitor(joinVisitor);
joinVisitor.setFromClause(ClauseType.GROUP_BY);
groupByManager.acceptVisitor(joinVisitor);
joinVisitor.setFromClause(ClauseType.HAVING);
havingManager.acceptVisitor(joinVisitor);
joinVisitor.setJoinWithObjectLeafAllowed(false);
joinVisitor.setFromClause(ClauseType.ORDER_BY);
orderByManager.acceptVisitor(joinVisitor);
joinVisitor.setJoinWithObjectLeafAllowed(true);
// No need to implicit join again if no mutation occurs
implicitJoinsApplied = true;
return joinVisitor;
}
void inlineSelectAlias(String selectAlias, Expression expression) {
final AliasReplacementVisitor aliasReplacementVisitor = new AliasReplacementVisitor(expression, selectAlias);
ExpressionModifierVisitor expressionModifierVisitor = new ExpressionModifierVisitor() {
@Override
public void visit(ExpressionModifier expressionModifier, ClauseType clauseType) {
Expression expr = expressionModifier.get();
Expression newExpr = expr.accept(aliasReplacementVisitor);
if (expr != newExpr) {
expressionModifier.set(newExpr);
}
}
};
joinManager.apply(expressionModifierVisitor);
selectManager.apply(expressionModifierVisitor);
whereManager.apply(expressionModifierVisitor);
havingManager.apply(expressionModifierVisitor);
groupByManager.apply(expressionModifierVisitor);
orderByManager.apply(expressionModifierVisitor);
}
protected void implicitJoinWhereClause() {
final JoinVisitor joinVisitor = new JoinVisitor(mainQuery, windowManager, null, joinManager, parameterManager, !mainQuery.jpaProvider.supportsSingleValuedAssociationIdExpressions());
joinVisitor.setJoinRequired(true);
joinVisitor.setFromClause(ClauseType.WHERE);
whereManager.acceptVisitor(joinVisitor);
}
protected void collectParameters() {
ParameterRegistrationVisitor parameterRegistrationVisitor = parameterManager.getParameterRegistrationVisitor();
ClauseType oldClauseType = parameterRegistrationVisitor.getClauseType();
AbstractCommonQueryBuilder oldQueryBuilder = parameterRegistrationVisitor.getQueryBuilder();
try {
parameterRegistrationVisitor.setQueryBuilder(this);
parameterRegistrationVisitor.setClauseType(ClauseType.SELECT);
selectManager.acceptVisitor(parameterRegistrationVisitor);
parameterRegistrationVisitor.setClauseType(ClauseType.JOIN);
joinManager.acceptVisitor(new OnClauseJoinNodeVisitor(parameterRegistrationVisitor));
parameterRegistrationVisitor.setClauseType(ClauseType.WHERE);
whereManager.acceptVisitor(parameterRegistrationVisitor);
parameterRegistrationVisitor.setClauseType(ClauseType.GROUP_BY);
groupByManager.acceptVisitor(parameterRegistrationVisitor);
parameterRegistrationVisitor.setClauseType(ClauseType.HAVING);
havingManager.acceptVisitor(parameterRegistrationVisitor);
parameterRegistrationVisitor.setClauseType(ClauseType.ORDER_BY);
orderByManager.acceptVisitor(parameterRegistrationVisitor);
} finally {
parameterRegistrationVisitor.setClauseType(oldClauseType);
parameterRegistrationVisitor.setQueryBuilder(oldQueryBuilder);
}
}
protected void applyVisitor(VisitorAdapter expressionVisitor) {
selectManager.acceptVisitor(expressionVisitor);
joinManager.acceptVisitor(new OnClauseJoinNodeVisitor(expressionVisitor));
whereManager.acceptVisitor(expressionVisitor);
groupByManager.acceptVisitor(expressionVisitor);
havingManager.acceptVisitor(expressionVisitor);
orderByManager.acceptVisitor(expressionVisitor);
}
public void applyExpressionTransformersAndBuildGroupByClauses(boolean addsGroupBy, JoinVisitor joinVisitor) {
groupByManager.resetCollected();
groupByManager.collectGroupByClauses(joinVisitor);
orderByManager.splitEmbeddables(joinVisitor);
int size = transformerGroups.size();
for (int i = 0; i < size; i++) {
@SuppressWarnings("unchecked")
ExpressionTransformerGroup transformerGroup = (ExpressionTransformerGroup) transformerGroups.get(i);
transformerGroup.applyExpressionTransformer(joinManager);
transformerGroup.applyExpressionTransformer(selectManager);
transformerGroup.applyExpressionTransformer(whereManager);
transformerGroup.applyExpressionTransformer(groupByManager);
transformerGroup.applyExpressionTransformer(havingManager);
transformerGroup.applyExpressionTransformer(orderByManager);
transformerGroup.afterTransformationGroup();
}
// After all transformations are done, we can finally check if aggregations are used
hasGroupBy = groupByManager.hasCollectedGroupByClauses();
hasGroupBy = hasGroupBy || Boolean.TRUE.equals(selectManager.acceptVisitor(AggregateDetectionVisitor.INSTANCE, true));
hasGroupBy = hasGroupBy || Boolean.TRUE.equals(joinManager.acceptVisitor(AggregateDetectionVisitor.INSTANCE, true));
hasGroupBy = hasGroupBy || Boolean.TRUE.equals(whereManager.acceptVisitor(AggregateDetectionVisitor.INSTANCE));
hasGroupBy = hasGroupBy || Boolean.TRUE.equals(orderByManager.acceptVisitor(AggregateDetectionVisitor.INSTANCE, true));
hasGroupBy = hasGroupBy || Boolean.TRUE.equals(havingManager.acceptVisitor(AggregateDetectionVisitor.INSTANCE));
if (hasGroupBy || addsGroupBy) {
if (mainQuery.getQueryConfiguration().isImplicitGroupByFromSelectEnabled()) {
selectManager.buildGroupByClauses(cbf.getMetamodel(), groupByManager, hasGroupBy, joinVisitor);
}
if (mainQuery.getQueryConfiguration().isImplicitGroupByFromHavingEnabled()) {
havingManager.buildGroupByClauses(groupByManager, hasGroupBy, joinVisitor);
}
if (mainQuery.getQueryConfiguration().isImplicitGroupByFromOrderByEnabled()) {
orderByManager.buildGroupByClauses(groupByManager, hasGroupBy, joinVisitor);
}
}
for (int i = 0; i < size; i++) {
ExpressionTransformerGroup transformerGroup = transformerGroups.get(i);
transformerGroup.afterAllTransformations();
}
}
public Class getResultType() {
return resultType;
}
public String getQueryString() {
prepareAndCheck();
return getExternalQueryString();
}
protected String getBaseQueryStringWithCheck() {
prepareAndCheck();
return getBaseQueryString();
}
protected final TypedQuery getTypedQueryForFinalOperationBuilder() {
try {
checkSetBuilderEnded = false;
return getTypedQuery();
} finally {
checkSetBuilderEnded = true;
}
}
protected TypedQuery getTypedQuery() {
// NOTE: This must happen first because it generates implicit joins
String baseQueryString = getBaseQueryStringWithCheck();
// We can only use the query directly if we have no ctes, entity functions or hibernate bugs
Set keyRestrictedLeftJoins = joinManager.getKeyRestrictedLeftJoins();
final boolean needsSqlReplacement = isMainQuery && mainQuery.cteManager.hasCtes() || joinManager.hasEntityFunctions() || !keyRestrictedLeftJoins.isEmpty() || !isMainQuery && hasLimit();
if (!needsSqlReplacement) {
TypedQuery baseQuery = getTypedQuery(baseQueryString);
parameterManager.parameterizeQuery(baseQuery);
return baseQuery;
}
TypedQuery baseQuery = getTypedQuery(baseQueryString);
Set parameterListNames = parameterManager.getParameterListNames(baseQuery);
String limit = null;
String offset = null;
// The main query will handle that separately
if (!isMainQuery) {
if (firstResult != 0) {
offset = Integer.toString(firstResult);
}
if (maxResults != Integer.MAX_VALUE) {
limit = Integer.toString(maxResults);
}
}
List keyRestrictedLeftJoinAliases = getKeyRestrictedLeftJoinAliases(baseQuery, keyRestrictedLeftJoins, Collections.EMPTY_SET);
List entityFunctionNodes = getEntityFunctionNodes(baseQuery);
boolean shouldRenderCteNodes = renderCteNodes(false);
List ctes = shouldRenderCteNodes ? getCteNodes(false) : Collections.EMPTY_LIST;
QuerySpecification querySpecification = new CustomQuerySpecification(
this, baseQuery, parameterManager.getParameters(), parameterListNames, limit, offset, keyRestrictedLeftJoinAliases, entityFunctionNodes, mainQuery.cteManager.isRecursive(), ctes, shouldRenderCteNodes
);
TypedQuery query = new CustomSQLTypedQuery(
querySpecification,
baseQuery,
parameterManager.getTransformers(),
parameterManager.getValuesParameters(),
parameterManager.getValuesBinders()
);
// The main query will use the native mechanism for limit/offset
if (isMainQuery) {
if (firstResult != 0) {
query.setFirstResult(firstResult);
}
if (maxResults != Integer.MAX_VALUE) {
query.setMaxResults(maxResults);
}
}
parameterManager.parameterizeQuery(query);
return applyObjectBuilder(query);
}
protected List getKeyRestrictedLeftJoinAliases(Query baseQuery, Set keyRestrictedLeftJoins, Set clauseExclusions) {
List keyRestrictedLeftJoinAliases = new ArrayList();
if (!keyRestrictedLeftJoins.isEmpty()) {
for (JoinNode node : keyRestrictedLeftJoins) {
if (!clauseExclusions.isEmpty() && clauseExclusions.containsAll(node.getClauseDependencies()) && !node.isCardinalityMandatory()) {
continue;
}
// The alias of the target entity table
String sqlAlias = cbf.getExtendedQuerySupport().getSqlAlias(em, baseQuery, node.getAliasInfo().getAlias());
keyRestrictedLeftJoinAliases.add(sqlAlias);
}
}
return keyRestrictedLeftJoinAliases;
}
protected List getEntityFunctionNodes(Query baseQuery) {
List entityFunctionNodes = new ArrayList();
ValuesStrategy strategy = mainQuery.dbmsDialect.getValuesStrategy();
String dummyTable = mainQuery.dbmsDialect.getDummyTable();
for (JoinNode node : joinManager.getEntityFunctionNodes()) {
Class clazz = node.getInternalEntityType().getJavaType();
String valueClazzAttributeName = node.getValuesLikeAttribute();
int valueCount = node.getValueCount();
boolean identifiableReference = node.getNodeType() instanceof EntityType && node.getValuesIdNames() != null;
String rootAlias = node.getAlias();
String castedParameter = node.getValuesCastedParameter();
String[] attributes = node.getValuesAttributes();
// We construct an example query representing the values clause with a SELECT clause that selects the fields in the right order which we need to construct SQL
// that uses proper aliases and filters null values which are there in the first place to pad up parameters in case we don't reach the desired value count
StringBuilder valuesSb = new StringBuilder(20 + valueCount * attributes.length * 3);
Query valuesExampleQuery = getValuesExampleQuery(clazz, valueCount, identifiableReference, valueClazzAttributeName, rootAlias, castedParameter, attributes, valuesSb, strategy, dummyTable, node);
String exampleQuerySql = mainQuery.cbf.getExtendedQuerySupport().getSql(mainQuery.em, valuesExampleQuery);
String exampleQuerySqlAlias = mainQuery.cbf.getExtendedQuerySupport().getSqlAlias(mainQuery.em, valuesExampleQuery, "e");
String exampleQueryCollectionSqlAlias = null;
if (!node.isValueClazzAttributeSingular()) {
exampleQueryCollectionSqlAlias = mainQuery.cbf.getExtendedQuerySupport().getSqlAlias(mainQuery.em, valuesExampleQuery, node.getValueClazzAlias("e_"));
}
StringBuilder whereClauseSb = new StringBuilder(exampleQuerySql.length());
String filterNullsTableAlias = "fltr_nulls_tbl_als_";
String valuesAliases = getValuesAliases(exampleQuerySqlAlias, attributes.length, exampleQuerySql, whereClauseSb, filterNullsTableAlias, strategy, dummyTable);
if (strategy == ValuesStrategy.SELECT_VALUES) {
valuesSb.insert(0, valuesAliases);
valuesSb.append(')');
valuesAliases = null;
} else if (strategy == ValuesStrategy.SELECT_UNION) {
valuesSb.insert(0, valuesAliases);
mainQuery.dbmsDialect.appendExtendedSql(valuesSb, DbmsStatementType.SELECT, true, true, null, Integer.toString(valueCount + 1), "1", null, null);
valuesSb.append(')');
valuesAliases = null;
}
boolean filterNulls = mainQuery.getQueryConfiguration().isValuesClauseFilterNullsEnabled();
if (filterNulls) {
valuesSb.insert(0, "(select * from ");
valuesSb.append(' ');
valuesSb.append(filterNullsTableAlias);
if (valuesAliases != null) {
valuesSb.append(valuesAliases);
valuesAliases = null;
}
valuesSb.append(whereClauseSb);
valuesSb.append(')');
}
String valuesClause = valuesSb.toString();
String valuesTableSqlAlias = exampleQuerySqlAlias;
String valuesTableJoin = null;
String pluralCollectionTableAlias = null;
String pluralTableAlias = null;
String syntheticPredicate = exampleQuerySql.substring(SqlUtils.indexOfWhere(exampleQuerySql) + " where ".length());
if (baseQuery != null) {
valuesTableSqlAlias = cbf.getExtendedQuerySupport().getSqlAlias(em, baseQuery, node.getAlias());
syntheticPredicate = syntheticPredicate.replace(exampleQuerySqlAlias, valuesTableSqlAlias);
if (exampleQueryCollectionSqlAlias != null) {
pluralTableAlias = cbf.getExtendedQuerySupport().getSqlAlias(em, baseQuery, node.getValueClazzAlias(node.getAlias() + "_"));
syntheticPredicate = syntheticPredicate.replace(exampleQueryCollectionSqlAlias, pluralTableAlias);
String baseQuerySql = cbf.getExtendedQuerySupport().getSql(em, baseQuery);
int[] indexRange = SqlUtils.indexOfFullJoin(baseQuerySql, pluralTableAlias);
String baseTableAlias = " " + valuesTableSqlAlias + " ";
int baseTableAliasIndex = baseQuerySql.indexOf(baseTableAlias);
int fullJoinStartIndex = baseTableAliasIndex + baseTableAlias.length();
if (fullJoinStartIndex != indexRange[0]) {
// TODO: find out pluralCollectionTableAlias
String onClause = " on ";
int onClauseIndex = baseQuerySql.indexOf(onClause, fullJoinStartIndex);
int[] collectionTableIndexRange = SqlUtils.rtrimBackwardsToFirstWhitespace(baseQuerySql, onClauseIndex);
pluralCollectionTableAlias = baseQuerySql.substring(collectionTableIndexRange[0], collectionTableIndexRange[1]);
}
valuesTableJoin = baseQuerySql.substring(fullJoinStartIndex, indexRange[1]);
}
}
entityFunctionNodes.add(new EntityFunctionNode(valuesClause, valuesAliases, node.getInternalEntityType().getName(), valuesTableSqlAlias, pluralCollectionTableAlias, pluralTableAlias, valuesTableJoin, syntheticPredicate));
}
return entityFunctionNodes;
}
private String getValuesAliases(String tableAlias, int attributeCount, String exampleQuerySql, StringBuilder whereClauseSb, String filterNullsTableAlias, ValuesStrategy strategy, String dummyTable) {
int startIndex = SqlUtils.indexOfSelect(exampleQuerySql);
int endIndex = exampleQuerySql.indexOf(" from ");
StringBuilder sb;
if (strategy == ValuesStrategy.VALUES) {
sb = new StringBuilder((endIndex - startIndex) - (tableAlias.length() + 3) * attributeCount);
sb.append('(');
} else if (strategy == ValuesStrategy.SELECT_VALUES) {
sb = new StringBuilder(endIndex - startIndex);
sb.append("(select ");
} else if (strategy == ValuesStrategy.SELECT_UNION) {
sb = new StringBuilder((endIndex - startIndex) - (tableAlias.length() + 3) * attributeCount);
sb.append("(select ");
} else {
throw new IllegalArgumentException("Unsupported values strategy: " + strategy);
}
whereClauseSb.append(" where");
String[] columnNames = SqlUtils.getSelectItemColumns(exampleQuerySql, startIndex);
for (int i = 0; i < attributeCount; i++) {
whereClauseSb.append(' ');
if (i > 0) {
whereClauseSb.append("or ");
}
whereClauseSb.append(filterNullsTableAlias);
whereClauseSb.append('.');
whereClauseSb.append(columnNames[i]);
whereClauseSb.append(" is not null");
if (strategy == ValuesStrategy.SELECT_VALUES) {
// TODO: This naming is actually H2 specific
sb.append('c');
sb.append(i + 1);
sb.append(' ');
} else if (strategy == ValuesStrategy.SELECT_UNION) {
sb.append("null as ");
}
sb.append(columnNames[i]);
sb.append(',');
}
if (strategy == ValuesStrategy.VALUES) {
sb.setCharAt(sb.length() - 1, ')');
} else if (strategy == ValuesStrategy.SELECT_VALUES) {
sb.setCharAt(sb.length() - 1, ' ');
sb.append(" from ");
} else if (strategy == ValuesStrategy.SELECT_UNION) {
sb.setCharAt(sb.length() - 1, ' ');
if (dummyTable != null) {
sb.append(" from ");
sb.append(dummyTable);
}
}
return sb.toString();
}
private Query getValuesExampleQuery(Class clazz, int valueCount, boolean identifiableReference, String valueClazzAttributeName, String prefix, String castedParameter, String[] attributes, StringBuilder valuesSb, ValuesStrategy strategy, String dummyTable, JoinNode valuesNode) {
String[] attributeParameter = new String[attributes.length];
// This size estimation roughly assumes a maximum attribute name length of 15
StringBuilder sb = new StringBuilder(50 + valueCount * prefix.length() * attributes.length * 50);
sb.append("SELECT ");
if (clazz == ValuesEntity.class) {
sb.append("e.");
attributeParameter[0] = mainQuery.dbmsDialect.needsCastParameters() ? castedParameter : "?";
sb.append(attributes[0]);
sb.append(',');
} else {
Map mapping = mainQuery.metamodel.getManagedType(ExtendedManagedType.class, clazz).getAttributes();
StringBuilder paramBuilder = new StringBuilder();
for (int i = 0; i < attributes.length; i++) {
ExtendedAttribute entry;
if (valuesNode.isValueClazzSimpleValue()) {
entry = mapping.get(valueClazzAttributeName);
} else {
entry = mapping.get(attributes[i]);
}
Attribute attribute = entry.getAttribute();
String[] columnTypes;
if (valuesNode.getQualificationExpression() == null) {
columnTypes = entry.getColumnTypes();
} else {
Collection types = entry.getJoinTable().getKeyColumnTypes().values();
columnTypes = types.toArray(new String[types.size()]);
}
attributeParameter[i] = getCastedParameters(paramBuilder, mainQuery.dbmsDialect, columnTypes);
// When the class for which we want a VALUES clause has *ToOne relations, we need to put their ids into the select
// otherwise we would fetch all of the types attributes, but the VALUES clause can only ever contain the id
if (attribute.getPersistentAttributeType() != Attribute.PersistentAttributeType.BASIC &&
attribute.getPersistentAttributeType() != Attribute.PersistentAttributeType.EMBEDDED &&
valuesNode.getQualificationExpression() == null) {
ManagedType managedAttributeType = mainQuery.metamodel.getManagedType(entry.getElementClass());
if (managedAttributeType == null || mainQuery.jpaProvider.needsElementCollectionIdCutoff() && valuesNode.getValuesLikeAttribute() != null && mapping.get(valuesNode.getValuesLikeAttribute()).getAttribute().getPersistentAttributeType() == Attribute.PersistentAttributeType.ELEMENT_COLLECTION) {
sb.append("e");
if (valuesNode.isValueClazzAttributeSingular()) {
sb.append('.');
sb.append(attributes[i]);
} else {
sb.append('_');
sb.append(valueClazzAttributeName.replace('.', '_'));
sb.append(attributes[i], valueClazzAttributeName.length(), attributes[i].length());
}
} else {
for (Attribute attributeTypeIdAttribute : JpaMetamodelUtils.getIdAttributes((IdentifiableType) managedAttributeType)) {
sb.append("e");
if (valuesNode.isValueClazzAttributeSingular()) {
sb.append('.');
sb.append(attributes[i]);
} else {
sb.append('_');
sb.append(valueClazzAttributeName.replace('.', '_'));
sb.append(attributes[i], valueClazzAttributeName.length(), attributes[i].length());
}
sb.append('.');
sb.append(attributeTypeIdAttribute.getName());
}
}
} else {
if (valuesNode.getQualificationExpression() != null) {
sb.append(valuesNode.getQualificationExpression()).append('(');
}
sb.append("e");
if (valuesNode.isValueClazzAttributeSingular()) {
sb.append('.');
sb.append(attributes[i]);
} else {
sb.append('_');
sb.append(valueClazzAttributeName.replace('.', '_'));
sb.append(attributes[i], valueClazzAttributeName.length(), attributes[i].length());
}
if (valuesNode.getQualificationExpression() != null) {
sb.append('_').append(valuesNode.getQualificationExpression().toLowerCase()).append(')');
}
}
sb.append(',');
}
}
sb.setCharAt(sb.length() - 1, ' ');
sb.append("FROM ");
sb.append(clazz.getName());
sb.append(" e");
if (!valuesNode.isValueClazzAttributeSingular()) {
sb.append(" LEFT JOIN e.");
if (valuesNode.isValueClazzSimpleValue()) {
sb.append(valueClazzAttributeName);
} else {
sb.append(valuesNode.getValuesLikeAttribute());
}
sb.append(" e_");
if (valuesNode.isValueClazzSimpleValue()) {
sb.append(valueClazzAttributeName.replace('.', '_'));
} else {
sb.append(valuesNode.getValuesLikeAttribute().replace('.', '_'));
}
if (valuesNode.getQualificationExpression() != null) {
sb.append('_').append(valuesNode.getQualificationExpression().toLowerCase());
}
}
sb.append(" WHERE ");
joinManager.renderValuesClausePredicate(sb, valuesNode, "e", false);
if (strategy == ValuesStrategy.SELECT_VALUES || strategy == ValuesStrategy.VALUES) {
valuesSb.append("(VALUES ");
} else if (strategy == ValuesStrategy.SELECT_UNION) {
// Nothing to do here
} else {
throw new IllegalArgumentException("Unsupported values strategy: " + strategy);
}
for (int i = 0; i < valueCount; i++) {
if (strategy == ValuesStrategy.SELECT_UNION) {
valuesSb.append(" union all select ");
} else {
valuesSb.append('(');
}
for (int j = 0; j < attributes.length; j++) {
valuesSb.append(attributeParameter[j]);
valuesSb.append(',');
}
if (strategy == ValuesStrategy.SELECT_UNION) {
valuesSb.setCharAt(valuesSb.length() - 1, ' ');
if (dummyTable != null) {
valuesSb.append("from ");
valuesSb.append(dummyTable);
valuesSb.append(' ');
}
} else {
valuesSb.setCharAt(valuesSb.length() - 1, ')');
valuesSb.append(',');
}
}
if (strategy == ValuesStrategy.SELECT_UNION) {
valuesSb.setCharAt(valuesSb.length() - 1, ' ');
} else {
valuesSb.setCharAt(valuesSb.length() - 1, ')');
}
String exampleQueryString = sb.toString();
return mainQuery.em.createQuery(exampleQueryString);
}
private static String getCastedParameters(StringBuilder sb, DbmsDialect dbmsDialect, String[] types) {
sb.setLength(0);
if (dbmsDialect.needsCastParameters()) {
for (int i = 0; i < types.length; i++) {
sb.append(dbmsDialect.cast("?", types[i]));
sb.append(',');
}
} else {
for (int i = 0; i < types.length; i++) {
sb.append("?,");
}
}
return sb.substring(0, sb.length() - 1);
}
protected boolean renderCteNodes(boolean isSubquery) {
return isMainQuery && !isSubquery;
}
protected List getCteNodes(boolean isSubquery) {
List cteNodes = new ArrayList();
// NOTE: Delete statements could cause CTEs to be generated for the cascading deletes
if (!isMainQuery || isSubquery || !mainQuery.dbmsDialect.supportsWithClause() || !mainQuery.cteManager.hasCtes() && statementType != DbmsStatementType.DELETE || statementType != DbmsStatementType.SELECT && !mainQuery.dbmsDialect.supportsWithClauseInModificationQuery()) {
return cteNodes;
}
StringBuilder sb = new StringBuilder();
for (CTEInfo cteInfo : mainQuery.cteManager.getCtes()) {
// Build queries and add as participating queries
Map modificationStates = cteInfo.nonRecursiveCriteriaBuilder.getModificationStates(explicitVersionEntities);
Query nonRecursiveQuery = cteInfo.nonRecursiveCriteriaBuilder.getQuery(modificationStates);
QuerySpecification nonRecursiveQuerySpecification = getQuerySpecification(nonRecursiveQuery);
Map nonRecursiveTableNameRemappings = null;
if (nonRecursiveQuery instanceof CustomSQLQuery) {
// EntityAlias -> CteName
nonRecursiveTableNameRemappings = cteInfo.nonRecursiveCriteriaBuilder.getModificationStateRelatedTableNameRemappings(explicitVersionEntities);
}
Query recursiveQuery;
QuerySpecification recursiveQuerySpecification = null;
Map recursiveTableNameRemappings = null;
if (cteInfo.recursive) {
modificationStates = cteInfo.nonRecursiveCriteriaBuilder.getModificationStates(explicitVersionEntities);
recursiveQuery = cteInfo.recursiveCriteriaBuilder.getQuery(modificationStates);
if (!mainQuery.dbmsDialect.supportsJoinsInRecursiveCte() && cteInfo.recursiveCriteriaBuilder.joinManager.hasNonEmulatableJoins()) {
throw new IllegalStateException("The dbms dialect does not support joins in the recursive part of a CTE!");
}
recursiveQuerySpecification = getQuerySpecification(recursiveQuery);
if (recursiveQuery instanceof CustomSQLQuery) {
// EntityAlias -> CteName
recursiveTableNameRemappings = cteInfo.recursiveCriteriaBuilder.getModificationStateRelatedTableNameRemappings(explicitVersionEntities);
}
}
String cteName = cteInfo.cteType.getName();
final List columnNames = cteInfo.columnNames;
String head;
String[] aliases;
if (mainQuery.dbmsDialect.supportsWithClauseHead()) {
sb.setLength(0);
sb.append(cteName);
sb.append('(');
for (int i = 0; i < columnNames.size(); i++) {
String column = columnNames.get(i);
if (i != 0) {
sb.append(", ");
}
sb.append(column);
}
sb.append(')');
head = sb.toString();
aliases = null;
} else {
sb.setLength(0);
sb.append(cteName);
List list = new ArrayList<>(columnNames.size());
for (int i = 0; i < columnNames.size(); i++) {
String[] columns = mainQuery.metamodel.getManagedType(ExtendedManagedType.class, cteInfo.cteType.getJavaType()).getAttribute(columnNames.get(i)).getColumnNames();
for (String column : columns) {
list.add(column);
}
}
head = sb.toString();
aliases = list.toArray(new String[list.size()]);
}
String nonRecursiveWithClauseSuffix = null;
if (!cteInfo.recursive && !mainQuery.dbmsDialect.supportsNonRecursiveWithClause()) {
sb.setLength(0);
sb.append("\nUNION ALL\n");
sb.append("SELECT ");
sb.append("NULL");
for (int i = 1; i < columnNames.size(); i++) {
sb.append(", ");
sb.append("NULL");
}
sb.append(" FROM DUAL WHERE 1=0");
nonRecursiveWithClauseSuffix = sb.toString();
}
cteNodes.add(new CTENode(
cteInfo.name,
cteInfo.cteType.getName(),
head,
aliases,
cteInfo.unionAll,
nonRecursiveQuerySpecification,
recursiveQuerySpecification,
nonRecursiveTableNameRemappings,
recursiveTableNameRemappings,
nonRecursiveWithClauseSuffix
));
}
return cteNodes;
}
protected Query getQuery() {
return getTypedQuery();
}
protected Query getQuery(Map includedModificationStates) {
return getQuery();
}
@SuppressWarnings("unchecked")
protected TypedQuery getTypedQuery(String queryString) {
TypedQuery query = (TypedQuery) em.createQuery(queryString, selectManager.getExpectedQueryResultType());
if (firstResult != 0) {
query.setFirstResult(firstResult);
}
if (maxResults != Integer.MAX_VALUE) {
query.setMaxResults(maxResults);
}
if (isCacheable()) {
mainQuery.jpaProvider.setCacheable(query);
}
return applyObjectBuilder(query);
}
@SuppressWarnings("unchecked")
public KeysetBuilder beforeKeyset() {
prepareForModification(ClauseType.WHERE);
return keysetManager.startBuilder(new KeysetBuilderImpl((BuilderType) this, keysetManager, KeysetMode.PREVIOUS));
}
public BuilderType beforeKeyset(Serializable... values) {
return beforeKeyset(new KeysetImpl(values));
}
@SuppressWarnings("unchecked")
public BuilderType beforeKeyset(Keyset keyset) {
prepareForModification(ClauseType.WHERE);
keysetManager.verifyBuilderEnded();
keysetManager.setKeysetLink(new SimpleKeysetLink(keyset, KeysetMode.PREVIOUS));
return (BuilderType) this;
}
@SuppressWarnings("unchecked")
public KeysetBuilder afterKeyset() {
prepareForModification(ClauseType.WHERE);
return keysetManager.startBuilder(new KeysetBuilderImpl((BuilderType) this, keysetManager, KeysetMode.NEXT));
}
public BuilderType afterKeyset(Serializable... values) {
return afterKeyset(new KeysetImpl(values));
}
@SuppressWarnings("unchecked")
public BuilderType afterKeyset(Keyset keyset) {
prepareForModification(ClauseType.WHERE);
keysetManager.verifyBuilderEnded();
keysetManager.setKeysetLink(new SimpleKeysetLink(keyset, KeysetMode.NEXT));
return (BuilderType) this;
}
protected String getBaseQueryString() {
if (cachedQueryString == null) {
cachedQueryString = buildBaseQueryString(false);
}
return cachedQueryString;
}
protected String getExternalQueryString() {
if (cachedExternalQueryString == null) {
cachedExternalQueryString = buildExternalQueryString();
}
return cachedExternalQueryString;
}
protected void prepareForModification(ClauseType changedClause) {
if (setOperationEnded) {
throw new IllegalStateException("Modifications to a query after connecting with a set operation is not allowed!");
}
needsCheck = true;
cachedQueryString = null;
cachedExternalQueryString = null;
cachedGroupByIdentifierExpressions = null;
implicitJoinsApplied = false;
if (changedClause == null || changedClause == ClauseType.WHERE) {
functionalDependencyAnalyzerVisitor.reset();
}
}
protected void prepareAndCheck() {
if (checkSetBuilderEnded) {
verifySetBuilderEnded();
}
if (!needsCheck) {
return;
}
verifyBuilderEnded();
joinManager.acceptVisitor(new JoinNodeVisitor() {
@Override
public void visit(JoinNode node) {
Class cteType = node.getJavaType();
// Except for VALUES clause from nodes, every cte type must be defined
if (node.getValueCount() == 0 && mainQuery.metamodel.getCte(cteType) != null) {
if (mainQuery.cteManager.getCte(cteType) == null) {
throw new IllegalStateException("Usage of CTE '" + cteType.getName() + "' without definition!");
}
}
}
});
// resolve unresolved aliases, object model etc.
// we must do implicit joining at the end because we can only do
// the aliases resolving at the end and alias resolving must happen before
// the implicit joins
// it makes no sense to do implicit joining before this point, since
// the user can call the api in arbitrary orders
// so where("b.c").join("a.b") but also
// join("a.b", "b").where("b.c")
// in the first case
JoinVisitor joinVisitor = applyImplicitJoins(null);
applyExpressionTransformersAndBuildGroupByClauses(false, joinVisitor);
hasCollections = joinManager.hasCollections();
if (keysetManager.hasKeyset()) {
// The last order by expression must be unique, otherwise keyset scrolling wouldn't work
List orderByExpressions = orderByManager.getOrderByExpressions(hasCollections, whereManager.rootPredicate.getPredicate(), hasGroupBy ? Arrays.asList(getGroupByIdentifierExpressions()) : Collections.emptyList(), joinVisitor);
if (!orderByExpressions.get(orderByExpressions.size() - 1).isResultUnique()) {
throw new IllegalStateException("The order by items of the query builder are not guaranteed to produce unique tuples! Consider also ordering by the entity identifier!");
}
keysetManager.initialize(orderByExpressions);
}
// No need to do all that stuff again if no mutation occurs
needsCheck = false;
}
protected ResolvedExpression[] getGroupByIdentifierExpressions() {
if (cachedGroupByIdentifierExpressions == null) {
Set resolvedExpressions = groupByManager.getCollectedGroupByClauses().keySet();
cachedGroupByIdentifierExpressions = resolvedExpressions.toArray(new ResolvedExpression[resolvedExpressions.size()]);
}
return cachedGroupByIdentifierExpressions;
}
protected String buildBaseQueryString(boolean externalRepresentation) {
StringBuilder sbSelectFrom = new StringBuilder();
buildBaseQueryString(sbSelectFrom, externalRepresentation);
return sbSelectFrom.toString();
}
protected void buildBaseQueryString(StringBuilder sbSelectFrom, boolean externalRepresentation) {
boolean originalExternalRepresentation = queryGenerator.isExternalRepresentation();
queryGenerator.setExternalRepresentation(externalRepresentation);
try {
appendSelectClause(sbSelectFrom, externalRepresentation);
List whereClauseEndConjuncts = this instanceof Subquery ? new ArrayList() : null;
List whereClauseConjuncts = new ArrayList<>();
List optionalWhereClauseConjuncts = new ArrayList<>();
joinManager.buildClause(sbSelectFrom, EnumSet.noneOf(ClauseType.class), null, false, externalRepresentation, false, optionalWhereClauseConjuncts, whereClauseConjuncts, whereClauseEndConjuncts, explicitVersionEntities, nodesToFetch, Collections.EMPTY_SET);
appendWhereClause(sbSelectFrom, whereClauseConjuncts, optionalWhereClauseConjuncts, whereClauseEndConjuncts);
appendGroupByClause(sbSelectFrom);
appendWindowClause(sbSelectFrom, externalRepresentation);
appendOrderByClause(sbSelectFrom);
if (externalRepresentation && !isMainQuery) {
// Don't render the LIMIT clause for subqueries, but let the parent render it in a LIMIT function
if (!(this instanceof SubqueryInternalBuilder)) {
applyJpaLimit(sbSelectFrom);
}
}
} finally {
queryGenerator.setExternalRepresentation(originalExternalRepresentation);
}
}
protected String buildExternalQueryString() {
StringBuilder sbSelectFrom = new StringBuilder();
buildExternalQueryString(sbSelectFrom);
return sbSelectFrom.toString();
}
protected void buildExternalQueryString(StringBuilder sbSelectFrom) {
if (isMainQuery) {
mainQuery.cteManager.buildClause(sbSelectFrom);
}
buildBaseQueryString(sbSelectFrom, true);
}
protected void appendSelectClause(StringBuilder sbSelectFrom, boolean externalRepresentation) {
selectManager.buildSelect(sbSelectFrom, false, externalRepresentation);
}
protected void appendWhereClause(StringBuilder sbSelectFrom, boolean externalRepresentation) {
boolean originalExternalRepresentation = queryGenerator.isExternalRepresentation();
queryGenerator.setExternalRepresentation(externalRepresentation);
try {
appendWhereClause(sbSelectFrom, Collections.emptyList(), Collections.emptyList(), Collections.emptyList());
} finally {
queryGenerator.setExternalRepresentation(originalExternalRepresentation);
}
}
protected void appendWhereClause(StringBuilder sbSelectFrom, List whereClauseConjuncts, List optionalWhereClauseConjuncts, List whereClauseEndConjuncts) {
KeysetLink keysetLink = keysetManager.getKeysetLink();
if (keysetLink == null || keysetLink.getKeysetMode() == KeysetMode.NONE) {
whereManager.buildClause(sbSelectFrom, whereClauseConjuncts, optionalWhereClauseConjuncts, whereClauseEndConjuncts);
} else {
sbSelectFrom.append(" WHERE ");
if (whereManager.hasPredicates()) {
whereManager.buildClausePredicate(sbSelectFrom, whereClauseConjuncts, optionalWhereClauseConjuncts, whereClauseEndConjuncts);
sbSelectFrom.append(" AND ");
}
int positionalOffset = parameterManager.getPositionalOffset();
if (mainQuery.getQueryConfiguration().isOptimizedKeysetPredicateRenderingEnabled()) {
keysetManager.buildOptimizedKeysetPredicate(sbSelectFrom, positionalOffset);
} else {
keysetManager.buildKeysetPredicate(sbSelectFrom, positionalOffset);
}
}
}
protected void appendGroupByClause(StringBuilder sbSelectFrom) {
if (hasGroupBy) {
groupByManager.buildGroupBy(sbSelectFrom);
havingManager.buildClause(sbSelectFrom);
}
}
protected void appendWindowClause(StringBuilder sbSelectFrom, boolean externalRepresentation) {
if (externalRepresentation) {
windowManager.buildWindow(sbSelectFrom);
}
}
protected void appendOrderByClause(StringBuilder sbSelectFrom) {
orderByManager.buildOrderBy(sbSelectFrom, false, false, false);
}
protected Map getModificationStates(Map, Map> explicitVersionEntities) {
return null;
}
protected Map getModificationStateRelatedTableNameRemappings(Map, Map> explicitVersionEntities) {
return null;
}
protected static boolean isEmpty(Set joinNodes, Set clauseExclusions) {
for (JoinNode node : joinNodes) {
if (!clauseExclusions.isEmpty() && clauseExclusions.containsAll(node.getClauseDependencies()) && !node.isCardinalityMandatory()) {
continue;
}
return false;
}
return true;
}
private QuerySpecification getQuerySpecification(Query query) {
if (query instanceof AbstractCustomQuery) {
return ((AbstractCustomQuery) query).getQuerySpecification();
}
return new DefaultQuerySpecification(statementType, query, em, parameterManager.getParameterListNames(query), cbf.getExtendedQuerySupport());
}
protected boolean hasLimit() {
return firstResult != 0 || maxResults != Integer.MAX_VALUE;
}
protected void applyJpaLimit(StringBuilder sbSelectFrom) {
if (hasLimit()) {
sbSelectFrom.append(" LIMIT ");
sbSelectFrom.append(maxResults);
if (firstResult > 0) {
sbSelectFrom.append(" OFFSET ");
sbSelectFrom.append(firstResult);
}
}
}
@SuppressWarnings("unchecked")
protected final TypedQuery applyObjectBuilder(TypedQuery query) {
ObjectBuilder selectObjectBuilder = selectManager.getSelectObjectBuilder();
if (selectObjectBuilder != null) {
return new ObjectBuilderTypedQuery<>(query, selectObjectBuilder);
} else {
return (TypedQuery) query;
}
}
// TODO: needs equals-hashCode implementation
}