diff --git a/querydsl-apt/src/main/java/com/mysema/query/apt/APTTypeModelFactory.java b/querydsl-apt/src/main/java/com/mysema/query/apt/APTTypeModelFactory.java index d48a29999..83d2d5e01 100644 --- a/querydsl-apt/src/main/java/com/mysema/query/apt/APTTypeModelFactory.java +++ b/querydsl-apt/src/main/java/com/mysema/query/apt/APTTypeModelFactory.java @@ -223,14 +223,14 @@ public class APTTypeModelFactory implements TypeVisitor { if (t.asElement() != null && t.asElement() instanceof TypeElement){ TypeElement typeElement = (TypeElement)t.asElement(); switch(typeElement.getKind()){ + case ENUM: case CLASS: return createClassType(t, typeElement, p); case INTERFACE: return createInterfaceType(t, typeElement, p); - case ENUM: return create(typeElement, TypeCategory.SIMPLE, p, t.getTypeArguments()); + default: throw new IllegalArgumentException("Illegal type " + typeElement); } }else{ throw new IllegalArgumentException("Unsupported element type " + t.asElement()); } - return null; } private TypeModel createInterfaceType(DeclaredType t, TypeElement typeElement, Elements p) { diff --git a/querydsl-apt/src/main/java/com/mysema/query/apt/EntityElementVisitor.java b/querydsl-apt/src/main/java/com/mysema/query/apt/EntityElementVisitor.java index a665b5901..3dddb8de4 100644 --- a/querydsl-apt/src/main/java/com/mysema/query/apt/EntityElementVisitor.java +++ b/querydsl-apt/src/main/java/com/mysema/query/apt/EntityElementVisitor.java @@ -185,8 +185,7 @@ public final class EntityElementVisitor extends SimpleElementVisitor6