通过Java Annotation Processor创建基类

时间:2014-09-18 18:02:03

标签: java annotations annotation-processing

我想使用注释处理器来生成要注释的类的基类。像这样:

@ClassGenerator(name="FakeClass") // Generates FakeClass
public class MyClass extends FakeClass {
    ...
}

这可能吗?或者我应该看看其他模式。

1 个答案:

答案 0 :(得分:0)

直接回答这个问题......当然,您只需要扫描ClassGenerator并生成缺少的课程。这是一个基本的工作处理器:

@SupportedAnnotationTypes("ClassGenerator")
@SupportedSourceVersion(SourceVersion.RELEASE_6)
public class BaseClassProcessor extends AbstractProcessor {

    private Elements elements;

    @Override
    public synchronized void init(ProcessingEnvironment processingEnv) {
        super.init(processingEnv);
        elements = processingEnv.getElementUtils();
    }

    @Override
    public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
        for (Element element : roundEnv.getElementsAnnotatedWith(ClassGenerator.class)) {
            TypeMirror mirror = ((TypeElement) element).getSuperclass();

            PackageElement pacakgeElement = elements.getPackageOf(element);
            String packageName = pacakgeElement.getQualifiedName().toString();
            String superClassName = ((DeclaredType) mirror).asElement().getSimpleName().toString();

            try {
                Writer file = processingEnv.getFiler().createSourceFile(packageName + "." + superClassName).openWriter();
                file.write(String.format("package %s; public class %s {}", packageName, superClassName));
                file.flush();
                file.close();
            } catch (IOException ex) {
                Logger.getLogger(BaseClassProcessor.class.getName()).log(Level.SEVERE, null, ex);
            }

        }
        return true;
    }
}

至于这种模式是否可行或最佳,您必须分析您的要求并确定这是否满足您的需求。