使用Antlr4插入额外的代码行

Question

目标是插入代码来监视Java同步块的进入和退出。

。

enteringSync();
synchronized(lockObj){
enteredSync();
   ...
leavingSync();
}
leftSync();

我最初的想法是实现进入/退出侦听器方法(在Java同步块周围添加子树)，然后打印结果AST.现在我意识到antlr4似乎不支持树修改，我应该考虑什么替代方案?

Answer 1

最好的解决方案是使用令牌流重写引擎，而不是操作解析树。书为例;http://amzn.com/1934356999。下面是将序列化标识符插入类体的代码片段。

public class InsertSerialIDListener extends JavaBaseListener {
    TokenStreamRewriter rewriter;
    public InsertSerialIDListener(TokenStream tokens) {
        rewriter = new TokenStreamRewriter(tokens);
    }
    @Override
    public void enterClassBody(JavaParser.ClassBodyContext ctx) {
        String field = "ntpublic static final long serialVersionUID = 1L;";       
        rewriter.insertAfter(ctx.start, field);
    }
}