java - 从 Java Annotation Processor 访问源代码

Question

我正在尝试从 Java 注释处理器中访问某种类型的实际原始源代码。这有可能吗？谢谢!

Answer 1

我遇到了一个问题，我必须访问一些源代码(非字符串/非原始常量的初始化代码)并通过 Compiler Tree API 访问源代码解决了这个问题.

这是一般的配方:

<强>1。创建自定义 TreePathScanner:

private static class CodeAnalyzerTreeScanner extends TreePathScanner<Object, Trees> {

private String fieldName;

private String fieldInitializer;

public void setFieldName(String fieldName) {
    this.fieldName = fieldName;
}

public String getFieldInitializer() {
    return this.fieldInitializer;
}

@Override
public Object visitVariable(VariableTree variableTree, Trees trees) {
    if (variableTree.getName().toString().equals(this.fieldName)) {
        this.fieldInitializer = variableTree.getInitializer().toString();
    }

    return super.visitVariable(variableTree, trees);
}

<强>2。在您的 AbstractProcessor 中，通过覆盖 init 方法保存对当前编译树的引用:

@Override
public void init(ProcessingEnvironment pe) {
    super.init(pe);
    this.trees = Trees.instance(pe);
}

<强>3。获取 VariableElement 的初始化源代码(在您的例子中是一个枚举):

// assuming theClass is a javax.lang.model.element.Element reference
// assuming theField is a javax.lang.model.element.VariableElement reference
String fieldName = theField.getSimpleName().toString();
CodeAnalyzerTreeScanner codeScanner = new CodeAnalyzerTreeScanner();
TreePath tp = this.trees.getPath(theClass);

codeScanner.setFieldName(fieldName);
codeScanner.scan(tp, this.trees);
String fieldInitializer = codeScanner.getFieldInitializer();

就是这样!最后，fieldInitiliazer 变量将包含用于初始化常量的确切代码行。通过一些调整，您应该能够使用相同的配方来访问源代码树中其他元素类型的源代码(即方法、包声明等)

有关更多阅读和示例，请阅读此 article: Source Code Analysis Using Java 6 APIs .