Test use of custom Hibernate id generators in native image

Author: yrodiere

integration-tests/jpa/src/main/java/io/quarkus/it/jpa/generatedvalue/EntityWithCustomGenericGeneratorReferencedAsClass.java

@@ -0,0 +1,15 @@
+package io.quarkus.it.jpa.generatedvalue;
+
+import jakarta.persistence.Entity;
+import jakarta.persistence.GeneratedValue;
+import jakarta.persistence.Id;
+
+import org.hibernate.annotations.GenericGenerator;
+
+@Entity
+public class EntityWithCustomGenericGeneratorReferencedAsClass {
+    @Id
+    @GeneratedValue(generator = "referenced-as-class")
+    @GenericGenerator(name = "referenced-as-class", type = MyCustomGenericGeneratorReferencedAsClass.class)
+    public String id;
+}

integration-tests/jpa/src/main/java/io/quarkus/it/jpa/generatedvalue/EntityWithCustomGenericGeneratorReferencedAsClassName.java

@@ -0,0 +1,15 @@
+package io.quarkus.it.jpa.generatedvalue;
+
+import jakarta.persistence.Entity;
+import jakarta.persistence.GeneratedValue;
+import jakarta.persistence.Id;
+
+import org.hibernate.annotations.GenericGenerator;
+
+@Entity
+public class EntityWithCustomGenericGeneratorReferencedAsClassName {
+    @Id
+    @GeneratedValue(generator = "referenced-as-class-name")
+    @GenericGenerator(name = "referenced-as-class-name", strategy = "io.quarkus.it.jpa.generatedvalue.MyCustomGenericGeneratorReferencedAsClassName")
+    public String id;
+}

integration-tests/jpa/src/main/java/io/quarkus/it/jpa/generatedvalue/EntityWithCustomIdGeneratorType.java

@@ -0,0 +1,13 @@
+package io.quarkus.it.jpa.generatedvalue;
+
+import jakarta.persistence.Entity;
+import jakarta.persistence.GeneratedValue;
+import jakarta.persistence.Id;
+
+@Entity
+public class EntityWithCustomIdGeneratorType {
+    @Id
+    @GeneratedValue
+    @MyCustomIdGeneratorAnnotation
+    public String id;
+}

integration-tests/jpa/src/main/java/io/quarkus/it/jpa/generatedvalue/EntityWithCustomValueGeneratorType.java

@@ -0,0 +1,15 @@
+package io.quarkus.it.jpa.generatedvalue;
+
+import jakarta.persistence.Entity;
+import jakarta.persistence.GeneratedValue;
+import jakarta.persistence.Id;
+
+@Entity
+public class EntityWithCustomValueGeneratorType {
+    @Id
+    @GeneratedValue
+    public Integer id;
+
+    @MyCustomValueGeneratorAnnotation
+    public String customGenerated;
+}

integration-tests/jpa/src/main/java/io/quarkus/it/jpa/generatedvalue/GeneratedValueResource.java

@@ -43,6 +43,30 @@ public String test() throws Exception {
                 e -> assertThat(e.generated).isNotNull(),
                 e -> assertThat(e.generatedColumn).isNotNull());
 
+        var entity2 = new EntityWithCustomIdGeneratorType();
+        assertThat(entity2.id).isNull();
+        em.persist(entity2);
+        em.flush();
+        assertThat(entity2.id).isEqualTo(MyCustomIdGenerator.STUB_VALUE);
+
+        var entity3 = new EntityWithCustomValueGeneratorType();
+        assertThat(entity3.customGenerated).isNull();
+        em.persist(entity3);
+        em.flush();
+        assertThat(entity3.customGenerated).isEqualTo(MyCustomValueGenerator.STUB_VALUE);
+
+        var entity4 = new EntityWithCustomGenericGeneratorReferencedAsClass();
+        assertThat(entity4.id).isNull();
+        em.persist(entity4);
+        em.flush();
+        assertThat(entity4.id).isEqualTo(MyCustomGenericGeneratorReferencedAsClass.STUB_VALUE);
+
+        var entity5 = new EntityWithCustomGenericGeneratorReferencedAsClassName();
+        assertThat(entity5.id).isNull();
+        em.persist(entity5);
+        em.flush();
+        assertThat(entity5.id).isEqualTo(MyCustomGenericGeneratorReferencedAsClassName.STUB_VALUE);
+
         return "OK";
     }
 }

integration-tests/jpa/src/main/java/io/quarkus/it/jpa/generatedvalue/MyCustomGenericGeneratorReferencedAsClass.java

@@ -0,0 +1,13 @@
+package io.quarkus.it.jpa.generatedvalue;
+
+import org.hibernate.engine.spi.SharedSessionContractImplementor;
+import org.hibernate.id.IdentifierGenerator;
+
+public class MyCustomGenericGeneratorReferencedAsClass implements IdentifierGenerator {
+    public static String STUB_VALUE = MyCustomGenericGeneratorReferencedAsClass.class.getName() + "_STUB_VALUE";
+
+    @Override
+    public Object generate(SharedSessionContractImplementor session, Object object) {
+        return STUB_VALUE;
+    }
+}

integration-tests/jpa/src/main/java/io/quarkus/it/jpa/generatedvalue/MyCustomGenericGeneratorReferencedAsClassName.java

@@ -0,0 +1,13 @@
+package io.quarkus.it.jpa.generatedvalue;
+
+import org.hibernate.engine.spi.SharedSessionContractImplementor;
+import org.hibernate.id.IdentifierGenerator;
+
+public class MyCustomGenericGeneratorReferencedAsClassName implements IdentifierGenerator {
+    public static String STUB_VALUE = MyCustomGenericGeneratorReferencedAsClassName.class.getName() + "_STUB_VALUE";
+
+    @Override
+    public Object generate(SharedSessionContractImplementor session, Object object) {
+        return STUB_VALUE;
+    }
+}

integration-tests/jpa/src/main/java/io/quarkus/it/jpa/generatedvalue/MyCustomIdGenerator.java

@@ -0,0 +1,13 @@
+package io.quarkus.it.jpa.generatedvalue;
+
+import org.hibernate.engine.spi.SharedSessionContractImplementor;
+import org.hibernate.id.IdentifierGenerator;
+
+public class MyCustomIdGenerator implements IdentifierGenerator {
+    public static String STUB_VALUE = MyCustomIdGenerator.class.getName() + "_STUB_VALUE";
+
+    @Override
+    public Object generate(SharedSessionContractImplementor session, Object object) {
+        return STUB_VALUE;
+    }
+}

integration-tests/jpa/src/main/java/io/quarkus/it/jpa/generatedvalue/MyCustomIdGeneratorAnnotation.java

@@ -0,0 +1,16 @@
+package io.quarkus.it.jpa.generatedvalue;
+
+import static java.lang.annotation.ElementType.FIELD;
+import static java.lang.annotation.ElementType.METHOD;
+import static java.lang.annotation.RetentionPolicy.RUNTIME;
+
+import java.lang.annotation.Retention;
+import java.lang.annotation.Target;
+
+import org.hibernate.annotations.IdGeneratorType;
+
+@IdGeneratorType(MyCustomIdGenerator.class)
+@Retention(RUNTIME)
+@Target({ FIELD, METHOD })
+public @interface MyCustomIdGeneratorAnnotation {
+}

integration-tests/jpa/src/main/java/io/quarkus/it/jpa/generatedvalue/MyCustomValueGenerator.java

@@ -0,0 +1,21 @@
+package io.quarkus.it.jpa.generatedvalue;
+
+import java.util.EnumSet;
+
+import org.hibernate.engine.spi.SharedSessionContractImplementor;
+import org.hibernate.generator.BeforeExecutionGenerator;
+import org.hibernate.generator.EventType;
+
+public class MyCustomValueGenerator implements BeforeExecutionGenerator {
+    public static String STUB_VALUE = MyCustomValueGenerator.class.getName() + "_STUB_VALUE";
+
+    @Override
+    public Object generate(SharedSessionContractImplementor session, Object owner, Object currentValue, EventType eventType) {
+        return STUB_VALUE;
+    }
+
+    @Override
+    public EnumSet<EventType> getEventTypes() {
+        return EnumSet.of(EventType.INSERT, EventType.UPDATE);
+    }
+}