feat: issue-32936 add support for TypeContributor and FunctionContributor

Author: NickBeginner

extensions/hibernate-orm/runtime/src/main/java/io/quarkus/hibernate/orm/runtime/PersistenceUnitUtil.java

@@ -36,10 +36,12 @@ public static Annotation qualifier(String persistenceUnitName) {
         }
     }
 
-    public static <T> InjectableInstance<T> singleExtensionInstanceForPersistenceUnit(Class<T> beanType,
+    public static <T> InjectableInstance<T> singleExtensionInstanceForPersistenceUnit(
+            Class<T> beanType,
             String persistenceUnitName,
-            Annotation... additionalQualifiers) {
-        InjectableInstance<T> instance = extensionInstanceForPersistenceUnit(beanType, persistenceUnitName,
+            Annotation... additionalQualifiers
+    ) {
+        InjectableInstance<T> instance = extensionInstancesForPersistenceUnit(beanType, persistenceUnitName,
                 additionalQualifiers);
         if (instance.isAmbiguous()) {
             List<String> ambiguousClassNames = instance.handlesStream().map(h -> h.getBean().getBeanClass().getCanonicalName())
@@ -52,8 +54,11 @@ public static <T> InjectableInstance<T> singleExtensionInstanceForPersistenceUni
         return instance;
     }
 
-    public static <T> InjectableInstance<T> extensionInstanceForPersistenceUnit(Class<T> beanType, String persistenceUnitName,
-            Annotation... additionalQualifiers) {
+    public static <T> InjectableInstance<T> extensionInstancesForPersistenceUnit(
+            Class<T> beanType,
+            String persistenceUnitName,
+            Annotation... additionalQualifiers
+    ) {
         if (additionalQualifiers.length == 0) {
             return Arc.container().select(beanType, new PersistenceUnitExtension.Literal(persistenceUnitName));
         } else {

extensions/hibernate-orm/runtime/src/main/java/io/quarkus/hibernate/orm/runtime/boot/FastBootMetadataBuilder.java

@@ -29,6 +29,7 @@
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.stream.Collectors;
 
+import io.quarkus.hibernate.orm.runtime.PersistenceUnitUtil;
 import jakarta.persistence.PersistenceException;
 import jakarta.persistence.PersistenceUnitTransactionType;
 
@@ -39,6 +40,8 @@
 import org.hibernate.boot.archive.scan.spi.Scanner;
 import org.hibernate.boot.beanvalidation.BeanValidationIntegrator;
 import org.hibernate.boot.internal.MetadataImpl;
+import org.hibernate.boot.model.FunctionContributor;
+import org.hibernate.boot.model.TypeContributor;
 import org.hibernate.boot.model.process.spi.ManagedResources;
 import org.hibernate.boot.model.process.spi.MetadataBuildingProcess;
 import org.hibernate.boot.registry.StandardServiceRegistry;
@@ -188,6 +191,9 @@ public FastBootMetadataBuilder(final QuarkusPersistenceUnitDefinition puDefiniti
 
         applyMetadataBuilderContributor();
 
+        applyTypeContributors();
+        applyFunctionContributors();
+
         // Unable to automatically handle:
         // AvailableSettings.ENHANCER_ENABLE_DIRTY_TRACKING,
         // AvailableSettings.ENHANCER_ENABLE_LAZY_INITIALIZATION,
@@ -663,6 +669,16 @@ private void applyMetadataBuilderContributor() {
         }
     }
 
+    private void applyTypeContributors() {
+        PersistenceUnitUtil.extensionInstancesForPersistenceUnit(TypeContributor.class, persistenceUnit.getName())
+                .stream().forEach(metamodelBuilder::applyTypes);
+    }
+
+    private void applyFunctionContributors() {
+        PersistenceUnitUtil.extensionInstancesForPersistenceUnit(FunctionContributor.class, persistenceUnit.getName())
+                .stream().forEach(metamodelBuilder::applyFunctions);
+    }
+
     @SuppressWarnings("unchecked")
     private <T> T loadSettingInstance(String settingName, Object settingValue, Class<T> clazz) {
         T instance = null;