Register hints for types exposed via PersistenceManagedTypes

Add binding hints for managed types and hints for
@EntityListeners, @IdClass and @Converter.

Closes gh-29096

spring-orm/src/test/java/org/springframework/orm/jpa/domain/Employee.java

@@ -0,0 +1,62 @@
+/*
+ * Copyright 2002-2022 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.orm.jpa.domain;
+
+import jakarta.persistence.Column;
+import jakarta.persistence.Entity;
+import jakarta.persistence.Id;
+import jakarta.persistence.IdClass;
+
+@Entity
+@IdClass(EmployeeId.class)
+public class Employee {
+
+	@Id
+	@Column
+	private String name;
+
+	@Id
+	@Column
+	private String department;
+
+	private EmployeeLocation location;
+
+
+	public String getName() {
+		return name;
+	}
+
+	public void setName(String name) {
+		this.name = name;
+	}
+
+	public String getDepartment() {
+		return department;
+	}
+
+	public void setDepartment(String department) {
+		this.department = department;
+	}
+
+	public EmployeeLocation getLocation() {
+		return location;
+	}
+
+	public void setLocation(EmployeeLocation location) {
+		this.location = location;
+	}
+}

spring-orm/src/test/java/org/springframework/orm/jpa/domain/EmployeeId.java

@@ -0,0 +1,29 @@
+/*
+ * Copyright 2002-2022 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.orm.jpa.domain;
+
+import java.io.Serial;
+import java.io.Serializable;
+
+public class EmployeeId implements Serializable {
+
+	@Serial
+	private static final long serialVersionUID = 1L;
+
+	private String name;
+	private String department;
+}

spring-orm/src/test/java/org/springframework/orm/jpa/domain/EmployeeLocation.java

@@ -0,0 +1,30 @@
+/*
+ * Copyright 2002-2022 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.orm.jpa.domain;
+
+public class EmployeeLocation {
+
+	private String location;
+
+	public String getLocation() {
+		return location;
+	}
+
+	public void setLocation(String location) {
+		this.location = location;
+	}
+}

spring-orm/src/test/java/org/springframework/orm/jpa/domain/EmployeeLocationConverter.java

@@ -0,0 +1,42 @@
+/*
+ * Copyright 2002-2022 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.orm.jpa.domain;
+
+import jakarta.persistence.AttributeConverter;
+import jakarta.persistence.Converter;
+
+@Converter(autoApply = true)
+public class EmployeeLocationConverter implements AttributeConverter<EmployeeLocation, String> {
+
+	@Override
+	public String convertToDatabaseColumn(EmployeeLocation employeeLocation) {
+		if (employeeLocation != null) {
+			return employeeLocation.getLocation();
+		}
+		return null;
+	}
+
+	@Override
+	public EmployeeLocation convertToEntityAttribute(String data) {
+		if (data != null) {
+			EmployeeLocation employeeLocation = new EmployeeLocation();
+			employeeLocation.setLocation(data);
+			return employeeLocation;
+		}
+		return null;
+	}
+}

spring-orm/src/test/java/org/springframework/orm/jpa/persistenceunit/PersistenceManagedTypesBeanRegistrationAotProcessorTests.java

@@ -17,11 +17,15 @@
 package org.springframework.orm.jpa.persistenceunit;
 
 import java.util.function.BiConsumer;
+import java.util.function.Consumer;
 
 import javax.sql.DataSource;
 
 import org.junit.jupiter.api.Test;
 
+import org.springframework.aot.hint.MemberCategory;
+import org.springframework.aot.hint.RuntimeHints;
+import org.springframework.aot.hint.predicate.RuntimeHintsPredicates;
 import org.springframework.aot.test.generate.TestGenerationContext;
 import org.springframework.aot.test.generate.compile.Compiled;
 import org.springframework.aot.test.generate.compile.TestCompiler;
@@ -35,7 +39,12 @@ import org.springframework.core.io.ResourceLoader;
 import org.springframework.orm.jpa.JpaVendorAdapter;
 import org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean;
 import org.springframework.orm.jpa.domain.DriversLicense;
+import org.springframework.orm.jpa.domain.Employee;
+import org.springframework.orm.jpa.domain.EmployeeId;
+import org.springframework.orm.jpa.domain.EmployeeLocation;
+import org.springframework.orm.jpa.domain.EmployeeLocationConverter;
 import org.springframework.orm.jpa.domain.Person;
+import org.springframework.orm.jpa.domain.PersonListener;
 import org.springframework.orm.jpa.vendor.Database;
 import org.springframework.orm.jpa.vendor.HibernateJpaVendorAdapter;
 
@@ -46,6 +55,7 @@ import static org.mockito.Mockito.mock;
  * Tests for {@link PersistenceManagedTypesBeanRegistrationAotProcessor}.
  *
  * @author Stephane Nicoll
+ * @author Sebastien Deleuze
  */
 class PersistenceManagedTypesBeanRegistrationAotProcessorTests {
 
@@ -59,13 +69,38 @@ class PersistenceManagedTypesBeanRegistrationAotProcessorTests {
 			PersistenceManagedTypes persistenceManagedTypes = freshApplicationContext.getBean(
 					"persistenceManagedTypes", PersistenceManagedTypes.class);
 			assertThat(persistenceManagedTypes.getManagedClassNames()).containsExactlyInAnyOrder(
-					DriversLicense.class.getName(), Person.class.getName());
+					DriversLicense.class.getName(), Person.class.getName(), Employee.class.getName(),
+					EmployeeLocationConverter.class.getName());
 			assertThat(persistenceManagedTypes.getManagedPackages()).isEmpty();
 			assertThat(freshApplicationContext.getBean(
 					EntityManagerWithPackagesToScanConfiguration.class).scanningInvoked).isFalse();
 		});
 	}
 
+	@Test
+	void contributeHints() {
+		GenericApplicationContext context = new AnnotationConfigApplicationContext();
+		context.registerBean(EntityManagerWithPackagesToScanConfiguration.class);
+		contributeHints(context, hints -> {
+			assertThat(RuntimeHintsPredicates.reflection().onType(DriversLicense.class)
+					.withMemberCategories(MemberCategory.DECLARED_FIELDS)).accepts(hints);
+			assertThat(RuntimeHintsPredicates.reflection().onType(Person.class)
+					.withMemberCategories(MemberCategory.DECLARED_FIELDS)).accepts(hints);
+			assertThat(RuntimeHintsPredicates.reflection().onType(PersonListener.class)
+					.withMemberCategories(MemberCategory.INVOKE_DECLARED_CONSTRUCTORS, MemberCategory.INVOKE_PUBLIC_METHODS))
+					.accepts(hints);
+			assertThat(RuntimeHintsPredicates.reflection().onType(Employee.class)
+					.withMemberCategories(MemberCategory.DECLARED_FIELDS)).accepts(hints);
+			assertThat(RuntimeHintsPredicates.reflection().onType(EmployeeId.class)
+					.withMemberCategories(MemberCategory.DECLARED_FIELDS)).accepts(hints);
+			assertThat(RuntimeHintsPredicates.reflection().onType(EmployeeLocationConverter.class)
+					.withMemberCategories(MemberCategory.INVOKE_DECLARED_CONSTRUCTORS, MemberCategory.INVOKE_PUBLIC_METHODS))
+					.accepts(hints);
+			assertThat(RuntimeHintsPredicates.reflection().onType(EmployeeLocation.class)
+					.withMemberCategories(MemberCategory.DECLARED_FIELDS)).accepts(hints);
+		});
+	}
+
 
 	@SuppressWarnings("unchecked")
 	private void compile(GenericApplicationContext applicationContext,
@@ -86,6 +121,13 @@ class PersistenceManagedTypesBeanRegistrationAotProcessorTests {
 		return freshApplicationContext;
 	}
 
+	private void contributeHints(GenericApplicationContext applicationContext, Consumer<RuntimeHints> result) {
+		ApplicationContextAotGenerator generator = new ApplicationContextAotGenerator();
+		TestGenerationContext generationContext = new TestGenerationContext();
+		generator.processAheadOfTime(applicationContext, generationContext);
+		result.accept(generationContext.getRuntimeHints());
+	}
+
 	@Configuration(proxyBeanMethods = false)
 	public static class EntityManagerWithPackagesToScanConfiguration {
 

spring-orm/src/test/java/org/springframework/orm/jpa/persistenceunit/PersistenceManagedTypesScannerTests.java

@@ -22,6 +22,8 @@ import org.springframework.context.testfixture.index.CandidateComponentsTestClas
 import org.springframework.core.io.ClassPathResource;
 import org.springframework.core.io.DefaultResourceLoader;
 import org.springframework.orm.jpa.domain.DriversLicense;
+import org.springframework.orm.jpa.domain.Employee;
+import org.springframework.orm.jpa.domain.EmployeeLocationConverter;
 import org.springframework.orm.jpa.domain.Person;
 import org.springframework.orm.jpa.domain2.entity.User;
 
@@ -40,7 +42,8 @@ class PersistenceManagedTypesScannerTests {
 	void scanPackageWithOnlyEntities() {
 		PersistenceManagedTypes managedTypes = this.scanner.scan("org.springframework.orm.jpa.domain");
 		assertThat(managedTypes.getManagedClassNames()).containsExactlyInAnyOrder(
-				Person.class.getName(), DriversLicense.class.getName());
+				Person.class.getName(), DriversLicense.class.getName(), Employee.class.getName(),
+				EmployeeLocationConverter.class.getName());
 		assertThat(managedTypes.getManagedPackages()).isEmpty();
 	}