AbstractJpaExpressionImpl.java

/*
 * Hibernate, Relational Persistence for Idiomatic Java
 *
 * License: Apache License, Version 2.0
 * See the LICENSE file in the root directory or visit http://www.apache.org/licenses/LICENSE-2.0
 */
package org.hibernate.test.sqm.parser.criteria.tree.expression;

import java.io.Serializable;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.util.Collection;
import javax.persistence.criteria.Expression;

import org.hibernate.query.sqm.produce.spi.criteria.JpaExpression;
import org.hibernate.query.sqm.produce.spi.criteria.JpaPredicate;
import org.hibernate.query.sqm.domain.SqmExpressableType;

import org.hibernate.test.sqm.parser.criteria.tree.CriteriaBuilderImpl;
import org.hibernate.test.sqm.parser.criteria.tree.expression.function.CastFunction;
import org.hibernate.test.sqm.parser.criteria.tree.select.AbstractSimpleSelection;

/**
 * Models an expression in the criteria query language.
 *
 * @author Steve Ebersole
 */
public abstract class AbstractJpaExpressionImpl<T>
		extends AbstractSimpleSelection<T>
		implements JpaExpression<T>, Serializable {
	private SqmExpressableType sqmType;

	public AbstractJpaExpressionImpl(
			CriteriaBuilderImpl criteriaBuilder,
			SqmExpressableType sqmType,
			Class<T> javaType) {
		super( criteriaBuilder, javaType );
		this.sqmType = sqmType;
	}

	@Override
	public SqmExpressableType getExpressionType() {
		return sqmType;
	}

	@Override
	@SuppressWarnings({ "unchecked" })
	public <X> JpaExpression<X> as(Class<X> type) {
		if ( type.equals( getJavaType() ) ) {
			return (JpaExpression<X>) this;
		}
		else {
			return new CastFunction(
					this,
					criteriaBuilder().consumerContext().getDomainMetamodel().resolveBasicType( type ),
					type,
					criteriaBuilder()
			);
		}
	}

	@Override
	public JpaPredicate isNull() {
		return criteriaBuilder().isNull( this );
	}

	@Override
	public JpaPredicate isNotNull() {
		return criteriaBuilder().isNotNull( this );
	}

	@Override
	public JpaPredicate in(Object... values) {
		return (JpaPredicate) criteriaBuilder().in( this, values );
	}

	@Override
	public JpaPredicate in(Expression<?>... values) {
		return (JpaPredicate) criteriaBuilder().in( this, values );
	}

	@Override
	public JpaPredicate in(Collection<?> values) {
		return (JpaPredicate) criteriaBuilder().in( this, values.toArray() );
	}

	@Override
	public JpaPredicate in(Expression<Collection<?>> values) {
		return (JpaPredicate) criteriaBuilder().in( this, values );
	}

	@Override
	@SuppressWarnings({ "unchecked" })
	public JpaExpression<Long> asLong() {
		resetJavaType( Long.class );
		return (JpaExpression<Long>) this;
	}

	@Override
	@SuppressWarnings({ "unchecked" })
	public JpaExpression<Integer> asInteger() {
		resetJavaType( Integer.class );
		return (JpaExpression<Integer>) this;
	}

	@Override
	@SuppressWarnings({ "unchecked" })
	public JpaExpression<Float> asFloat() {
		resetJavaType( Float.class );
		return (JpaExpression<Float>) this;
	}

	@Override
	@SuppressWarnings({ "unchecked" })
	public JpaExpression<Double> asDouble() {
		resetJavaType( Double.class );
		return (JpaExpression<Double>) this;
	}

	@Override
	@SuppressWarnings({ "unchecked" })
	public JpaExpression<BigDecimal> asBigDecimal() {
		resetJavaType( BigDecimal.class );
		return (JpaExpression<BigDecimal>) this;
	}

	@Override
	@SuppressWarnings({ "unchecked" })
	public JpaExpression<BigInteger> asBigInteger() {
		resetJavaType( BigInteger.class );
		return (JpaExpression<BigInteger>) this;
	}

	@Override
	@SuppressWarnings({ "unchecked" })
	public JpaExpression<String> asString() {
		resetJavaType( String.class );
		return (JpaExpression<String>) this;
	}
}