HHH-9966 - Improve schema tooling support for creating catalogs and schemas

Author: sebersole

hibernate-core/src/main/java/org/hibernate/tool/hbm2ddl/SchemaUpdate.java

@@ -32,7 +32,7 @@
 import org.hibernate.internal.CoreMessageLogger;
 import org.hibernate.internal.log.DeprecationLogger;
 import org.hibernate.service.ServiceRegistry;
-import org.hibernate.tool.schema.extract.internal.legacy.DatabaseInformationImpl;
+import org.hibernate.tool.schema.extract.internal.DatabaseInformationImpl;
 import org.hibernate.tool.schema.extract.spi.DatabaseInformation;
 import org.hibernate.tool.schema.internal.TargetDatabaseImpl;
 import org.hibernate.tool.schema.internal.TargetFileImpl;

hibernate-core/src/main/java/org/hibernate/tool/hbm2ddl/SchemaValidator.java

@@ -30,7 +30,7 @@
 import org.hibernate.internal.CoreMessageLogger;
 import org.hibernate.internal.log.DeprecationLogger;
 import org.hibernate.service.ServiceRegistry;
-import org.hibernate.tool.schema.extract.internal.legacy.DatabaseInformationImpl;
+import org.hibernate.tool.schema.extract.internal.DatabaseInformationImpl;
 import org.hibernate.tool.schema.extract.spi.DatabaseInformation;
 import org.hibernate.tool.schema.spi.SchemaManagementTool;
 

hibernate-core/src/main/java/org/hibernate/tool/schema/extract/internal/DatabaseInformationImpl.java

@@ -6,94 +6,150 @@
  */
 package org.hibernate.tool.schema.extract.internal;
 
+import java.sql.SQLException;
 import java.util.HashMap;
 import java.util.Map;
 
 import org.hibernate.boot.model.naming.Identifier;
 import org.hibernate.boot.model.relational.Namespace;
 import org.hibernate.boot.model.relational.QualifiedSequenceName;
 import org.hibernate.boot.model.relational.QualifiedTableName;
+import org.hibernate.engine.jdbc.connections.spi.JdbcConnectionAccess;
+import org.hibernate.engine.jdbc.env.spi.JdbcEnvironment;
+import org.hibernate.service.ServiceRegistry;
 import org.hibernate.tool.schema.extract.spi.DatabaseInformation;
 import org.hibernate.tool.schema.extract.spi.ExtractionContext;
+import org.hibernate.tool.schema.extract.spi.InformationExtractor;
 import org.hibernate.tool.schema.extract.spi.SequenceInformation;
 import org.hibernate.tool.schema.extract.spi.TableInformation;
 
 /**
+ * Access to information from the existing database.
+ *
  * @author Steve Ebersole
  */
 public class DatabaseInformationImpl implements DatabaseInformation, ExtractionContext.DatabaseObjectAccess {
-	private final Map<QualifiedTableName,TableInformation> tableInformationMap = new HashMap<QualifiedTableName, TableInformation>();
+	private final InformationExtractor extractor;
+	private final ExtractionContext extractionContext;
+
+	private final JdbcEnvironment jdbcEnvironment;
+
 	private final Map<QualifiedSequenceName,SequenceInformation> sequenceInformationMap = new HashMap<QualifiedSequenceName, SequenceInformation>();
 
-	public DatabaseInformationImpl() {
+	public DatabaseInformationImpl(
+			ServiceRegistry serviceRegistry,
+			JdbcEnvironment jdbcEnvironment,
+			JdbcConnectionAccess jdbcConnectionAccess,
+			Identifier defaultCatalogName,
+			Identifier defaultSchemaName) throws SQLException {
+		this.jdbcEnvironment = jdbcEnvironment;
+
+		this.extractionContext = new ExtractionContextImpl(
+				serviceRegistry,
+				jdbcEnvironment,
+				jdbcConnectionAccess,
+				this,
+				defaultCatalogName,
+				defaultSchemaName
+		);
+
+		// todo : make this pluggable
+		this.extractor = new InformationExtractorJdbcDatabaseMetaDataImpl( extractionContext );
+
+		// legacy code did initialize sequences...
+		initializeSequences();
 	}
 
-
-	// DatabaseInformation implementation ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+	private void initializeSequences() throws SQLException {
+		Iterable<SequenceInformation> itr = jdbcEnvironment.getDialect().getSequenceInformationExtractor().extractMetadata( extractionContext );
+		for ( SequenceInformation sequenceInformation : itr ) {
+			sequenceInformationMap.put(
+					// for now, follow the legacy behavior of storing just the
+					// unqualified sequence name.
+					new QualifiedSequenceName(
+							null,
+							null,
+							sequenceInformation.getSequenceName().getSequenceName()
+					),
+					sequenceInformation
+			);
+		}
+	}
 
 	@Override
 	public boolean schemaExists(Namespace.Name schema) {
-		return false;
+		return extractor.schemaExists( schema.getCatalog(), schema.getSchema() );
 	}
 
 	@Override
-	public TableInformation getTableInformation(Identifier catalogName, Identifier schemaName, Identifier tableName) {
-		return locateTableInformation( new QualifiedTableName( catalogName, schemaName, tableName ) );
+	public TableInformation getTableInformation(
+			Identifier catalogName,
+			Identifier schemaName,
+			Identifier tableName) {
+		return getTableInformation( new QualifiedTableName( catalogName, schemaName, tableName ) );
 	}
 
 	@Override
-	public TableInformation getTableInformation(Namespace.Name schemaName, Identifier tableName) {
-		return locateTableInformation( new QualifiedTableName( schemaName, tableName ) );
+	public TableInformation getTableInformation(
+			Namespace.Name schemaName,
+			Identifier tableName) {
+		return getTableInformation( new QualifiedTableName( schemaName, tableName ) );
 	}
 
 	@Override
-	public TableInformation getTableInformation(QualifiedTableName tableName) {
-		return locateTableInformation( tableName );
+	public TableInformation getTableInformation(QualifiedTableName qualifiedTableName) {
+		if ( qualifiedTableName.getObjectName() == null ) {
+			throw new IllegalArgumentException( "Passed table name cannot be null" );
+		}
+
+		return extractor.getTable(
+				qualifiedTableName.getCatalogName(),
+				qualifiedTableName.getSchemaName(),
+				qualifiedTableName.getTableName()
+		);
 	}
 
 	@Override
-	public SequenceInformation getSequenceInformation(Identifier catalogName, Identifier schemaName, Identifier sequenceName) {
-		return locateSequenceInformation( new QualifiedSequenceName( catalogName, schemaName, sequenceName ) );
+	public void registerTable(TableInformation tableInformation) {
 	}
 
 	@Override
-	public SequenceInformation getSequenceInformation(Namespace.Name schemaName, Identifier sequenceName) {
-		return locateSequenceInformation( new QualifiedSequenceName( schemaName, sequenceName ) );
+	public SequenceInformation getSequenceInformation(
+			Identifier catalogName,
+			Identifier schemaName,
+			Identifier sequenceName) {
+		return getSequenceInformation( new QualifiedSequenceName( catalogName, schemaName, sequenceName ) );
 	}
 
 	@Override
-	public SequenceInformation getSequenceInformation(QualifiedSequenceName sequenceName) {
-		return locateSequenceInformation( sequenceName );
+	public SequenceInformation getSequenceInformation(
+			Namespace.Name schemaName,
+			Identifier sequenceName) {
+		return getSequenceInformation( new QualifiedSequenceName( schemaName, sequenceName ) );
 	}
 
 	@Override
-	public boolean catalogExists(Identifier catalog) {
-		return false;
-	}
-
-
-	// RegisteredObjectAccess implementation ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
-
-	public void registerTableInformation(TableInformation tableInformation) {
-		tableInformationMap.put( tableInformation.getName(), tableInformation );
+	public SequenceInformation getSequenceInformation(QualifiedSequenceName qualifiedSequenceName) {
+		return locateSequenceInformation( qualifiedSequenceName );
 	}
 
-	public void registerSequenceInformation(SequenceInformation sequenceInformation) {
-		sequenceInformationMap.put( sequenceInformation.getSequenceName(), sequenceInformation );
+	@Override
+	public boolean catalogExists(Identifier catalog) {
+		return extractor.catalogExists( catalog );
 	}
 
 	@Override
 	public TableInformation locateTableInformation(QualifiedTableName tableName) {
-		return tableInformationMap.get( tableName );
+		return getTableInformation( tableName );
 	}
 
 	@Override
 	public SequenceInformation locateSequenceInformation(QualifiedSequenceName sequenceName) {
-		return sequenceInformationMap.get( sequenceName );
-	}
+		// again, follow legacy behavior
+		if ( sequenceName.getCatalogName() != null || sequenceName.getSchemaName() != null ) {
+			sequenceName = new QualifiedSequenceName( null, null, sequenceName.getSequenceName() );
+		}
 
-	@Override
-	public void registerTable(TableInformation tableInformation) {
-		tableInformationMap.put( tableInformation.getName(), tableInformation );
+		return sequenceInformationMap.get( sequenceName );
 	}
 }