This commit is contained in:
Lana Steuck 2015-01-29 15:35:04 -08:00
commit 9853c62a2b

View File

@ -135,7 +135,7 @@ public class SchemaGenerator extends AbstractProcessor {
private void filterClass(List<Reference> classes, Collection<? extends Element> elements) {
for (Element element : elements) {
if (element.getKind().equals(ElementKind.CLASS)) {
if (element.getKind().equals(ElementKind.CLASS) || element.getKind().equals(ElementKind.ENUM)) {
classes.add(new Reference((TypeElement) element, processingEnv));
filterClass(classes, ElementFilter.typesIn(element.getEnclosedElements()));
}