1 files changed, 7 insertions, 1 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 713b948..297b9d8 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,9 +77,15 @@ public class MyProcessor implements Processor {
             System.out.println("claimed = " + claimed);
             return claimed;
         } catch (CompilerException e) {
+            e.printStackTrace(System.err);
             Messager messager = processingEnv.getMessager();
             messager.printMessage(Diagnostic.Kind.ERROR, e.getMessage(), e.element);
             return true;
+        } catch (InternalErrorException e) {
+            e.printStackTrace(System.err);
+            Messager messager = processingEnv.getMessager();
+            messager.printMessage(Diagnostic.Kind.ERROR, e.getMessage());
+            return true;
         } catch (RuntimeException e) {
             throw e;
         } catch (Exception e) {
@@ -116,7 +122,7 @@ public class MyProcessor implements Processor {
                     new LogHandler(processingEnv).processLog((TypeElement) element);
                 }
                 if (types.isSameType(entity.asType(), annotationType)) {
-                    entityHandler.processEntity((TypeElement) element);
+                    entityHandler.recordEntity((TypeElement) element);
                 }
             }
         }