Процессор аннотаций

Question

У меня проблема с AnnotationProcessor.

Сначала мой исходный код:

@SupportedAnnotationTypes("*")
@SupportedSourceVersion(SourceVersion.RELEASE_8)
public class TreeAnnotationProcessor extends AbstractProcessor{

  private Trees trees;
  private Tree tree;

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

  @Override
  public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
    for (Element element : roundEnv.getRootElements()) {
      tree = trees.getTree(element);
    }

    return true;
  }

  public Tree getTree() {
    return tree;
  }
}

Этот аннотационный процессор собирает дерево компилятора. В этом процессоре все нормально. Если я вызову функцию getTree после процесса компиляции, дерево не будет завершено. Все дети Дерева (Узла) отсутствуют.

...
JavaCompiler.CompilationTask task = compiler.getTask(null, fileManager, null, null, null, compilationUnits);
TreeAnnotationProcessor treeAnnotationProcessor = new TreeAnnotationProcessor();
task.setProcessors(Collections.singletonList(treeAnnotationProcessor));
task.call();
Tree tree = treeAnnotationProcessor.getTree();
...

Спасибо за любую помощь.

Answer 1

Я нашел решение. Tree-интерфейс реализуется классом com.sun.tools.javac.tree.JCTree. Этот класс реализует метод клонирования. Когда я использую этот метод, клон завершается после процесса компиляции:

@SupportedAnnotationTypes("*")
@SupportedSourceVersion(SourceVersion.RELEASE_8)
public class TreeAnnotationProcessor extends AbstractProcessor{

  private Trees trees;
  private Tree tree;

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

  @Override
  public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
    for (Element element : roundEnv.getRootElements()) {
      tree = trees.getTree(element);

      try {
        Method cloneMethod = tree.getClass().getMethod("clone");
        Object cloneTree = cloneMethod.invoke(tree);
        this.tree = (Tree) cloneTree;
      } catch (NoSuchMethodException | IllegalAccessException | InvocationTargetException e) {
        e.printStackTrace();
      }
    }

    return true;
  }

  public Tree getTree() {
    return tree;
  }
}