From df92538ab3d83da9839f08b28fc8a67317565463 Mon Sep 17 00:00:00 2001
From: Trygve Laugstøl <trygvis@inamo.no>
Date: Sat, 3 Aug 2013 20:21:19 +0200
Subject: wip

---
 .../src/main/java/io/trygvis/container/compiler/MyProcessor.java  | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'container-compiler-plugin/src/main/java/io/trygvis/container/compiler/MyProcessor.java')

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);
                 }
             }
         }
-- 
cgit v1.2.3