java - 使用Antlr4插入额外的代码行

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

即

enteringSync();
synchronized(lockObj){
enteredSync();

   ...

leavingSync();
}
leftSync();

我最初的想法是实现enter / exit侦听器方法（该方法在Java syncnizd块周围添加子树），然后打印出结果AST。现在我意识到antlr4似乎不支持树修改，我应该考虑哪些替代方案？

最佳答案

最好的解决方案是使用令牌流重写引擎，而不是操纵解析树。以书为例； 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 = "\n\tpublic static final long serialVersionUID = 1L;";
        rewriter.insertAfter(ctx.start, field);
    }
}