org.hibernate.tool.schema.extract.internal.SequenceInformationExtractorMariaDBDatabaseImpl Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of hibernate-core-jakarta Show documentation

Hibernate O/RM implementation of the Jakarta Persistence specification

There is a newer version: 5.6.15.Final

Show newest version

/*
 * 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.util.ArrayList;
import java.util.Collections;
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;

/**
 * @author Vlad Mihalcea, Magnus Hagström
 */
public class SequenceInformationExtractorMariaDBDatabaseImpl extends SequenceInformationExtractorLegacyImpl {
	/**
	 * Singleton access
	 */
	public static final SequenceInformationExtractorMariaDBDatabaseImpl INSTANCE = new SequenceInformationExtractorMariaDBDatabaseImpl();

	// SQL to get metadata from individual sequence
	private static final String SQL_SEQUENCE_QUERY =
			"SELECT '%1$s' as sequence_name, minimum_value, maximum_value, start_value, increment, cache_size FROM %1$s ";

	private static final String UNION_ALL =
			"UNION ALL ";

	@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 List sequenceNames = extractionContext.getQueryResults( lookupSql, null, resultSet -> {
			final List sequences = new ArrayList<>();
			while ( resultSet.next() ) {
				sequences.add( resultSetSequenceName( resultSet ) );
			}
			return sequences;
		});

		if ( !sequenceNames.isEmpty() ) {
			StringBuilder sequenceInfoQueryBuilder = new StringBuilder();
			for ( String sequenceName : sequenceNames ) {
				if ( sequenceInfoQueryBuilder.length() > 0 ) {
					sequenceInfoQueryBuilder.append( UNION_ALL );
				}
				sequenceInfoQueryBuilder.append( String.format( SQL_SEQUENCE_QUERY, sequenceName ) );
			}
			return extractionContext.getQueryResults(
					sequenceInfoQueryBuilder.toString(),
					null,
					(ExtractionContext.ResultSetProcessor>) resultSet -> {
						final List sequenceInformationList = new ArrayList<>();
						final IdentifierHelper identifierHelper = extractionContext.getJdbcEnvironment()
								.getIdentifierHelper();

						while ( resultSet.next() ) {
							SequenceInformation sequenceInformation = new SequenceInformationImpl(
									new QualifiedSequenceName(
											null,
											null,
											identifierHelper.toIdentifier( resultSetSequenceName(resultSet) )
									),
									resultSetStartValueSize(resultSet),
									resultSetMinValue(resultSet),
									resultSetMaxValue(resultSet),
									resultSetIncrementValue(resultSet)
							);
							sequenceInformationList.add(sequenceInformation);
						}
						return sequenceInformationList;
					});
		}

		return Collections.emptyList();
	}

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

	@Override
	protected String sequenceCatalogColumn() {
		return null;
	}

	@Override
	protected String sequenceSchemaColumn() {
		return null;
	}

}