table partitioning tests

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

@@ -4614,7 +4614,10 @@ public void augmentRecognizedTableTypes(List<String> tableTypesList) {
 	}
 
 	/**
-	 * Does is dialect support {@code partition by}?
+	 * Does is dialect support {@code partition by} in window
+	 * functions?
+	 *
+	 * @apiNote This has nothing to do with table partitioning.
 	 *
 	 * @since 5.2
 	 */
@@ -4624,10 +4627,14 @@ public boolean supportsPartitionBy() {
 
 	/**
 	 * Does this dialect require that the columns listed in
-	 * {@code partition by} also occur in the primary key?
+	 * {@code partition by} also occur in the primary key,
+	 * when defining table partitioning?
+	 *
+	 * @apiNote This has nothing to do with window functions.
 	 *
 	 * @since 7.1
 	 */
+	@Incubating
 	public boolean addPartitionKeyToPrimaryKey() {
 		return false;
 	}

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

@@ -676,6 +676,11 @@ public String getCurrentTimestampSelectString() {
 
 	// Overridden informational metadata ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 
+	@Override
+	public boolean addPartitionKeyToPrimaryKey() {
+		return true;
+	}
+
 	@Override
 	public boolean supportsResultSetPositionQueryMethodsOnForwardOnlyCursor() {
 		return false;

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

@@ -579,6 +579,11 @@ public boolean supportsRowValueConstructorSyntaxInInList() {
 		return false;
 	}
 
+	@Override
+	public boolean addPartitionKeyToPrimaryKey() {
+		return true;
+	}
+
 	private static class JTDSSchemaNameResolver implements SchemaNameResolver {
 		@Override
 		public String resolveSchemaName(Connection connection, Dialect dialect) throws SQLException {

hibernate-core/src/test/java/org/hibernate/orm/test/sql/partition/Db2PartitionedTableTest.java

@@ -0,0 +1,49 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.orm.test.sql.partition;
+
+import jakarta.persistence.Entity;
+import jakarta.persistence.Id;
+import jakarta.persistence.Table;
+import org.hibernate.annotations.PartitionKey;
+import org.hibernate.dialect.DB2Dialect;
+import org.hibernate.testing.orm.junit.EntityManagerFactoryScope;
+import org.hibernate.testing.orm.junit.Jpa;
+import org.hibernate.testing.orm.junit.RequiresDialect;
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+
+@RequiresDialect(DB2Dialect.class)
+@Jpa(annotatedClasses = Db2PartitionedTableTest.Partitioned.class)
+class Db2PartitionedTableTest {
+	@Test void test(EntityManagerFactoryScope scope) {
+		scope.inTransaction( session -> {
+			Partitioned partitioned = new Partitioned();
+			partitioned.id = 1L;
+			partitioned.pid = 500L;
+			session.persist( partitioned );
+		} );
+		scope.inTransaction( session -> {
+			Partitioned partitioned = session.find( Partitioned.class, 1L );
+			assertNotNull( partitioned );
+			partitioned.text = "updated";
+		} );
+	}
+	@Entity
+	@Table(name = "db2parts",
+		options =
+			"""
+			PARTITION BY RANGE (pid) (
+				STARTING FROM (0) ENDING AT (1000),
+				ENDING AT (2000)
+			)
+			""")
+	static class Partitioned {
+		@Id Long id;
+		@PartitionKey Long pid;
+		String text = "";
+	}
+}

hibernate-core/src/test/java/org/hibernate/orm/test/sql/partition/InformixPartitionedTableTest.java

@@ -0,0 +1,49 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.orm.test.sql.partition;
+
+import jakarta.persistence.Entity;
+import jakarta.persistence.Id;
+import jakarta.persistence.Table;
+import org.hibernate.annotations.PartitionKey;
+import org.hibernate.community.dialect.InformixDialect;
+import org.hibernate.testing.orm.junit.EntityManagerFactoryScope;
+import org.hibernate.testing.orm.junit.Jpa;
+import org.hibernate.testing.orm.junit.RequiresDialect;
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+
+@RequiresDialect(InformixDialect.class)
+@Jpa(annotatedClasses = InformixPartitionedTableTest.Partitioned.class)
+class InformixPartitionedTableTest {
+	@Test void test(EntityManagerFactoryScope scope) {
+		scope.inTransaction( session -> {
+			Partitioned partitioned = new Partitioned();
+			partitioned.id = 1L;
+			partitioned.pid = 500L;
+			session.persist( partitioned );
+		} );
+		scope.inTransaction( session -> {
+			Partitioned partitioned = session.find( Partitioned.class, 1L );
+			assertNotNull( partitioned );
+			partitioned.text = "updated";
+		} );
+	}
+	@Entity
+	@Table(name = "infoparts",
+		options =
+			"""
+			FRAGMENT BY RANGE (pid)
+			INTERVAL (1000) STORE IN (rootdbs)
+				PARTITION p1 VALUES < 1000 IN rootdbs,
+				PARTITION p2 VALUES < 2000 IN rootdbs
+			""")
+	static class Partitioned {
+		@Id Long id;
+		@PartitionKey Long pid;
+		String text = "";
+	}
+}

hibernate-core/src/test/java/org/hibernate/orm/test/sql/partition/MySQLPartitionedTableTest.java

@@ -0,0 +1,51 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.orm.test.sql.partition;
+
+import jakarta.persistence.Entity;
+import jakarta.persistence.Id;
+import jakarta.persistence.Table;
+import org.hibernate.annotations.PartitionKey;
+import org.hibernate.dialect.MariaDBDialect;
+import org.hibernate.dialect.MySQLDialect;
+import org.hibernate.testing.orm.junit.EntityManagerFactoryScope;
+import org.hibernate.testing.orm.junit.Jpa;
+import org.hibernate.testing.orm.junit.RequiresDialect;
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+
+@RequiresDialect(MySQLDialect.class)
+@RequiresDialect(MariaDBDialect.class)
+@Jpa(annotatedClasses = MySQLPartitionedTableTest.Partitioned.class)
+class MySQLPartitionedTableTest {
+	@Test void test(EntityManagerFactoryScope scope) {
+		scope.inTransaction( session -> {
+			Partitioned partitioned = new Partitioned();
+			partitioned.id = 1L;
+			partitioned.pid = 500L;
+			session.persist( partitioned );
+		} );
+		scope.inTransaction( session -> {
+			Partitioned partitioned = session.find( Partitioned.class, 1L );
+			assertNotNull( partitioned );
+			partitioned.text = "updated";
+		} );
+	}
+	@Entity
+	@Table(name = "myparts",
+		options =
+			"""
+			PARTITION BY RANGE (pid) (
+				PARTITION p1 VALUES LESS THAN (1000),
+				PARTITION p2 VALUES LESS THAN (2000)
+			)
+			""")
+	static class Partitioned {
+		@Id Long id;
+		@PartitionKey Long pid;
+		String text = "";
+	}
+}

hibernate-core/src/test/java/org/hibernate/orm/test/sql/partition/OraclePartitionedTableTest.java

@@ -0,0 +1,49 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.orm.test.sql.partition;
+
+import jakarta.persistence.Entity;
+import jakarta.persistence.Id;
+import jakarta.persistence.Table;
+import org.hibernate.annotations.PartitionKey;
+import org.hibernate.dialect.OracleDialect;
+import org.hibernate.testing.orm.junit.EntityManagerFactoryScope;
+import org.hibernate.testing.orm.junit.Jpa;
+import org.hibernate.testing.orm.junit.RequiresDialect;
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+
+@RequiresDialect(OracleDialect.class)
+@Jpa(annotatedClasses = OraclePartitionedTableTest.Partitioned.class)
+class OraclePartitionedTableTest {
+	@Test void test(EntityManagerFactoryScope scope) {
+		scope.inTransaction( session -> {
+			Partitioned partitioned = new Partitioned();
+			partitioned.id = 1L;
+			partitioned.pid = 500L;
+			session.persist( partitioned );
+		} );
+		scope.inTransaction( session -> {
+			Partitioned partitioned = session.find( Partitioned.class, 1L );
+			assertNotNull( partitioned );
+			partitioned.text = "updated";
+		} );
+	}
+	@Entity
+	@Table(name = "oraparts",
+		options =
+			"""
+			PARTITION BY RANGE (pid) (
+				PARTITION p1 VALUES LESS THAN (1000),
+				PARTITION p2 VALUES LESS THAN (2000)
+			)
+			""")
+	static class Partitioned {
+		@Id Long id;
+		@PartitionKey Long pid;
+		String text = "";
+	}
+}

hibernate-core/src/test/java/org/hibernate/orm/test/sql/partition/PostgresPartitionedTableTest.java

@@ -0,0 +1,54 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.orm.test.sql.partition;
+
+import jakarta.persistence.Entity;
+import jakarta.persistence.Id;
+import jakarta.persistence.Table;
+import org.hibernate.annotations.PartitionKey;
+import org.hibernate.cfg.SchemaToolingSettings;
+import org.hibernate.dialect.PostgreSQLDialect;
+import org.hibernate.testing.orm.junit.EntityManagerFactoryScope;
+import org.hibernate.testing.orm.junit.Jpa;
+import org.hibernate.testing.orm.junit.RequiresDialect;
+import org.hibernate.testing.orm.junit.Setting;
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+
+@RequiresDialect(PostgreSQLDialect.class)
+@Jpa(annotatedClasses = PostgresPartitionedTableTest.Partitioned.class,
+		integrationSettings = {
+				@Setting(name = SchemaToolingSettings.JAKARTA_HBM2DDL_CREATE_SCRIPT_SOURCE,
+						value = "org/hibernate/orm/test/sql/partition/postgrespartitions-create.sql"),
+				@Setting(name = SchemaToolingSettings.JAKARTA_HBM2DDL_CREATE_SOURCE,
+						value = "metadata-then-script"),
+				@Setting(name = SchemaToolingSettings.JAKARTA_HBM2DDL_DROP_SCRIPT_SOURCE,
+						value = "org/hibernate/orm/test/sql/partition/postgrespartitions-drop.sql"),
+				@Setting(name = SchemaToolingSettings.JAKARTA_HBM2DDL_DROP_SOURCE,
+						value = "script-then-metadata")})
+class PostgresPartitionedTableTest {
+	@Test void test(EntityManagerFactoryScope scope) {
+		scope.inTransaction( session -> {
+			Partitioned partitioned = new Partitioned();
+			partitioned.id = 1L;
+			partitioned.pid = 500L;
+			session.persist( partitioned );
+		} );
+		scope.inTransaction( session -> {
+			Partitioned partitioned = session.find( Partitioned.class, 1L );
+			assertNotNull( partitioned );
+			partitioned.text = "updated";
+		} );
+	}
+	@Entity
+	@Table(name = "pgparts",
+		options = "partition by range (pid)")
+	static class Partitioned {
+		@Id Long id;
+		@PartitionKey Long pid;
+		String text = "";
+	}
+}

hibernate-core/src/test/java/org/hibernate/orm/test/sql/partition/SQLServerPartitionedTableTest.java

@@ -0,0 +1,54 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.orm.test.sql.partition;
+
+import jakarta.persistence.Entity;
+import jakarta.persistence.Id;
+import jakarta.persistence.Table;
+import org.hibernate.annotations.PartitionKey;
+import org.hibernate.cfg.SchemaToolingSettings;
+import org.hibernate.dialect.SQLServerDialect;
+import org.hibernate.testing.orm.junit.EntityManagerFactoryScope;
+import org.hibernate.testing.orm.junit.Jpa;
+import org.hibernate.testing.orm.junit.RequiresDialect;
+import org.hibernate.testing.orm.junit.Setting;
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+
+@RequiresDialect(SQLServerDialect.class)
+@Jpa(annotatedClasses = SQLServerPartitionedTableTest.Partitioned.class,
+		integrationSettings = {
+				@Setting(name = SchemaToolingSettings.JAKARTA_HBM2DDL_CREATE_SCRIPT_SOURCE,
+						value = "org/hibernate/orm/test/sql/partition/sqlserverpartitions-create.sql"),
+				@Setting(name = SchemaToolingSettings.JAKARTA_HBM2DDL_CREATE_SOURCE,
+						value = "script-then-metadata"),
+				@Setting(name = SchemaToolingSettings.JAKARTA_HBM2DDL_DROP_SCRIPT_SOURCE,
+						value = "org/hibernate/orm/test/sql/partition/sqlserverpartitions-drop.sql"),
+				@Setting(name = SchemaToolingSettings.JAKARTA_HBM2DDL_DROP_SOURCE,
+						value = "metadata-then-script")})
+class SQLServerPartitionedTableTest {
+	@Test void test(EntityManagerFactoryScope scope) {
+		scope.inTransaction( session -> {
+			Partitioned partitioned = new Partitioned();
+			partitioned.id = 1L;
+			partitioned.pid = 500L;
+			session.persist( partitioned );
+		} );
+		scope.inTransaction( session -> {
+			Partitioned partitioned = session.find( Partitioned.class, 1L );
+			assertNotNull( partitioned );
+			partitioned.text = "updated";
+		} );
+	}
+	@Entity
+	@Table(name = "msparts",
+		options = "on partScheme(pid)")
+	static class Partitioned {
+		@Id Long id;
+		@PartitionKey Long pid;
+		String text = "";
+	}
+}