如何有效地区分ANTLR4中规则中的不同令牌？

Question

expr    : expr (EQUALS | NOT_EQUALS) expr
        | literal;
literal : ...; // omitted here

EQUALS     : '=';
NOT_EQUALS : '!=';

在我的代码中，我想区分相等和不相等的大小写。我想知道如何有效地做到这一点。目前，我以以下方式实现访问者：

public Expression visitExpr(ExprContext ctx) {
    if (ctx.EQUALS() != null) {
        return EqualsExpression.construct(ctx.expr());
    } else if (ctx.NOT_EQUALS() != null) {
        return NotEqualsExpression.construct(ctx.expr());
    } else if (ctx.literal() != null) {
        return LiteralExpression.construct(ctx.literal());
    }
    throw new IllegalStateException();
}

我不确定这是否非常有效，因为equals()/not_equals()调用gettoken()，它基本上循环所有子级。这样做了多次，所以我不确定这是否明智。另外，我调用literal()两次。关于后者，我知道我可以缓存在一个局部变量中，但如果需要考虑多个子规则，这将很快导致相当丑陋的代码。

有没有办法更有效地做到这一点？基于某种令牌标识符或分支标识符的switch/case语句会更理想吗？

expr         : expr_eq | expr_not_eq | expr_literal
expr_eq      : expr EQUALS expr
expr_not_eq  : expr NOT_EQUALS expr
expr_literal : literal

public Expression visitExprEx(ExprEqContext ctx) {
    return EqualsExpression.construct(ctx.expr());
}

public Expression visitExprNotEq(ExprNotEqContext ctx) {
    return NotEqualsExpression.construct(ctx.expr());
}

public Expression visitExprLiteral(ExprLiteralContext ctx) {
    return LiteralExpression.construct(ctx.literal());
}

Answer 1

不要仅仅从代码中猜测性能。量一下！

子列表非常短，这意味着getToken在找到equals或not_equals之前只执行1-2个循环。代码的其他部分可能需要比查找更多的时间。

但是，如果您想要获得每一个性能片段，请避免使用方便的方法，而手动操作，因为您可以根据对语法的了解来优化访问。在这种特殊情况下，expr规则只能有一个或三个子规则。查看第一个变体，它可以获得一个literal规则子节点或expr规则子节点、内部alt的令牌子节点和另一个expr规则子节点。你所要做的就是检查：

if (ctx.getChildCount() > 1 && ((TerminalNode)ctx.getChild(1)).getSymbol().getType() == YourParser.EQUALS)