org.hibernate.query.sqm.consume.multitable.internal.StandardIdTableSupport 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.Alpha3

/*
 * 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.consume.multitable.internal;

import org.hibernate.boot.spi.SessionFactoryOptions;
import org.hibernate.engine.jdbc.env.spi.JdbcEnvironment;
import org.hibernate.engine.jdbc.spi.JdbcServices;
import org.hibernate.metamodel.model.domain.spi.EntityTypeDescriptor;
import org.hibernate.metamodel.model.relational.spi.PhysicalTable;
import org.hibernate.naming.Identifier;
import org.hibernate.query.sqm.consume.multitable.spi.idtable.IdTable;
import org.hibernate.query.sqm.consume.multitable.spi.idtable.IdTableManagementTransactionality;
import org.hibernate.query.sqm.consume.multitable.spi.idtable.IdTableSupport;
import org.hibernate.tool.schema.spi.Exporter;

/**
 * @author Steve Ebersole
 */
public class StandardIdTableSupport implements IdTableSupport {
	private final Exporter idTableExporter;
	private IdTableManagementTransactionality tableManagementTransactionality;

	public StandardIdTableSupport(Exporter idTableExporter) {
		this.idTableExporter = idTableExporter;
	}

	public StandardIdTableSupport(Exporter idTableExporter, IdTableManagementTransactionality tableManagementTransactionality) {
		this(idTableExporter);
		this.tableManagementTransactionality = tableManagementTransactionality;
	}

	@Override
	public Identifier determineIdTableName(EntityTypeDescriptor entityDescriptor, SessionFactoryOptions sessionFactoryOptions) {
		return determineIdTableName(
				entityDescriptor,
				sessionFactoryOptions.getServiceRegistry().getService( JdbcServices.class ).getJdbcEnvironment()
		);
	}

	protected Identifier determineIdTableName(
			EntityTypeDescriptor entityDescriptor,
			JdbcEnvironment jdbcEnvironment) {
		final String idTableNameBase = determineIdTableNameBase( entityDescriptor, jdbcEnvironment );
		final String idTableName = determineIdTableName( idTableNameBase );

		return jdbcEnvironment.getIdentifierHelper().toIdentifier( idTableName );
	}

	private String determineIdTableNameBase(EntityTypeDescriptor entityDescriptor, JdbcEnvironment jdbcEnvironment) {
		if ( entityDescriptor.getPrimaryTable() instanceof PhysicalTable ) {
			return ( (PhysicalTable) entityDescriptor.getPrimaryTable() ).getTableName().render( jdbcEnvironment.getDialect() );
		}
		else {
			return entityDescriptor.getJpaEntityName();
		}
	}

	protected String determineIdTableName(String baseName) {
		return "HT_" + baseName;
	}

	@Override
	public Exporter getIdTableExporter() {
		return idTableExporter;
	}

	@Override
	public IdTableManagementTransactionality geIdTableManagementTransactionality() {
		return tableManagementTransactionality;
	}
}