java - 从 ANTLR AST 识别 Java 变量

Question

我想列出使用 ANTLR4 的 Java 源代码中使用的所有变量。对于以下源，结果应为 a,b,i。

class Item {}
class First {
    public static void main(String[] args) {
        int a = 3;
        int b;
        b = a + 1;
        Item i = new Item();
        i = new Item();
        System.out.println(b);
    }
}

使用the official grammar我可以这样提取 AST: ( view larger )

问题是我不知道如何区分变量和例如系统，因为它们都被标记为表达式 -> 主要。

我想这是一个常见问题，但我找不到任何合适的答案。谁能帮我？

Answer 1

只需重写 JavaBaseListener 中的 enterVariableDeclarator(...) 方法即可获取文本 variableDeclaratorId 匹配项:

public class Main {

    public static void main(String[] args) throws Exception {

        final String source = "class Item {}\n" +
                "class First {\n" +
                "    public static void main(String[] args) {\n" +
                "        int a = 3;\n" +
                "        int b;\n" +
                "        b = a + 1;\n" +
                "        Item i = new Item();\n" +
                "        i = new Item();\n" +
                "        System.out.println(b);\n" +
                "    }\n" +
                "}";

        final JavaLexer lexer = new JavaLexer(new ANTLRInputStream(source));
        final JavaParser parser = new JavaParser(new CommonTokenStream(lexer));
        final List<String> variables = new ArrayList<>();

        ParseTreeWalker.DEFAULT.walk(new JavaBaseListener() {
            // variableDeclarator
            //     :   variableDeclaratorId ('=' variableInitializer)?
            //     ;
            @Override
            public void enterVariableDeclarator(@NotNull JavaParser.VariableDeclaratorContext ctx) {
                variables.add(ctx.variableDeclaratorId().getText());
            }
        }, parser.compilationUnit());

        System.out.printf("variable=%s", variables);
    }
}

将打印:

variable=[a, b, i]