wip

1 files changed, 8 insertions, 2 deletions

diff --git a/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/MyProcessor.java b/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/MyProcessor.java
index be0b446..89da439 100644
--- a/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/MyProcessor.java
+++ b/container-compiler-plugin/src/main/java/io/trygvis/container/compiler/MyProcessor.java
@@ -77,7 +77,7 @@ public class MyProcessor implements Processor {
 //            System.out.println("claimed = " + claimed);
             return claimed;
         } catch (CompilerException e) {
-            e.printStackTrace(System.err);
+//            e.printStackTrace(System.out);
             Messager messager = processingEnv.getMessager();
             messager.printMessage(Diagnostic.Kind.ERROR, e.getMessage(), e.element);
             return true;
@@ -122,7 +122,13 @@ public class MyProcessor implements Processor {
                     new LogHandler(processingEnv).processLog((TypeElement) element);
                 }
                 if (types.isSameType(entity.asType(), annotationType)) {
-                    entityHandler.recordEntity((TypeElement) element);
+                    try {
+                        entityHandler.recordEntity((TypeElement) element);
+                    } catch (CompilerException e) {
+//                        e.printStackTrace(System.out);
+                        Messager messager = processingEnv.getMessager();
+                        messager.printMessage(Diagnostic.Kind.ERROR, e.getMessage(), e.element);
+                    }
                 }
             }
         }