org.hibernate.query.criteria.internal.CriteriaSubqueryImpl Maven / Gradle / Ivy
/*
* Hibernate, Relational Persistence for Idiomatic Java
*
* License: GNU Lesser General Public License (LGPL), version 2.1 or later.
* See the lgpl.txt file in the root directory or .
*/
package org.hibernate.query.criteria.internal;
import java.io.Serializable;
import java.util.List;
import java.util.Set;
import javax.persistence.criteria.AbstractQuery;
import javax.persistence.criteria.CollectionJoin;
import javax.persistence.criteria.CommonAbstractCriteria;
import javax.persistence.criteria.Expression;
import javax.persistence.criteria.Join;
import javax.persistence.criteria.ListJoin;
import javax.persistence.criteria.MapJoin;
import javax.persistence.criteria.ParameterExpression;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
import javax.persistence.criteria.SetJoin;
import javax.persistence.criteria.Subquery;
import javax.persistence.metamodel.EntityType;
import org.hibernate.query.criteria.internal.compile.RenderingContext;
import org.hibernate.query.criteria.internal.expression.DelegatedExpressionImpl;
import org.hibernate.query.criteria.internal.expression.ExpressionImpl;
import org.hibernate.query.criteria.internal.path.RootImpl;
/**
* The Hibernate implementation of the JPA {@link Subquery} contract. Mostlty a set of delegation to its internal
* {@link QueryStructure}.
*
* @author Steve Ebersole
*/
public class CriteriaSubqueryImpl extends ExpressionImpl implements Subquery, Serializable {
private final CommonAbstractCriteria parent;
private final QueryStructure queryStructure;
public CriteriaSubqueryImpl(
CriteriaBuilderImpl criteriaBuilder,
Class javaType,
CommonAbstractCriteria parent) {
super( criteriaBuilder, javaType);
this.parent = parent;
this.queryStructure = new QueryStructure( this, criteriaBuilder );
}
@Override
public AbstractQuery getParent() {
if ( ! AbstractQuery.class.isInstance( parent ) ) {
throw new IllegalStateException( "Cannot call getParent on update/delete criterias" );
}
return (AbstractQuery) parent;
}
@Override
public CommonAbstractCriteria getContainingQuery() {
return parent;
}
@Override
public void registerParameters(ParameterRegistry registry) {
for ( ParameterExpression param : queryStructure.getParameters() ) {
registry.registerParameter( param );
}
}
@Override
public Class getResultType() {
return getJavaType();
}
// ROOTS ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@Override
public Set> getRoots() {
return queryStructure.getRoots();
}
@Override
public Root from(EntityType entityType) {
return queryStructure.from( entityType );
}
@Override
public Root from(Class entityClass) {
return queryStructure.from( entityClass );
}
// SELECTION ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@Override
public Subquery distinct(boolean applyDistinction) {
queryStructure.setDistinct( applyDistinction );
return this;
}
@Override
public boolean isDistinct() {
return queryStructure.isDistinct();
}
private Expression wrappedSelection;
@Override
public Expression getSelection() {
if ( wrappedSelection == null ) {
if ( queryStructure.getSelection() == null ) {
return null;
}
wrappedSelection = new SubquerySelection( (ExpressionImpl) queryStructure.getSelection(), this );
}
return wrappedSelection;
}
@Override
public Subquery select(Expression expression) {
queryStructure.setSelection( expression );
return this;
}
public static class SubquerySelection extends DelegatedExpressionImpl {
private final CriteriaSubqueryImpl subQuery;
public SubquerySelection(ExpressionImpl wrapped, CriteriaSubqueryImpl subQuery) {
super( wrapped );
this.subQuery = subQuery;
}
@Override
public String render(RenderingContext renderingContext) {
return subQuery.render( renderingContext );
}
@Override
public String renderProjection(RenderingContext renderingContext) {
return render( renderingContext );
}
}
// RESTRICTION ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@Override
public Predicate getRestriction() {
return queryStructure.getRestriction();
}
@Override
public Subquery where(Expression expression) {
queryStructure.setRestriction( criteriaBuilder().wrap( expression ) );
return this;
}
@Override
public Subquery where(Predicate... predicates) {
// TODO : assuming this should be a conjuntion, but the spec does not say specifically...
queryStructure.setRestriction( criteriaBuilder().and( predicates ) );
return this;
}
// GROUPING ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@Override
public List> getGroupList() {
return queryStructure.getGroupings();
}
@Override
public Subquery groupBy(Expression... groupings) {
queryStructure.setGroupings( groupings );
return this;
}
@Override
public Subquery groupBy(List> groupings) {
queryStructure.setGroupings( groupings );
return this;
}
@Override
public Predicate getGroupRestriction() {
return queryStructure.getHaving();
}
@Override
public Subquery having(Expression expression) {
queryStructure.setHaving( criteriaBuilder().wrap( expression ) );
return this;
}
@Override
public Subquery having(Predicate... predicates) {
queryStructure.setHaving( criteriaBuilder().and( predicates ) );
return this;
}
// CORRELATIONS ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@Override
public Set> getCorrelatedJoins() {
return queryStructure.collectCorrelatedJoins();
}
@Override
public Root correlate(Root source) {
final RootImpl correlation = ( ( RootImpl ) source ).correlateTo( this );
queryStructure.addCorrelationRoot( correlation );
return correlation;
}
@Override
public Join correlate(Join source) {
final JoinImplementor correlation = ( (JoinImplementor) source ).correlateTo( this );
queryStructure.addCorrelationRoot( correlation );
return correlation;
}
@Override
public CollectionJoin correlate(CollectionJoin source) {
final CollectionJoinImplementor correlation = ( (CollectionJoinImplementor) source ).correlateTo( this );
queryStructure.addCorrelationRoot( correlation );
return correlation;
}
@Override
public SetJoin correlate(SetJoin source) {
final SetJoinImplementor correlation = ( (SetJoinImplementor) source ).correlateTo( this );
queryStructure.addCorrelationRoot( correlation );
return correlation;
}
@Override
public ListJoin correlate(ListJoin source) {
final ListJoinImplementor correlation = ( (ListJoinImplementor) source ).correlateTo( this );
queryStructure.addCorrelationRoot( correlation );
return correlation;
}
@Override
public MapJoin correlate(MapJoin source) {
final MapJoinImplementor correlation = ( (MapJoinImplementor) source ).correlateTo( this );
queryStructure.addCorrelationRoot( correlation );
return correlation;
}
@Override
public Subquery subquery(Class subqueryType) {
return queryStructure.subquery( subqueryType );
}
// rendering ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@Override
public String render(RenderingContext renderingContext) {
StringBuilder subqueryBuffer = new StringBuilder( "(" );
queryStructure.render( subqueryBuffer, renderingContext );
subqueryBuffer.append( ')' );
return subqueryBuffer.toString();
}
@Override
public String renderProjection(RenderingContext renderingContext) {
throw new IllegalStateException( "Subquery cannot occur in select clause" );
}
}