HHH-19521 Implement temporary and cte table derivation based on boot model

Author: beikov

hibernate-community-dialects/src/main/java/org/hibernate/community/dialect/SQLServerLegacyDialect.java

@@ -44,6 +44,8 @@
 import org.hibernate.dialect.sequence.SQLServer16SequenceSupport;
 import org.hibernate.dialect.sequence.SQLServerSequenceSupport;
 import org.hibernate.dialect.sequence.SequenceSupport;
+import org.hibernate.dialect.temptable.SQLServerLocalTemporaryTableStrategy;
+import org.hibernate.dialect.temptable.TemporaryTableStrategy;
 import org.hibernate.dialect.type.SQLServerCastingXmlArrayJdbcTypeConstructor;
 import org.hibernate.dialect.type.SQLServerCastingXmlJdbcType;
 import org.hibernate.dialect.unique.AlterTableUniqueIndexDelegate;
@@ -1149,19 +1151,14 @@ public void appendDateTimeLiteral(
 		}
 	}
 
+	@Override
+	public TemporaryTableStrategy getLocalTemporaryTableStrategy() {
+		return SQLServerLocalTemporaryTableStrategy.INSTANCE;
+	}
+
 	@Override
 	public String getCreateTemporaryTableColumnAnnotation(int sqlTypeCode) {
-		switch (sqlTypeCode) {
-			case Types.CHAR:
-			case Types.NCHAR:
-			case Types.VARCHAR:
-			case Types.NVARCHAR:
-			case Types.LONGVARCHAR:
-			case Types.LONGNVARCHAR:
-				return "collate database_default";
-			default:
-				return "";
-		}
+		return SQLServerLocalTemporaryTableStrategy.INSTANCE.getCreateTemporaryTableColumnAnnotation( sqlTypeCode );
 	}
 
 	@Override

hibernate-core/src/main/java/org/hibernate/dialect/Dialect.java

@@ -58,7 +58,6 @@
 import org.hibernate.dialect.temptable.LegacyTemporaryTableStrategy;
 import org.hibernate.dialect.temptable.PersistentTemporaryTableStrategy;
 import org.hibernate.dialect.temptable.StandardTemporaryTableExporter;
-import org.hibernate.dialect.temptable.TemporaryTable;
 import org.hibernate.dialect.temptable.TemporaryTableExporter;
 import org.hibernate.dialect.temptable.TemporaryTableKind;
 import org.hibernate.dialect.temptable.TemporaryTableStrategy;
@@ -3209,15 +3208,7 @@ public SqmMultiTableMutationStrategy getFallbackSqmMutationStrategy(
 	public SqmMultiTableInsertStrategy getFallbackSqmInsertStrategy(
 			EntityMappingType entityDescriptor,
 			RuntimeModelCreationContext runtimeModelCreationContext) {
-		return new PersistentTableInsertStrategy(
-				TemporaryTable.createEntityTable(
-						entityDescriptor,
-						name -> TemporaryTable.ENTITY_TABLE_PREFIX + name,
-						this,
-						runtimeModelCreationContext
-				),
-				runtimeModelCreationContext.getSessionFactory()
-		);
+		return new PersistentTableInsertStrategy( entityDescriptor, runtimeModelCreationContext );
 	}
 
 	// UDT support ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

hibernate-core/src/main/java/org/hibernate/dialect/SQLServerDialect.java

@@ -32,6 +32,8 @@
 import org.hibernate.dialect.sequence.SQLServerSequenceSupport;
 import org.hibernate.dialect.sequence.SequenceSupport;
 import org.hibernate.dialect.sql.ast.SQLServerSqlAstTranslator;
+import org.hibernate.dialect.temptable.SQLServerLocalTemporaryTableStrategy;
+import org.hibernate.dialect.temptable.TemporaryTableStrategy;
 import org.hibernate.dialect.type.SQLServerCastingXmlArrayJdbcTypeConstructor;
 import org.hibernate.dialect.type.SQLServerCastingXmlJdbcType;
 import org.hibernate.dialect.unique.AlterTableUniqueIndexDelegate;
@@ -1068,13 +1070,14 @@ public void appendDateTimeLiteral(
 		}
 	}
 
+	@Override
+	public TemporaryTableStrategy getLocalTemporaryTableStrategy() {
+		return SQLServerLocalTemporaryTableStrategy.INSTANCE;
+	}
+
 	@Override
 	public String getCreateTemporaryTableColumnAnnotation(int sqlTypeCode) {
-		return switch (sqlTypeCode) {
-			case Types.CHAR, Types.NCHAR, Types.VARCHAR, Types.NVARCHAR, Types.LONGVARCHAR, Types.LONGNVARCHAR ->
-					"collate database_default";
-			default -> "";
-		};
+		return SQLServerLocalTemporaryTableStrategy.INSTANCE.getCreateTemporaryTableColumnAnnotation( sqlTypeCode );
 	}
 
 	@Override

hibernate-core/src/main/java/org/hibernate/dialect/temptable/DB2GlobalTemporaryTableStrategy.java

@@ -11,6 +11,11 @@ public class DB2GlobalTemporaryTableStrategy extends StandardGlobalTemporaryTabl
 
 	public static final DB2GlobalTemporaryTableStrategy INSTANCE = new DB2GlobalTemporaryTableStrategy();
 
+	@Override
+	public String adjustTemporaryTableName(String desiredTableName) {
+		return "session." + desiredTableName;
+	}
+
 	@Override
 	public String getTemporaryTableCreateOptions() {
 		return "not logged";
@@ -20,4 +25,9 @@ public String getTemporaryTableCreateOptions() {
 	public String getTemporaryTableCreateCommand() {
 		return "declare global temporary table";
 	}
+
+	@Override
+	public boolean supportsTemporaryTablePrimaryKey() {
+		return false;
+	}
 }

hibernate-core/src/main/java/org/hibernate/dialect/temptable/SQLServerLocalTemporaryTableStrategy.java

@@ -0,0 +1,25 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.dialect.temptable;
+
+import org.hibernate.type.SqlTypes;
+
+/**
+ * SQL Server specific local temporary table strategy.
+ */
+public class SQLServerLocalTemporaryTableStrategy extends TransactSQLLocalTemporaryTableStrategy {
+
+	public static final SQLServerLocalTemporaryTableStrategy INSTANCE = new SQLServerLocalTemporaryTableStrategy();
+
+	@Override
+	public String getCreateTemporaryTableColumnAnnotation(int sqlTypeCode) {
+		return switch ( sqlTypeCode ) {
+			case SqlTypes.CHAR, SqlTypes.VARCHAR, SqlTypes.CLOB, SqlTypes.NCHAR, SqlTypes.NVARCHAR, SqlTypes.NCLOB ->
+					"collate database_default";
+			default -> "";
+		};
+	}
+
+}