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.AbstractCTECriteriaBuilder Maven / Gradle / Ivy
/*
* Copyright 2014 - 2020 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.BaseCTECriteriaBuilder;
import com.blazebit.persistence.SelectBuilder;
import com.blazebit.persistence.impl.query.CTEQuerySpecification;
import com.blazebit.persistence.impl.query.CustomSQLQuery;
import com.blazebit.persistence.impl.query.EntityFunctionNode;
import com.blazebit.persistence.impl.query.QuerySpecification;
import com.blazebit.persistence.parser.expression.ExpressionCopyContext;
import com.blazebit.persistence.spi.DbmsStatementType;
import com.blazebit.persistence.spi.ExtendedAttribute;
import com.blazebit.persistence.spi.ExtendedManagedType;
import com.blazebit.persistence.spi.SetOperationType;
import javax.persistence.Query;
import javax.persistence.metamodel.EntityType;
import java.util.Arrays;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
/**
*
* @param The criteria builder returned after the cte builder
* @param The concrete builder type
* @param The builder type that should be returned on set operations
* @author Christian Beikov
* @since 1.1.0
*/
public abstract class AbstractCTECriteriaBuilder, Z, W> extends AbstractCommonQueryBuilder> implements BaseCTECriteriaBuilder, SelectBuilder, CTEInfoBuilder {
protected final Y result;
protected final CTEBuilderListener listener;
protected final CTEManager.CTEKey cteKey;
protected final boolean inline;
protected final EntityType cteType;
protected final Map> attributeEntries;
protected final Map bindingMap;
protected final Map columnBindingMap;
protected final CTEBuilderListenerImpl subListener;
private CTEInfo info;
public AbstractCTECriteriaBuilder(MainQuery mainQuery, QueryContext queryContext, CTEManager.CTEKey cteKey, boolean inline, Class clazz, Y result, CTEBuilderListener listener, BaseFinalSetOperationCTECriteriaBuilderImpl finalSetOperationBuilder, AliasManager parentAliasManager, JoinManager parentJoinManager) {
super(mainQuery, queryContext, false, DbmsStatementType.SELECT, clazz, null, finalSetOperationBuilder, false, parentAliasManager, parentJoinManager);
this.cteKey = cteKey;
this.result = result;
this.listener = listener;
this.cteType = mainQuery.metamodel.entity(clazz);
this.attributeEntries = mainQuery.metamodel.getManagedType(ExtendedManagedType.class, clazz).getOwnedSingularAttributes();
this.inline = inline;
this.bindingMap = new LinkedHashMap<>(attributeEntries.size());
this.columnBindingMap = new LinkedHashMap<>(attributeEntries.size());
this.subListener = new CTEBuilderListenerImpl();
}
public AbstractCTECriteriaBuilder(AbstractCTECriteriaBuilder builder, MainQuery mainQuery, QueryContext queryContext, Map joinManagerMapping, ExpressionCopyContext copyContext) {
super(builder, mainQuery, queryContext, joinManagerMapping, copyContext);
this.cteKey = new CTEManager.CTEKey(builder.cteKey.getName(), joinManagerMapping.get(builder.cteKey.getOwner()));
this.result = null;
this.listener = null;
this.cteType = builder.cteType;
this.attributeEntries = builder.attributeEntries;
this.inline = builder.inline;
this.bindingMap = new LinkedHashMap<>(builder.bindingMap);
this.columnBindingMap = new LinkedHashMap<>(builder.columnBindingMap);
this.subListener = null;
}
public CTEBuilderListenerImpl getSubListener() {
return subListener;
}
@Override
protected void buildExternalQueryString(StringBuilder sbSelectFrom) {
buildBaseQueryString(sbSelectFrom, true, null);
}
@Override
protected Query getQuery() {
// NOTE: This must happen first because it generates implicit joins
prepareAndCheck();
String baseQueryString = buildBaseQueryString(false);
return getQuery(baseQueryString);
}
protected Query getQuery(String baseQueryString) {
Set keyRestrictedLeftJoins = getKeyRestrictedLeftJoins();
Query query;
if (hasLimit() || joinManager.hasEntityFunctions() || !keyRestrictedLeftJoins.isEmpty()) {
// We need to change the underlying sql when doing a limit
query = em.createQuery(baseQueryString);
Set parameterListNames = parameterManager.getParameterListNames(query);
String limit = null;
String offset = null;
// The main query will handle that separately
if (!isMainQuery) {
if (firstResult != 0) {
query.setFirstResult(firstResult);
offset = Integer.toString(firstResult);
}
if (maxResults != Integer.MAX_VALUE) {
query.setMaxResults(maxResults);
limit = Integer.toString(maxResults);
}
}
List keyRestrictedLeftJoinAliases = getKeyRestrictedLeftJoinAliases(query, keyRestrictedLeftJoins, Collections.emptySet());
List entityFunctionNodes = getEntityFunctionNodes(query);
QuerySpecification querySpecification = new CTEQuerySpecification(
this,
query,
parameterManager.getParameters(),
parameterListNames,
limit,
offset,
keyRestrictedLeftJoinAliases,
entityFunctionNodes
);
query = new CustomSQLQuery(
querySpecification,
query,
parameterManager.getTransformers(),
parameterManager.getValuesParameters(),
parameterManager.getValuesBinders()
);
} else {
query = em.createQuery(baseQueryString);
}
parameterManager.parameterizeQuery(query);
return query;
}
public SelectBuilder bind(String cteAttribute) {
ExtendedAttribute attributeEntry = attributeEntries.get(cteAttribute);
if (attributeEntry == null) {
throw new IllegalArgumentException("The cte attribute [" + cteAttribute + "] does not exist!");
}
// TODO: move this check and a type check to end()
// if (attributeEntry.getAttribute().getPersistentAttributeType() == Attribute.PersistentAttributeType.EMBEDDED) {
// throw new IllegalArgumentException("Can't bind the embeddable cte attribute [" + cteAttribute + "] directly! Please bind the respective sub attributes.");
// }
if (bindingMap.put(cteAttribute, selectManager.getSelectInfos().size()) != null) {
throw new IllegalArgumentException("The cte attribute [" + cteAttribute + "] has already been bound!");
}
for (String column : attributeEntry.getColumnNames()) {
if (columnBindingMap.put(column, cteAttribute) != null) {
throw new IllegalArgumentException("The cte column [" + column + "] has already been bound!");
}
}
return this;
}
public Y end() {
listener.onBuilderEnded(this);
return result;
}
@Override
protected void prepareAndCheck() {
if (!needsCheck) {
return;
}
try {
List attributes = prepareAndGetAttributes();
List columns = prepareAndGetColumnNames();
super.prepareAndCheck();
info = new CTEInfo(cteKey.getName(), cteKey.getOwner(), inline, cteType, attributes, columns, false, false, this, null);
} catch (RuntimeException ex) {
needsCheck = true;
throw ex;
}
}
public CTEInfo createCTEInfo() {
prepareAndCheck();
return info;
}
protected List prepareAndGetAttributes() {
JpaUtils.expandBindings(bindingMap, columnBindingMap, attributeEntries, ClauseType.SELECT, this, null, true);
String[] attributes = new String[bindingMap.size()];
for (Map.Entry entry : bindingMap.entrySet()) {
attributes[entry.getValue()] = entry.getKey();
}
return Arrays.asList(attributes);
}
protected List prepareAndGetColumnNames() {
StringBuilder sb = null;
for (ExtendedAttribute entry : attributeEntries.values()) {
for (String column : entry.getColumnNames()) {
if (!columnBindingMap.containsKey(column)) {
if (sb == null) {
sb = new StringBuilder();
sb.append("[");
} else {
sb.append(", ");
}
sb.append(column);
}
}
}
if (sb != null) {
sb.insert(0, "The following column names have not been bound: ");
sb.append("]");
throw new IllegalStateException(sb.toString());
}
String[] columns = new String[columnBindingMap.size()];
for (Map.Entry columnBinding : columnBindingMap.entrySet()) {
columns[bindingMap.get(columnBinding.getValue())] = columnBinding.getKey();
}
return Arrays.asList(columns);
}
protected BaseFinalSetOperationCTECriteriaBuilderImpl createFinalSetOperationBuilder(SetOperationType operator, boolean nested, boolean isSubquery) {
FullSelectCTECriteriaBuilderImpl newInitiator = finalSetOperationBuilder == null ? null : finalSetOperationBuilder.getInitiator();
return createFinalSetOperationBuilder(operator, nested, isSubquery, newInitiator);
}
@SuppressWarnings("unchecked")
protected BaseFinalSetOperationCTECriteriaBuilderImpl createFinalSetOperationBuilder(SetOperationType operator, boolean nested, boolean isSubquery, FullSelectCTECriteriaBuilderImpl initiator) {
CTEBuilderListener newListener = finalSetOperationBuilder == null ? listener : finalSetOperationBuilder.getSubListener();
Y newResult = finalSetOperationBuilder == null ? result : (Y) finalSetOperationBuilder.getResult();
if (isSubquery) {
return new OngoingFinalSetOperationCTECriteriaBuilderImpl(mainQuery, queryContext, (Class) cteType.getJavaType(), newResult, operator, nested, newListener, initiator);
} else {
return new FinalSetOperationCTECriteriaBuilderImpl(mainQuery, queryContext, (Class) cteType.getJavaType(), newResult, operator, nested, newListener, initiator);
}
}
@SuppressWarnings("unchecked")
protected LeafOngoingSetOperationCTECriteriaBuilderImpl createLeaf(BaseFinalSetOperationCTECriteriaBuilderImpl finalSetOperationBuilder) {
CTEBuilderListener newListener = finalSetOperationBuilder.getSubListener();
LeafOngoingSetOperationCTECriteriaBuilderImpl next = new LeafOngoingSetOperationCTECriteriaBuilderImpl(mainQuery, queryContext, cteKey, inline, (Class) cteType.getJavaType(), result, newListener, (FinalSetOperationCTECriteriaBuilderImpl) finalSetOperationBuilder, aliasManager.getParent(), joinManager.getParent());
newListener.onBuilderStarted(next);
return next;
}
@SuppressWarnings("unchecked")
protected > StartOngoingSetOperationCTECriteriaBuilderImpl createStartOngoing(BaseFinalSetOperationCTECriteriaBuilderImpl finalSetOperationBuilder, T endSetResult) {
// TODO: This is such an ugly hack, but I don't know how else to fix this generics issue for now
finalSetOperationBuilder.setEndSetResult((T) endSetResult);
CTEBuilderListener newListener = finalSetOperationBuilder.getSubListener();
StartOngoingSetOperationCTECriteriaBuilderImpl next = new StartOngoingSetOperationCTECriteriaBuilderImpl(mainQuery, queryContext, cteKey, inline, (Class) cteType.getJavaType(), result, newListener, (OngoingFinalSetOperationCTECriteriaBuilderImpl) finalSetOperationBuilder, endSetResult, aliasManager.getParent(), joinManager.getParent());
newListener.onBuilderStarted(next);
return next;
}
@SuppressWarnings("unchecked")
protected > OngoingSetOperationCTECriteriaBuilderImpl createOngoing(BaseFinalSetOperationCTECriteriaBuilderImpl finalSetOperationBuilder, T endSetResult) {
// TODO: This is such an ugly hack, but I don't know how else to fix this generics issue for now
finalSetOperationBuilder.setEndSetResult((T) endSetResult);
CTEBuilderListener newListener = finalSetOperationBuilder.getSubListener();
OngoingSetOperationCTECriteriaBuilderImpl next = new OngoingSetOperationCTECriteriaBuilderImpl(mainQuery, queryContext, cteKey, inline, (Class) cteType.getJavaType(), result, newListener, (OngoingFinalSetOperationCTECriteriaBuilderImpl) finalSetOperationBuilder, endSetResult, aliasManager.getParent(), joinManager.getParent());
newListener.onBuilderStarted(next);
return next;
}
}