编译器树API:在注解处理器中修改源代码
利用javac的Tree API和FileManager,在编译期对源代码进行结构化分析、生成、甚至原地修改,实现自定义代码检查器或代码增强器。 · 难度:入门 · +10XP
编译器树API:在注解处理器中修改源代码
注解处理器不仅能生成新文件,还能通过Tree API(com.sun.source.tree)读写抽象语法树。本教程教你如何编写一个处理器,在编译时检测并自动插入日志代码,或修改方法参数(如添加@NonNull检查),并利用TreePathScanner和Translator实现精细化的树变换。
// 简单的树扫描器
public class MethodLoggerProcessor extends AbstractProcessor {
@Override
public boolean process(Set extends TypeElement> annotations, RoundEnvironment roundEnv) {
for (Element element : roundEnv.getElementsAnnotatedWith(Loggable.class)) {
Trees trees = Trees.instance(processingEnv);
TreePath path = trees.getPath(element);
MethodTree methodTree = (MethodTree) path.getLeaf();
// 在方法体开头插入日志语句
// 使用TreeMaker创建新语句
}
return true;
}
}