#58 - Add support for @EbeanComponent (to register other types like B…

…eanPersistController, BeanPersistListener etc)
ebean-orm-deprecated · Jun 26, 2020 · 73738e9 · 73738e9
1 parent c4aa14d
commit 73738e9
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/src/main/java/io/ebean/querybean/generator/Constants.java b/src/main/java/io/ebean/querybean/generator/Constants.java
@@ -14,6 +14,7 @@ interface Constants {
   String ENTITY = "javax.persistence.Entity";
   String EMBEDDABLE = "javax.persistence.Embeddable";
   String CONVERTER = "javax.persistence.Converter";
+  String EBEAN_COMPONENT = "io.ebean.annotation.EbeanComponent";
 
   String DBARRAY = "io.ebean.annotation.DbArray";
   String DBJSON = "io.ebean.annotation.DbJson";

diff --git a/src/main/java/io/ebean/querybean/generator/ProcessingContext.java b/src/main/java/io/ebean/querybean/generator/ProcessingContext.java
@@ -113,6 +113,10 @@ TypeElement converterAnnotation() {
     return elementUtils.getTypeElement(CONVERTER);
   }
 
+  TypeElement componentAnnotation() {
+    return elementUtils.getTypeElement(EBEAN_COMPONENT);
+  }
+
   private String generatedAnnotation(boolean jdk8) {
     if (jdk8) {
       return isTypeAvailable(GENERATED_8) ? GENERATED_8 : null;

diff --git a/src/main/java/io/ebean/querybean/generator/Processor.java b/src/main/java/io/ebean/querybean/generator/Processor.java
@@ -32,6 +32,7 @@ public Set<String> getSupportedAnnotationTypes() {
     annotations.add(ENTITY);
     annotations.add(EMBEDDABLE);
     annotations.add(CONVERTER);
+    annotations.add(EBEAN_COMPONENT);
     annotations.add(MODULEINFO);
     return annotations;
   }
@@ -74,6 +75,9 @@ private void processOthers(RoundEnvironment roundEnv) {
     for (Element element : roundEnv.getElementsAnnotatedWith(processingContext.converterAnnotation())) {
       processingContext.addOther(element);
     }
+    for (Element element : roundEnv.getElementsAnnotatedWith(processingContext.componentAnnotation())) {
+      processingContext.addOther(element);
+    }
   }
 
   private void writeModuleInfoBean() {