如何通过AnnotationProcessor访问TypeUse注释

时间:2019-03-18 09:29:23

标签: java annotation-processing

问题:

  1. 是否可以通过注释处理器访问带有@Target(ElementType.TYPE_USE)注释的元素?
  2. 是否可以通过注释处理器访问带注释的类型边界?

高度赞赏我错过的相关文档的链接。

上下文:

注释:

@Target(ElementType.TYPE_USE)
@Retention(RetentionPolicy.SOURCE)
public @interface TypeUseAnno {}

一个示例类:

public class SomeClass extends HashMap<@TypeUseAnno String, String> {}

处理器:

@SupportedSourceVersion(SourceVersion.RELEASE_8)
@SupportedAnnotationTypes("base.annotations.TypeUseAnno")
public class Processor extends AbstractProcessor {

    @Override
    public synchronized void init(ProcessingEnvironment processingEnv) {
        super.init(processingEnv);
        this.processingEnv.getMessager().printMessage(Diagnostic.Kind.WARNING, "Initialized.");
    }

    @Override
    public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
        this.processingEnv.getMessager().printMessage(Diagnostic.Kind.WARNING, "Invoked.");
        for (TypeElement annotation : annotations) {
            this.processingEnv.getMessager().printMessage(Diagnostic.Kind.WARNING, "" + roundEnv.getElementsAnnotatedWith(annotation));
        }
        return true;
    }
}

将以上SomeClassProcessor一起在类路径上进行编译将显示"Intialized"消息,但从未调用process(...)方法。 当方法参数上存在注释时,使用@Target(ElementType.PARAMETER)向处理器添加另一个注释会很好。如果方法参数用@TypeUseAnno注释,则过程将再次忽略该元素。

1 个答案:

答案 0 :(得分:5)

TYPE_USE注释有点棘手,因为编译器对待它们的方式与“旧用法”注释不同。

因此,正如您正确观察到的那样,它们不会传递给注释处理器,并且您的process()方法将永远不会收到它们。

那么如何在编译时使用它们?

在Java 8中,引入了这些注释,还引入了附加到Java编译的新方法。现在,您可以将侦听器附加到编译任务,并触发您自己对源代码的遍历。因此,访问批注的任务将分为两部分。

  1. 连接到编译器
  2. 实施分析仪

广告1。 在Java 8中有2个钩挂在编译器上的选项 1.使用新的编译器插件API(https://docs.oracle.com/javase/8/docs/jdk/api/javac/tree/com/sun/source/util/Plugin.html) 2.使用注释处理器

我没有太多使用选项#1,因为它需要明确地指定为javac参数。因此,我将介绍选项1:

您必须将TaskListener附加到正确的编译阶段。有多个阶段。下一个是唯一的一个,在此期间,您具有可访问的语法树,该语法树表示包括方法主体在内的完整源代码(请记住,TYPE_USE注释甚至可以在局部变量声明中使用。

@SupportedSourceVersion(SourceVersion.RELEASE_8)
public class EndProcessor extends AbstractProcessor {

    @Override
    public synchronized void init(ProcessingEnvironment env) {
        super.init(env);
        Trees trees = Trees.instance(env);
        JavacTask.instance(env).addTaskListener(new TaskListener() {

            @Override
            public void started(TaskEvent taskEvent) {
                // Nothing to do on task started event.
            }

            @Override
            public void finished(TaskEvent taskEvent) {
                if(taskEvent.getKind() == ANALYZE) {
                    new MyTreeScanner(trees).scan(taskEvent.getCompilationUnit(), null);
                }
            }

        });
    }

    @Override
    public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
        // We don't care about this method, as it will never be invoked for our annotation.
        return false;
    }
}

广告2。 现在MyTreeScanner可以扫描完整的源代码,并找到注释。无论您使用Plugin还是AnnotationProcessor方法,这都适用。这仍然很棘手。您必须实现TreeScanner,或通常扩展TreePathScanner。 这代表了一种访问者模式,您必须在其中进行适当的分析,以了解哪些元素是您要访问的。

让我们举一个简单的例子,它可以对局部变量声明做出某种反应(给我5分钟):

class MyTreeScanner extends TreePathScanner<Void, Void> {
    private final Trees trees;

    public MyTreeScanner(Trees trees) {
        this.trees = trees;
    }

    @Override
    public Void visitVariable(VariableTree tree, Void aVoid) {
        super.visitVariable(variableTree, aVoid);
        // This method might be invoked in case of
        //  1. method field definition
        //  2. method parameter
        //  3. local variable declaration
        // Therefore you have to filter out somehow what you don't need.
        if(tree.getKind() == Tree.Kind.VARIABLE) {
            Element variable = trees.getElement(trees.getPath(getCurrentPath().getCompilationUnit(), tree));
            MyUseAnnotation annotation = variable.getAnnotation(MyUseAnnotation.class);
            // Here you have your annotation.
            // You can process it now.
        }
        return aVoid;
    }
}

这是非常简短的介绍。对于真实的示例,您可以查看以下项目源代码: https://github.com/c0stra/fluent-api-end-check/tree/master/src/main/java/fluent/api/processors

在开发此类功能时进行良好的测试也很重要,因此您可以调试,反向工程并解决您将在该领域遇到的所有棘手问题;) 为此,您还可以在这里获得启发: https://github.com/c0stra/fluent-api-end-check/blob/master/src/test/java/fluent/api/EndProcessorTest.java

也许是我的最后一句话,因为javac确实使用了不同的注释,因此存在一些限制。例如。它不适合触发Java代码生成,因为编译器不会选择在此阶段创建的文件进行进一步编译。