Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Query clone support with pre-existing entityManager/connection #672 #713

Merged
merged 2 commits into from
Apr 14, 2014
Merged

Query clone support with pre-existing entityManager/connection #672 #713

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
f5e3cf1
Query clone support with pre-existing entityManager/connection #672
TuomasKiviaho Apr 13, 2014
488b5a9
Delete AbstractSQLSubQuery.java
TuomasKiviaho Apr 14, 2014
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
302 changes: 14 additions & 288 deletions querydsl-jpa/src/main/java/com/mysema/query/jpa/AbstractSQLQuery.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,49 +13,26 @@
*/
package com.mysema.query.jpa;

import javax.annotation.Nullable;
import javax.persistence.Entity;

import com.mysema.query.JoinFlag;
import com.mysema.query.Query;
import com.mysema.query.QueryFlag;
import com.mysema.query.QueryFlag.Position;
import com.mysema.query.QueryMetadata;
import com.mysema.query.sql.ForeignKey;
import com.mysema.query.sql.RelationalFunctionCall;
import com.mysema.query.sql.RelationalPath;
import com.mysema.query.sql.SQLCommonQuery;
import com.mysema.query.sql.SQLOps;
import com.mysema.query.sql.SQLTemplates;
import com.mysema.query.sql.Union;
import com.mysema.query.sql.UnionImpl;
import com.mysema.query.sql.UnionUtils;
import com.mysema.query.sql.WithBuilder;
import com.mysema.query.support.Expressions;
import com.mysema.query.support.ProjectableQuery;
import com.mysema.query.sql.Configuration;
import com.mysema.query.sql.ProjectableSQLQuery;
import com.mysema.query.sql.SQLSerializer;
import com.mysema.query.support.QueryMixin;
import com.mysema.query.types.EntityPath;
import com.mysema.query.types.Expression;
import com.mysema.query.types.ExpressionUtils;
import com.mysema.query.types.Operation;
import com.mysema.query.types.OperationImpl;
import com.mysema.query.types.Path;
import com.mysema.query.types.Predicate;
import com.mysema.query.types.SubQueryExpression;
import com.mysema.query.types.TemplateExpression;
import com.mysema.query.types.expr.Wildcard;
import com.mysema.query.types.query.ListSubQuery;
import com.mysema.query.types.template.NumberTemplate;
import com.mysema.query.types.template.SimpleTemplate;

/**
* Abstract super class for SQLQuery implementation for JPA and Hibernate
*
* @author tiwe
*
* @param <T> concrete subtype
* @param <Q> concrete subtype
*/
public abstract class AbstractSQLQuery<T extends AbstractSQLQuery<T> & Query<T>> extends ProjectableQuery<T> implements SQLCommonQuery<T> {
public abstract class AbstractSQLQuery<Q extends AbstractSQLQuery<Q>> extends ProjectableSQLQuery<Q> {

private static final class NativeQueryMixin<T> extends QueryMixin<T> {
private NativeQueryMixin(QueryMetadata metadata) {
Expand All @@ -68,280 +45,29 @@ public <RT> Expression<RT> convert(Expression<RT> expr, boolean forOrder) {
}
}

protected final QueryMixin<T> queryMixin;

@Nullable
protected Expression<?> union;

protected boolean unionAll;

@SuppressWarnings("unchecked")
public AbstractSQLQuery(QueryMetadata metadata) {
super(new NativeQueryMixin<T>(metadata));
this.queryMixin = super.queryMixin;
this.queryMixin.setSelf((T)this);
public AbstractSQLQuery(QueryMetadata metadata, Configuration configuration) {
super(new NativeQueryMixin<Q>(metadata), configuration);
this.queryMixin.setSelf((Q) this);
}

protected boolean isEntityExpression(Expression<?> expr) {
protected static boolean isEntityExpression(Expression<?> expr) {
return expr instanceof EntityPath || expr.getType().isAnnotationPresent(Entity.class);
}

protected Expression<?> extractEntityExpression(Expression<?> expr) {
protected static Expression<?> extractEntityExpression(Expression<?> expr) {
if (expr instanceof Operation) {
return ((Operation)expr).getArg(0);
return ((Operation<?>)expr).getArg(0);
} else if (expr instanceof TemplateExpression) {
return (Expression<?>) ((TemplateExpression)expr).getArg(0);
return (Expression<?>) ((TemplateExpression<?>)expr).getArg(0);
} else {
return expr;
}
}

@Override
public long count() {
Number number = uniqueResult(Wildcard.countAsInt);
return number.longValue();
}

@Override
public boolean exists() {
return limit(1).uniqueResult(NumberTemplate.ONE) != null;
}

public T from(Expression<?> arg) {
return queryMixin.from(arg);
}

@Override
public T from(Expression<?>... args) {
return queryMixin.from(args);
}

@Override
@SuppressWarnings("unchecked")
public T from(SubQueryExpression<?> subQuery, Path<?> alias) {
return queryMixin.from(ExpressionUtils.as((Expression) subQuery, alias));
}

@Override
public T fullJoin(EntityPath<?> o) {
return queryMixin.fullJoin(o);
}

@Override
public <E> T fullJoin(RelationalFunctionCall<E> target, Path<E> alias) {
return queryMixin.fullJoin(target, alias);
}

@Override
public <E> T fullJoin(ForeignKey<E> key, RelationalPath<E> entity) {
return queryMixin.fullJoin(entity).on(key.on(entity));
}

@Override
public T fullJoin(SubQueryExpression<?> o, Path<?> alias) {
return queryMixin.fullJoin(o, alias);
}

public QueryMetadata getMetadata() {
return queryMixin.getMetadata();
}

@Override
public T innerJoin(EntityPath<?> o) {
return queryMixin.innerJoin(o);
}

@Override
public <E> T innerJoin(RelationalFunctionCall<E> target, Path<E> alias) {
return queryMixin.innerJoin(target, alias);
}

@Override
public <E> T innerJoin(ForeignKey<E> key, RelationalPath<E> entity) {
return queryMixin.innerJoin(entity).on(key.on(entity));
}

@Override
public T innerJoin(SubQueryExpression<?> o, Path<?> alias) {
return queryMixin.innerJoin(o, alias);
}

@Override
public T join(EntityPath<?> o) {
return queryMixin.join(o);
}

@Override
public <E> T join(RelationalFunctionCall<E> target, Path<E> alias) {
return queryMixin.join(target, alias);
}

@Override
public <E> T join(ForeignKey<E> key, RelationalPath<E> entity) {
return queryMixin.join(entity).on(key.on(entity));
}

@Override
public T join(SubQueryExpression<?> o, Path<?> alias) {
return queryMixin.join(o, alias);
}

@Override
public T leftJoin(EntityPath<?> o) {
return queryMixin.leftJoin(o);
}

@Override
public <E> T leftJoin(RelationalFunctionCall<E> target, Path<E> alias) {
return queryMixin.leftJoin(target, alias);
}

@Override
public <E> T leftJoin(ForeignKey<E> key, RelationalPath<E> entity) {
return queryMixin.innerJoin(entity).on(key.on(entity));
}

@Override
public T leftJoin(SubQueryExpression<?> o, Path<?> alias) {
return queryMixin.leftJoin(o, alias);
}

public T on(Predicate condition) {
return queryMixin.on(condition);
}

@Override
public T on(Predicate... conditions) {
return queryMixin.on(conditions);
}

@Override
public T rightJoin(EntityPath<?> o) {
return queryMixin.rightJoin(o);
}

@Override
public <E> T rightJoin(RelationalFunctionCall<E> target, Path<E> alias) {
return queryMixin.rightJoin(target, alias);
}

@Override
public <E> T rightJoin(ForeignKey<E> key, RelationalPath<E> entity) {
return queryMixin.innerJoin(entity).on(key.on(entity));
}

@Override
public T rightJoin(SubQueryExpression<?> o, Path<?> alias) {
return queryMixin.rightJoin(o, alias);
}

public <RT> Union<RT> union(ListSubQuery<RT>... sq) {
return innerUnion(sq);
}

public <RT> Union<RT> union(SubQueryExpression<RT>... sq) {
return innerUnion(sq);
}

public <RT> Union<RT> unionAll(ListSubQuery<RT>... sq) {
unionAll = true;
return innerUnion(sq);
}

public <RT> Union<RT> unionAll(SubQueryExpression<RT>... sq) {
unionAll = true;
return innerUnion(sq);
}

public <RT> T union(Path<?> alias, ListSubQuery<RT>... sq) {
return from(UnionUtils.union(sq, alias, false));
}

public <RT> T union(Path<?> alias, SubQueryExpression<RT>... sq) {
return from(UnionUtils.union(sq, alias, false));
}

public <RT> T unionAll(Path<?> alias, ListSubQuery<RT>... sq) {
return from(UnionUtils.union(sq, alias, true));
}

public <RT> T unionAll(Path<?> alias, SubQueryExpression<RT>... sq) {
return from(UnionUtils.union(sq, alias, true));
}

@Override
public T withRecursive(Path<?> alias, SubQueryExpression<?> query) {
queryMixin.addFlag(new QueryFlag(QueryFlag.Position.WITH, SQLTemplates.RECURSIVE));
return with(alias, query);
}

@Override
public T withRecursive(Path<?> alias, Expression<?> query) {
queryMixin.addFlag(new QueryFlag(QueryFlag.Position.WITH, SQLTemplates.RECURSIVE));
return with(alias, query);
}

@Override
public WithBuilder<T> withRecursive(Path<?> alias, Path<?>... columns) {
queryMixin.addFlag(new QueryFlag(QueryFlag.Position.WITH, SQLTemplates.RECURSIVE));
return with(alias, columns);
}

@Override
public T with(Path<?> alias, Expression<?> query) {
Expression<?> expr = OperationImpl.create(alias.getType(), SQLOps.WITH_ALIAS, alias, query);
return queryMixin.addFlag(new QueryFlag(QueryFlag.Position.WITH, expr));
}

@Override
public WithBuilder<T> with(Path<?> alias, Path<?>... columns) {
Expression<?> columnsCombined = ExpressionUtils.list(Object.class, columns);
Expression<?> aliasCombined = Expressions.operation(alias.getType(), SQLOps.WITH_COLUMNS, alias, columnsCombined);
return new WithBuilder<T>(queryMixin, aliasCombined);
}

@Override
public T with(Path<?> alias, SubQueryExpression<?> query) {
Expression<?> expr = OperationImpl.create(alias.getType(), SQLOps.WITH_ALIAS, alias, query);
return queryMixin.addFlag(new QueryFlag(QueryFlag.Position.WITH, expr));
}

@SuppressWarnings("unchecked")
protected <RT> Union<RT> innerUnion(SubQueryExpression<?>... sq) {
queryMixin.getMetadata().setValidate(false);
if (!queryMixin.getMetadata().getJoins().isEmpty()) {
throw new IllegalArgumentException("Don't mix union and from");
}
this.union = UnionUtils.union(sq, unionAll);
return new UnionImpl<T, RT>((T)this, sq[0].getMetadata().getProjection());
}

@Override
public T addJoinFlag(String flag) {
return addJoinFlag(flag, JoinFlag.Position.BEFORE_TARGET);
}

@Override
@SuppressWarnings("unchecked")
public T addJoinFlag(String flag, JoinFlag.Position position) {
queryMixin.addJoinFlag(new JoinFlag(flag, position));
return (T)this;
}

@Override
public T addFlag(Position position, String prefix, Expression<?> expr) {
Expression<?> flag = SimpleTemplate.create(expr.getType(), prefix + "{0}", expr);
return queryMixin.addFlag(new QueryFlag(position, flag));
}

@Override
public T addFlag(Position position, String flag) {
return queryMixin.addFlag(new QueryFlag(position, flag));
}

@Override
public T addFlag(Position position, Expression<?> flag) {
return queryMixin.addFlag(new QueryFlag(position, flag));
protected SQLSerializer createSerializer() {
return new NativeSQLSerializer(configuration, true);
}

}