org.hibernate.query.sqm.tree.from.SqmRoot Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of hibernate-core Show documentation
Hibernate's core ORM functionality
There is a newer version: 7.0.0.Beta1
/*
 * 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 http://www.gnu.org/licenses/lgpl-2.1.html
 */
package org.hibernate.query.sqm.tree.from;

import java.util.function.Consumer;

import org.hibernate.metamodel.mapping.JdbcMapping;
import org.hibernate.metamodel.model.domain.EntityDomainType;
import org.hibernate.persister.entity.EntityPersister;
import org.hibernate.query.PathException;
import org.hibernate.query.criteria.JpaEntityJoin;
import org.hibernate.query.criteria.JpaRoot;
import org.hibernate.query.sqm.NodeBuilder;
import org.hibernate.query.sqm.SemanticQueryWalker;
import org.hibernate.query.sqm.sql.internal.DomainResultProducer;
import org.hibernate.query.sqm.tree.SqmJoinType;
import org.hibernate.query.sqm.tree.domain.AbstractSqmFrom;
import org.hibernate.query.sqm.tree.domain.SqmPath;
import org.hibernate.query.sqm.tree.domain.SqmTreatedPath;
import org.hibernate.query.sqm.tree.domain.SqmTreatedRoot;
import org.hibernate.sql.ast.Clause;
import org.hibernate.sql.results.spi.DomainResult;
import org.hibernate.sql.results.spi.DomainResultCreationState;
import org.hibernate.type.descriptor.java.JavaTypeDescriptor;
import org.hibernate.type.spi.TypeConfiguration;

/**
 * @author Steve Ebersole
 */
public class SqmRoot extends AbstractSqmFrom implements JpaRoot, DomainResultProducer {
	public SqmRoot(
			EntityDomainType entityType,
			String alias,
			NodeBuilder nodeBuilder) {
		super( entityType, alias, nodeBuilder );
	}

	@Override
	public SqmPath getLhs() {
		// a root has no LHS
		return null;
	}

	@Override
	public SqmRoot findRoot() {
		return this;
	}

	@Override
	public EntityDomainType getReferencedPathSource() {
		return (EntityDomainType) super.getReferencedPathSource();
	}

	public String getEntityName() {
		return getReferencedPathSource().getHibernateEntityName();
	}

	@Override
	public JavaTypeDescriptor getJavaTypeDescriptor() {
		return getReferencedPathSource().getExpressableJavaTypeDescriptor();
	}

	@Override
	public String toString() {
		return getExplicitAlias() == null
				? getEntityName()
				: getEntityName() + " as " + getExplicitAlias();
	}

	@Override
	public  X accept(SemanticQueryWalker walker) {
		return walker.visitRootPath( this );
	}


	// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
	// JPA

	@Override
	public EntityDomainType getManagedType() {
		return getReferencedPathSource();
	}


	@Override
	public  JpaEntityJoin join(Class entityJavaType) {
		return join( nodeBuilder().getDomainModel().entity( entityJavaType ) );
	}

	@Override
	public  JpaEntityJoin join(EntityDomainType entity) {
		final SqmEntityJoin join = new SqmEntityJoin<>( entity, null, SqmJoinType.CROSS, this );
		//noinspection unchecked
		addSqmJoin( (SqmEntityJoin) join );
		return join;
	}

	@Override
	public EntityDomainType getModel() {
		return getReferencedPathSource();
	}

	@Override
	public  SqmTreatedRoot treatAs(Class treatJavaType) throws PathException {
		return (SqmTreatedRoot) treatAs( nodeBuilder().getDomainModel().entity( treatJavaType ) );
	}

	@Override
	public  SqmTreatedPath treatAs(EntityDomainType treatTarget) throws PathException {
		return new SqmTreatedRoot<>( this, treatTarget, nodeBuilder() );
	}

	@Override
	public void visitJdbcTypes(Consumer action, TypeConfiguration typeConfiguration) {
		final String entityName = getReferencedPathSource().getHibernateEntityName();
		final EntityPersister entityDescriptor = typeConfiguration.getSessionFactory()
				.getMetamodel()
				.getEntityDescriptor( entityName );
		entityDescriptor.visitSubParts(
				valueMapping -> valueMapping.visitJdbcTypes(
						action,
						Clause.IRRELEVANT,
						typeConfiguration
				),
				entityDescriptor
		);
	}


	@Override
	public DomainResult createDomainResult(
			String resultVariable,
			DomainResultCreationState creationState) {
		final String entityName = getReferencedPathSource().getHibernateEntityName();
		final EntityPersister entityDescriptor = creationState.getSqlAstCreationState()
				.getCreationContext()
				.getDomainModel()
				.getEntityDescriptor( entityName );
		return entityDescriptor.createDomainResult(
				getNavigablePath(),
				creationState.getSqlAstCreationState().getFromClauseAccess().findTableGroup( getNavigablePath() ),
				resultVariable,
				creationState
		);
	}

	@Override
	public void applySqlSelections(DomainResultCreationState creationState) {
		final String entityName = getReferencedPathSource().getHibernateEntityName();
		final EntityPersister entityDescriptor = creationState.getSqlAstCreationState()
				.getCreationContext()
				.getDomainModel()
				.getEntityDescriptor( entityName );
		entityDescriptor.applySqlSelections(
				getNavigablePath(),
				creationState.getSqlAstCreationState().getFromClauseAccess().findTableGroup( getNavigablePath() ),
				creationState
		);

	}
}