org.hibernate.tool.schema.extract.internal.SequenceInformationExtractorLegacyImpl 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.Alpha1
/*
 * 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 .
 */
package org.hibernate.tool.schema.extract.internal;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;

import org.hibernate.boot.model.relational.QualifiedSequenceName;
import org.hibernate.engine.jdbc.env.spi.IdentifierHelper;
import org.hibernate.tool.schema.extract.spi.ExtractionContext;
import org.hibernate.tool.schema.extract.spi.SequenceInformation;
import org.hibernate.tool.schema.extract.spi.SequenceInformationExtractor;

/**
 * @author Steve Ebersole
 */
public class SequenceInformationExtractorLegacyImpl implements SequenceInformationExtractor {
	/**
	 * Singleton access
	 */
	public static final SequenceInformationExtractorLegacyImpl INSTANCE = new SequenceInformationExtractorLegacyImpl();

	@Override
	public Iterable extractMetadata(ExtractionContext extractionContext) throws SQLException {
		final String lookupSql = extractionContext.getJdbcEnvironment().getDialect().getQuerySequencesString();

		// *should* never happen, but to be safe in the interest of performance...
		if ( lookupSql == null ) {
			return SequenceInformationExtractorNoOpImpl.INSTANCE.extractMetadata( extractionContext );
		}

		final IdentifierHelper identifierHelper = extractionContext.getJdbcEnvironment().getIdentifierHelper();
		final Statement statement = extractionContext.getJdbcConnection().createStatement();
		try {
			final ResultSet resultSet = statement.executeQuery( lookupSql );
			try {
				final List sequenceInformationList = new ArrayList<>();
				while ( resultSet.next() ) {
					sequenceInformationList.add(
							new SequenceInformationImpl(
									new QualifiedSequenceName(
											identifierHelper.toIdentifier(
												resultSetCatalogName( resultSet )
											),
											identifierHelper.toIdentifier(
													resultSetSchemaName( resultSet )
											),
											identifierHelper.toIdentifier(
													resultSetSequenceName( resultSet )
											)
									),
									resultSetStartValueSize( resultSet ),
									resultSetMinValue( resultSet ),
									resultSetMaxValue( resultSet ),
									resultSetIncrementValue( resultSet )
							)
					);
				}
				return sequenceInformationList;
			}
			finally {
				try {
					resultSet.close();
				}
				catch (SQLException ignore) {
				}
			}
		}
		finally {
			try {
				statement.close();
			}
			catch (SQLException ignore) {
			}
		}
	}

	protected String sequenceNameColumn() {
		return "sequence_name";
	}

	protected String sequenceCatalogColumn() {
		return "sequence_catalog";
	}

	protected String sequenceSchemaColumn() {
		return "sequence_schema";
	}

	protected String sequenceStartValueColumn() {
		return "start_value";
	}

	protected String sequenceMinValueColumn() {
		return "minimum_value";
	}

	protected String sequenceMaxValueColumn() {
		return "maximum_value";
	}

	protected String sequenceIncrementColumn() {
		return "increment";
	}

	protected String resultSetSequenceName(ResultSet resultSet) throws SQLException {
		return resultSet.getString( sequenceNameColumn() );
	}

	protected String resultSetCatalogName(ResultSet resultSet) throws SQLException {
		String column = sequenceCatalogColumn();
		return column != null ? resultSet.getString( column ) : null;
	}

	protected String resultSetSchemaName(ResultSet resultSet) throws SQLException {
		String column = sequenceSchemaColumn();
		return column != null ? resultSet.getString( column ) : null;
	}

	protected Long resultSetStartValueSize(ResultSet resultSet) throws SQLException {
		String column = sequenceStartValueColumn();
		return column != null ? resultSet.getLong( column ) : null;
	}

	protected Long resultSetMinValue(ResultSet resultSet) throws SQLException {
		String column = sequenceMinValueColumn();
		return column != null ? resultSet.getLong( column ) : null;
	}

	protected Long resultSetMaxValue(ResultSet resultSet) throws SQLException {
		String column = sequenceMaxValueColumn();
		return column != null ? resultSet.getLong( column ) : null;
	}

	protected Long resultSetIncrementValue(ResultSet resultSet) throws SQLException {
		String column = sequenceIncrementColumn();
		return column != null ? resultSet.getLong( column ) : null;
	}
}