com.mysema.query.jpa.AbstractSQLQuery Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of querydsl-jpa Show documentation
JPA support for Querydsl
There is a newer version: 3.7.4
/*
 * Copyright 2011, Mysema Ltd
 *
 * 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.mysema.query.jpa;

import javax.annotation.Nullable;

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.SQLTemplates;
import com.mysema.query.sql.Union;
import com.mysema.query.sql.UnionImpl;
import com.mysema.query.sql.UnionUtils;
import com.mysema.query.support.ProjectableQuery;
import com.mysema.query.support.QueryMixin;
import com.mysema.query.types.Expression;
import com.mysema.query.types.ExpressionUtils;
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.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  concrete subtype
 */
public abstract class AbstractSQLQuery & Query> extends ProjectableQuery implements SQLCommonQuery {

    private static final class NativeQueryMixin extends QueryMixin {
        private NativeQueryMixin(QueryMetadata metadata) {
            super(metadata, false);
        }

        @Override
        public  Expression convert(Expression expr) {
            return super.convert(Conversions.convertForNativeQuery(expr));
        }
    }

    protected final QueryMixin queryMixin;

    @Nullable
    protected Expression union;

    protected boolean unionAll;

    @SuppressWarnings("unchecked")
    public AbstractSQLQuery(QueryMetadata metadata) {
        super(new NativeQueryMixin(metadata));
        this.queryMixin = super.queryMixin;
        this.queryMixin.setSelf((T)this);
    }

    @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(RelationalPath o) {
        return queryMixin.fullJoin(o);
    }

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

    @Override
    public  T fullJoin(ForeignKey key, RelationalPath 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(RelationalPath o) {
        return queryMixin.innerJoin(o);
    }

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

    @Override
    public  T innerJoin(ForeignKey key, RelationalPath 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(RelationalPath o) {
        return queryMixin.join(o);
    }

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

    @Override
    public  T join(ForeignKey key, RelationalPath 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(RelationalPath o) {
        return queryMixin.leftJoin(o);
    }

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

    @Override
    public  T leftJoin(ForeignKey key, RelationalPath 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(RelationalPath o) {
        return queryMixin.rightJoin(o);
    }

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

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

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

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

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

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

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

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

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

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

    public  T unionAll(Path alias, SubQueryExpression... 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 with(Path alias, SubQueryExpression query) {
        Expression expr = OperationImpl.create(alias.getType(), SQLTemplates.WITH_ALIAS, alias, query);
        return queryMixin.addFlag(new QueryFlag(QueryFlag.Position.WITH, expr));
    }

    @SuppressWarnings("unchecked")
    protected  Union 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)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));
    }

}