如何为ANTLR访问者方法添加参数以适配延续传递风格（CPS）

一、修改ANTLR生成逻辑（不推荐，门槛高） #

ANTLR默认通过StringTemplate模板生成访问器/监听器代码，要给visit方法添加额外参数，需要修改对应的模板文件：

• 找到ANTLR对应语言的模板（比如Java是Java.stg，C#是CSharp.stg）
• 定位到访问器方法的模板定义（比如visitRule模板），修改方法签名，添加Continuation cnt参数
• 使用修改后的模板重新生成解析器和访问器代码

注意：这种方法需要熟悉ANTLR的模板系统，且修改后无法直接使用官方ANTLR Jar，维护成本高，一般不建议。

二、无需修改生成代码的适配方案（推荐） #

1. 自定义访问器基类，维护延续上下文 #

编写一个包装ANTLR生成的访问器的基类，将延续参数存储为实例变量，对外暴露带延续参数的visit方法：

public abstract class CPSBaseVisitor<T> extends YourLangBaseVisitor<T> {
    private Continuation currentCont;

    // 对外调用的入口方法，传入上下文和延续
    public T visitWithCont(ParserRuleContext ctx, Continuation cont) {
        Continuation oldCont = currentCont;
        currentCont = cont;
        try {
            return super.visit(ctx);
        } finally {
            // 恢复之前的延续，避免递归时污染上下文
            currentCont = oldCont;
        }
    }

    // 子类可直接获取当前延续
    protected Continuation getCurrentCont() {
        return currentCont;
    }
}

然后在具体的访问器实现中使用：

public class YourLangCPSVisitor extends CPSBaseVisitor<Value> {
    @Override
    public Value visitStart(StartContext ctx) {
        Continuation cont = getCurrentCont();
        // 执行CPS风格的求值逻辑，比如将表达式结果传给cont
        Value exprVal = visit(ctx.expr());
        return cont.apply(exprVal);
    }
}

2. 让访问器返回延续函数（纯CPS实现） #

直接将每个visit方法的返回值定义为接受延续的函数，完全在访问器层面实现CPS转换，无需额外参数：

public class YourLangCPSVisitor extends YourLangBaseVisitor<Function<Continuation, Value>> {
    @Override
    public Function<Continuation, Value> visitStart(StartContext ctx) {
        // 返回一个接受初始延续的函数
        return cont -> visit(ctx.expr()).apply(cont);
    }

    @Override
    public Function<Continuation, Value> visitAdd(AddContext ctx) {
        // 先求值左表达式，再求值右表达式，最后应用延续
        return cont -> {
            Function<Continuation, Value> leftThunk = visit(ctx.left());
            return leftThunk.apply(leftVal -> {
                Function<Continuation, Value> rightThunk = visit(ctx.right());
                return rightThunk.apply(rightVal -> cont.apply(leftVal.add(rightVal)));
            });
        };
    }
}

调用方式：

YourLangCPSVisitor visitor = new YourLangCPSVisitor();
// 传入初始延续（比如打印结果）
visitor.visit(startCtx).apply(val -> {
    System.out.println("求值结果：" + val);
    return val;
});

3. 监听器模式下的延续栈维护 #

如果使用监听器而非访问器，可以维护一个延续栈，在进入节点时推入对应延续，离开时弹出：

public class YourLangCPSListener extends YourLangBaseListener {
    private Deque<Continuation> contStack = new ArrayDeque<>();
    // 存储节点的求值结果，用于后续传递给延续
    private Map<ParserRuleContext, Value> resultMap = new HashMap<>();

    public void enterStart(StartContext ctx) {
        // 推入初始延续
        contStack.push(val -> {
            System.out.println("最终结果：" + val);
            return val;
        });
    }

    public void exitAdd(AddContext ctx) {
        Continuation cont = contStack.pop();
        Value leftVal = resultMap.get(ctx.left());
        Value rightVal = resultMap.get(ctx.right());
        // 将结果传给延续，并将延续的结果存入当前节点的结果映射
        resultMap.put(ctx, cont.apply(leftVal.add(rightVal)));
    }
}

这种方式适合事件驱动的遍历，但需要手动管理结果和延续的关联，逻辑相对复杂。

内容的提问来源于stack exchange，提问作者Plegeus